mshtml: COM cleanup for the IHTMLDOMNode2 iface.
[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     hr = IKindaEnumWidget_Next(pKEW, &pWidget);
999     ok_ole_success(hr, IKindaEnumWidget_Next);
1000
1001     IKindaEnumWidget_Release(pKEW);
1002
1003     hr = IWidget_QueryInterface(pWidget, &IID_IDispatch, (void **)&pDispatch);
1004
1005     /* call put_Name */
1006     VariantInit(&vararg[0]);
1007     dispparams.cNamedArgs = 1;
1008     dispparams.rgdispidNamedArgs = &dispidNamed;
1009     dispparams.cArgs = 1;
1010     dispparams.rgvarg = vararg;
1011     VariantInit(&varresult);
1012     hr = IDispatch_Invoke(pDispatch, DISPID_TM_NAME, &IID_NULL, LOCALE_NEUTRAL, DISPATCH_PROPERTYPUT, &dispparams, &varresult, &excepinfo, NULL);
1013     ok_ole_success(hr, IDispatch_Invoke);
1014     ok(excepinfo.wCode == 0x0 && excepinfo.scode == S_OK,
1015         "EXCEPINFO differs from expected: wCode = 0x%x, scode = 0x%08x\n",
1016         excepinfo.wCode, excepinfo.scode);
1017     VariantClear(&varresult);
1018
1019     /* call put_Name (direct) */
1020     bstr = SysAllocString(szSuperman);
1021     hr = IWidget_put_Name(pWidget, bstr);
1022     ok_ole_success(hr, IWidget_put_Name);
1023     SysFreeString(bstr);
1024
1025     /* call get_Name */
1026     dispparams.cNamedArgs = 0;
1027     dispparams.rgdispidNamedArgs = NULL;
1028     dispparams.cArgs = 0;
1029     dispparams.rgvarg = NULL;
1030     VariantInit(&varresult);
1031     hr = IDispatch_Invoke(pDispatch, DISPID_TM_NAME, &IID_NULL, LOCALE_NEUTRAL, DISPATCH_PROPERTYGET, &dispparams, &varresult, &excepinfo, NULL);
1032     ok_ole_success(hr, IDispatch_Invoke);
1033     ok(excepinfo.wCode == 0x0 && excepinfo.scode == S_OK,
1034         "EXCEPINFO differs from expected: wCode = 0x%x, scode = 0x%08x\n",
1035         excepinfo.wCode, excepinfo.scode);
1036     trace("Name = %s\n", wine_dbgstr_w(V_BSTR(&varresult)));
1037     VariantClear(&varresult);
1038
1039     /* call get_Name (direct) */
1040     bstr = NULL;
1041     hr = IWidget_get_Name(pWidget, &bstr);
1042     ok_ole_success(hr, IWidget_get_Name);
1043     ok(!lstrcmpW(bstr, szCat), "IWidget_get_Name should have returned string \"Cat\" instead of %s\n", wine_dbgstr_w(bstr));
1044     SysFreeString(bstr);
1045
1046     /* call DoSomething */
1047     VariantInit(&vararg[0]);
1048     VariantInit(&vararg[1]);
1049     V_VT(&vararg[1]) = VT_R8;
1050     V_R8(&vararg[1]) = 3.141;
1051     dispparams.cNamedArgs = 0;
1052     dispparams.cArgs = 2;
1053     dispparams.rgdispidNamedArgs = NULL;
1054     dispparams.rgvarg = vararg;
1055     VariantInit(&varresult);
1056     hr = IDispatch_Invoke(pDispatch, DISPID_TM_DOSOMETHING, &IID_NULL, LOCALE_NEUTRAL, DISPATCH_METHOD, &dispparams, &varresult, &excepinfo, NULL);
1057     ok_ole_success(hr, IDispatch_Invoke);
1058     ok(V_VT(&varresult) == VT_EMPTY, "varresult should be VT_EMPTY\n");
1059     VariantClear(&varresult);
1060
1061     /* call get_State */
1062     dispparams.cNamedArgs = 0;
1063     dispparams.cArgs = 0;
1064     dispparams.rgdispidNamedArgs = NULL;
1065     dispparams.rgvarg = NULL;
1066     hr = IDispatch_Invoke(pDispatch, DISPID_TM_STATE, &IID_NULL, LOCALE_NEUTRAL, DISPATCH_PROPERTYGET, &dispparams, &varresult, &excepinfo, NULL);
1067     ok_ole_success(hr, IDispatch_Invoke);
1068     ok((V_VT(&varresult) == VT_I4) && (V_I4(&varresult) == STATE_WIDGETIFIED), "Return val mismatch\n");
1069
1070     /* call get_State (direct) */
1071     hr = IWidget_get_State(pWidget, &the_state);
1072     ok_ole_success(hr, IWidget_get_state);
1073     ok(the_state == STATE_WIDGETIFIED, "should have returned WIDGET_WIDGETIFIED instead of %d\n", the_state);
1074
1075     /* call put_State */
1076     the_state = STATE_WIDGETIFIED;
1077     VariantInit(&vararg[0]);
1078     V_VT(&vararg[0]) = VT_BYREF|VT_I4;
1079     V_I4REF(&vararg[0]) = (int *)&the_state;
1080     dispparams.cNamedArgs = 1;
1081     dispparams.cArgs = 1;
1082     dispparams.rgdispidNamedArgs = &dispidNamed;
1083     dispparams.rgvarg = vararg;
1084     hr = IDispatch_Invoke(pDispatch, DISPID_TM_STATE, &IID_NULL, LOCALE_NEUTRAL, DISPATCH_PROPERTYPUT, &dispparams, &varresult, &excepinfo, NULL);
1085     ok_ole_success(hr, IDispatch_Invoke);
1086
1087     /* call Map */
1088     bstr = SysAllocString(szTestTest);
1089     VariantInit(&vararg[0]);
1090     V_VT(&vararg[0]) = VT_BYREF|VT_BSTR;
1091     V_BSTRREF(&vararg[0]) = &bstr;
1092     dispparams.cNamedArgs = 0;
1093     dispparams.cArgs = 1;
1094     dispparams.rgdispidNamedArgs = NULL;
1095     dispparams.rgvarg = vararg;
1096     VariantInit(&varresult);
1097     hr = IDispatch_Invoke(pDispatch, DISPID_TM_MAP, &IID_NULL, LOCALE_NEUTRAL, DISPATCH_METHOD, &dispparams, &varresult, &excepinfo, NULL);
1098     ok_ole_success(hr, IDispatch_Invoke);
1099     ok(V_VT(&varresult) == VT_BSTR, "Return value should be of type BSTR instead of %d\n", V_VT(&varresult));
1100     ok(!lstrcmpW(V_BSTR(&varresult), szTestTest), "Return value should have been \"TestTest\" instead of %s\n", wine_dbgstr_w(V_BSTR(&varresult)));
1101     VariantClear(&varresult);
1102
1103     /* call SetOleColor with large negative VT_I4 param */
1104     VariantInit(&vararg[0]);
1105     V_VT(&vararg[0]) = VT_I4;
1106     V_I4(&vararg[0]) = 0x80000005;
1107     dispparams.cNamedArgs = 0;
1108     dispparams.cArgs = 1;
1109     dispparams.rgdispidNamedArgs = NULL;
1110     dispparams.rgvarg = vararg;
1111     hr = IDispatch_Invoke(pDispatch, DISPID_TM_SETOLECOLOR, &IID_NULL, LOCALE_NEUTRAL, DISPATCH_METHOD, &dispparams, NULL, &excepinfo, NULL);
1112     ok_ole_success(hr, IDispatch_Invoke);
1113
1114     /* call GetOleColor */
1115     dispparams.cNamedArgs = 0;
1116     dispparams.cArgs = 0;
1117     dispparams.rgdispidNamedArgs = NULL;
1118     dispparams.rgvarg = NULL;
1119     VariantInit(&varresult);
1120     hr = IDispatch_Invoke(pDispatch, DISPID_TM_GETOLECOLOR, &IID_NULL, LOCALE_NEUTRAL, DISPATCH_METHOD, &dispparams, &varresult, &excepinfo, NULL);
1121     ok_ole_success(hr, IDispatch_Invoke);
1122     VariantClear(&varresult);
1123
1124     /* call StructArgs (direct) */
1125     mystruct = MYSTRUCT_BYPTR;
1126     memcpy(mystructArray, MYSTRUCT_ARRAY, sizeof(mystructArray));
1127     hr = IWidget_StructArgs(pWidget, MYSTRUCT_BYVAL, &mystruct, mystructArray);
1128     ok_ole_success(hr, IWidget_StructArgs);
1129
1130     /* call Clone */
1131     dispparams.cNamedArgs = 0;
1132     dispparams.cArgs = 0;
1133     dispparams.rgdispidNamedArgs = NULL;
1134     dispparams.rgvarg = NULL;
1135     VariantInit(&varresult);
1136     hr = IDispatch_Invoke(pDispatch, DISPID_TM_CLONE, &IID_NULL, LOCALE_NEUTRAL, DISPATCH_PROPERTYGET, &dispparams, &varresult, &excepinfo, NULL);
1137     ok_ole_success(hr, IDispatch_Invoke);
1138     ok(V_VT(&varresult) == VT_DISPATCH, "vt %x\n", V_VT(&varresult));
1139     VariantClear(&varresult);
1140
1141     /* call CloneInterface */
1142     dispparams.cNamedArgs = 0;
1143     dispparams.cArgs = 0;
1144     dispparams.rgdispidNamedArgs = NULL;
1145     dispparams.rgvarg = NULL;
1146     VariantInit(&varresult);
1147     hr = IDispatch_Invoke(pDispatch, DISPID_TM_CLONEINTERFACE, &IID_NULL, LOCALE_NEUTRAL, DISPATCH_PROPERTYGET, &dispparams, &varresult, &excepinfo, NULL);
1148     ok_ole_success(hr, IDispatch_Invoke);
1149     ok(V_VT(&varresult) == VT_DISPATCH, "vt %x\n", V_VT(&varresult));
1150     VariantClear(&varresult);
1151
1152     /* call CloneDispatch with automatic value getting */
1153     V_VT(&vararg[0]) = VT_I2;
1154     V_I2(&vararg[0]) = 1;
1155     dispparams.cNamedArgs = 0;
1156     dispparams.rgdispidNamedArgs = NULL;
1157     dispparams.cArgs = 1;
1158     dispparams.rgvarg = vararg;
1159     VariantInit(&varresult);
1160     hr = IDispatch_Invoke(pDispatch, DISPID_TM_CLONEDISPATCH, &IID_NULL, LOCALE_NEUTRAL, DISPATCH_PROPERTYGET, &dispparams, &varresult, &excepinfo, NULL);
1161     ok_ole_success(hr, IDispatch_Invoke);
1162
1163     ok(excepinfo.wCode == 0x0 && excepinfo.scode == S_OK,
1164         "EXCEPINFO differs from expected: wCode = 0x%x, scode = 0x%08x\n",
1165         excepinfo.wCode, excepinfo.scode);
1166
1167     ok(V_VT(&varresult) == VT_I2, "V_VT(&varresult) was %d instead of VT_I2\n", V_VT(&varresult));
1168     ok(V_I2(&varresult) == 1234, "V_I2(&varresult) was %d instead of 1234\n", V_I2(&varresult));
1169     VariantClear(&varresult);
1170
1171     /* call CloneCoclass */
1172     dispparams.cNamedArgs = 0;
1173     dispparams.cArgs = 0;
1174     dispparams.rgdispidNamedArgs = NULL;
1175     dispparams.rgvarg = NULL;
1176     VariantInit(&varresult);
1177     hr = IDispatch_Invoke(pDispatch, DISPID_TM_CLONECOCLASS, &IID_NULL, LOCALE_NEUTRAL, DISPATCH_PROPERTYGET, &dispparams, &varresult, &excepinfo, NULL);
1178     ok_ole_success(hr, IDispatch_Invoke);
1179
1180     ok(excepinfo.wCode == 0x0 && excepinfo.scode == S_OK,
1181        "EXCEPINFO differs from expected: wCode = 0x%x, scode = 0x%08x\n",
1182        excepinfo.wCode, excepinfo.scode);
1183
1184     ok(V_VT(&varresult) == VT_DISPATCH, "V_VT(&varresult) was %d instead of VT_DISPATCH\n", V_VT(&varresult));
1185     ok(!V_DISPATCH(&varresult), "V_DISPATCH(&varresult) should be NULL instead of %p\n", V_DISPATCH(&varresult));
1186     VariantClear(&varresult);
1187
1188     /* call Value with a VT_VARIANT|VT_BYREF type */
1189     V_VT(&vararg[0]) = VT_VARIANT|VT_BYREF;
1190     V_VARIANTREF(&vararg[0]) = &vararg[1];
1191     V_VT(&vararg[1]) = VT_I2;
1192     V_I2(&vararg[1]) = 1;
1193     dispparams.cNamedArgs = 0;
1194     dispparams.rgdispidNamedArgs = NULL;
1195     dispparams.cArgs = 1;
1196     dispparams.rgvarg = vararg;
1197     VariantInit(&varresult);
1198     hr = IDispatch_Invoke(pDispatch, DISPID_VALUE, &IID_NULL, LOCALE_NEUTRAL, DISPATCH_PROPERTYGET, &dispparams, &varresult, &excepinfo, NULL);
1199     ok_ole_success(hr, IDispatch_Invoke);
1200
1201     ok(excepinfo.wCode == 0x0 && excepinfo.scode == S_OK,
1202         "EXCEPINFO differs from expected: wCode = 0x%x, scode = 0x%08x\n",
1203         excepinfo.wCode, excepinfo.scode);
1204
1205     ok(V_VT(&varresult) == VT_I2, "V_VT(&varresult) was %d instead of VT_I2\n", V_VT(&varresult));
1206     ok(V_I2(&varresult) == 1234, "V_I2(&varresult) was %d instead of 1234\n", V_I2(&varresult));
1207     VariantClear(&varresult);
1208
1209     /* call Variant - exercises variant copying in ITypeInfo::Invoke and
1210      * handling of void return types */
1211     /* use a big type to ensure that the variant was properly copied into the
1212      * destination function's args */
1213     V_VT(&vararg[0]) = VT_CY;
1214     S(V_CY(&vararg[0])).Hi = 0xdababe;
1215     S(V_CY(&vararg[0])).Lo = 0xdeadbeef;
1216     dispparams.cNamedArgs = 0;
1217     dispparams.cArgs = 1;
1218     dispparams.rgdispidNamedArgs = NULL;
1219     dispparams.rgvarg = vararg;
1220     VariantInit(&varresult);
1221     hr = IDispatch_Invoke(pDispatch, DISPID_TM_VARIANT, &IID_NULL, LOCALE_NEUTRAL, DISPATCH_METHOD, &dispparams, NULL, NULL, NULL);
1222     ok_ole_success(hr, IDispatch_Invoke);
1223     VariantClear(&varresult);
1224
1225     /* call VarArg */
1226     VariantInit(&vararg[3]);
1227     V_VT(&vararg[3]) = VT_I4;
1228     V_I4(&vararg[3]) = 3;
1229     VariantInit(&vararg[2]);
1230     V_VT(&vararg[2]) = VT_I4;
1231     V_I4(&vararg[2]) = 0;
1232     VariantInit(&vararg[1]);
1233     V_VT(&vararg[1]) = VT_I4;
1234     V_I4(&vararg[1]) = 1;
1235     VariantInit(&vararg[0]);
1236     V_VT(&vararg[0]) = VT_I4;
1237     V_I4(&vararg[0]) = 2;
1238     dispparams.cNamedArgs = 0;
1239     dispparams.cArgs = 4;
1240     dispparams.rgdispidNamedArgs = NULL;
1241     dispparams.rgvarg = vararg;
1242     hr = IDispatch_Invoke(pDispatch, DISPID_TM_VARARG, &IID_NULL, LOCALE_NEUTRAL, DISPATCH_METHOD, &dispparams, NULL, NULL, NULL);
1243     ok_ole_success(hr, ITypeInfo_Invoke);
1244
1245     /* call VarArg, even one (non-optional, non-safearray) named argument is not allowed */
1246     dispidNamed = 0;
1247     dispparams.cNamedArgs = 1;
1248     dispparams.rgdispidNamedArgs = &dispidNamed;
1249     hr = IDispatch_Invoke(pDispatch, DISPID_TM_VARARG, &IID_NULL, LOCALE_NEUTRAL, DISPATCH_METHOD, &dispparams, NULL, NULL, NULL);
1250     ok(hr == DISP_E_NONAMEDARGS, "IDispatch_Invoke should have returned DISP_E_NONAMEDARGS instead of 0x%08x\n", hr);
1251     dispidNamed = DISPID_PROPERTYPUT;
1252
1253     /* call Error */
1254     dispparams.cNamedArgs = 0;
1255     dispparams.cArgs = 0;
1256     dispparams.rgdispidNamedArgs = NULL;
1257     dispparams.rgvarg = NULL;
1258     VariantInit(&varresult);
1259     hr = IDispatch_Invoke(pDispatch, DISPID_TM_ERROR, &IID_NULL, LOCALE_NEUTRAL, DISPATCH_METHOD, &dispparams, NULL, &excepinfo, NULL);
1260     ok(hr == DISP_E_EXCEPTION, "IDispatch_Invoke should have returned DISP_E_EXCEPTION instead of 0x%08x\n", hr);
1261     ok(excepinfo.wCode == 0x0 && excepinfo.scode == E_NOTIMPL,
1262         "EXCEPINFO differs from expected: wCode = 0x%x, scode = 0x%08x\n",
1263         excepinfo.wCode, excepinfo.scode);
1264     VariantClear(&varresult);
1265
1266     /* call BstrRet */
1267     pTypeInfo = NonOleAutomation_GetTypeInfo();
1268     dispparams.cNamedArgs = 0;
1269     dispparams.cArgs = 0;
1270     dispparams.rgdispidNamedArgs = NULL;
1271     dispparams.rgvarg = NULL;
1272     VariantInit(&varresult);
1273     hr = ITypeInfo_Invoke(pTypeInfo, &NonOleAutomation, DISPID_NOA_BSTRRET, DISPATCH_METHOD, &dispparams, &varresult, &excepinfo, NULL);
1274     ok_ole_success(hr, ITypeInfo_Invoke);
1275     ok(V_VT(&varresult) == VT_BSTR, "V_VT(&varresult) should be VT_BSTR instead of %d\n", V_VT(&varresult));
1276     ok(V_BSTR(&varresult) != NULL, "V_BSTR(&varresult) should not be NULL\n");
1277
1278     VariantClear(&varresult);
1279
1280     dispparams.cNamedArgs = 0;
1281     dispparams.cArgs = 0;
1282     dispparams.rgdispidNamedArgs = NULL;
1283     dispparams.rgvarg = NULL;
1284     hr = ITypeInfo_Invoke(pTypeInfo, &NonOleAutomation, DISPID_NOA_ERROR, DISPATCH_METHOD, &dispparams, &varresult, &excepinfo, NULL);
1285     ok(hr == DISP_E_EXCEPTION, "ITypeInfo_Invoke should have returned DISP_E_EXCEPTION instead of 0x%08x\n", hr);
1286     ok(V_VT(&varresult) == VT_EMPTY, "V_VT(&varresult) should be VT_EMPTY instead of %d\n", V_VT(&varresult));
1287     ok(excepinfo.wCode == 0x0 && excepinfo.scode == E_NOTIMPL,
1288         "EXCEPINFO differs from expected: wCode = 0x%x, scode = 0x%08x\n",
1289         excepinfo.wCode, excepinfo.scode);
1290     VariantClear(&varresult);
1291
1292     ITypeInfo_Release(pTypeInfo);
1293
1294     /* tests call put_Name without named arg */
1295     VariantInit(&vararg[0]);
1296     dispparams.cNamedArgs = 0;
1297     dispparams.rgdispidNamedArgs = NULL;
1298     dispparams.cArgs = 1;
1299     dispparams.rgvarg = vararg;
1300     VariantInit(&varresult);
1301     hr = IDispatch_Invoke(pDispatch, DISPID_TM_NAME, &IID_NULL, LOCALE_NEUTRAL, DISPATCH_PROPERTYPUT, &dispparams, &varresult, &excepinfo, NULL);
1302     ok(hr == DISP_E_PARAMNOTFOUND, "IDispatch_Invoke should have returned DISP_E_PARAMNOTFOUND instead of 0x%08x\n", hr);
1303     VariantClear(&varresult);
1304
1305     /* tests param type that cannot be coerced */
1306     VariantInit(&vararg[0]);
1307     V_VT(&vararg[0]) = VT_UNKNOWN;
1308     V_UNKNOWN(&vararg[0]) = NULL;
1309     dispparams.cNamedArgs = 1;
1310     dispparams.rgdispidNamedArgs = &dispidNamed;
1311     dispparams.cArgs = 1;
1312     dispparams.rgvarg = vararg;
1313     VariantInit(&varresult);
1314     hr = IDispatch_Invoke(pDispatch, DISPID_TM_NAME, &IID_NULL, LOCALE_NEUTRAL, DISPATCH_PROPERTYPUT, &dispparams, &varresult, &excepinfo, NULL);
1315     ok(hr == DISP_E_TYPEMISMATCH, "IDispatch_Invoke should have returned DISP_E_TYPEMISMATCH instead of 0x%08x\n", hr);
1316     VariantClear(&varresult);
1317
1318     /* tests bad param type */
1319     VariantInit(&vararg[0]);
1320     V_VT(&vararg[0]) = VT_CLSID;
1321     V_BYREF(&vararg[0]) = NULL;
1322     dispparams.cNamedArgs = 1;
1323     dispparams.rgdispidNamedArgs = &dispidNamed;
1324     dispparams.cArgs = 1;
1325     dispparams.rgvarg = vararg;
1326     VariantInit(&varresult);
1327     hr = IDispatch_Invoke(pDispatch, DISPID_TM_NAME, &IID_NULL, LOCALE_NEUTRAL, DISPATCH_PROPERTYPUT, &dispparams, &varresult, &excepinfo, NULL);
1328     ok(hr == DISP_E_BADVARTYPE, "IDispatch_Invoke should have returned DISP_E_BADVARTYPE instead of 0x%08x\n", hr);
1329     VariantClear(&varresult);
1330
1331     /* tests too small param count */
1332     dispparams.cNamedArgs = 0;
1333     dispparams.rgdispidNamedArgs = NULL;
1334     dispparams.cArgs = 0;
1335     dispparams.rgvarg = NULL;
1336     VariantInit(&varresult);
1337     hr = IDispatch_Invoke(pDispatch, DISPID_TM_DOSOMETHING, &IID_NULL, LOCALE_NEUTRAL, DISPATCH_METHOD, &dispparams, &varresult, &excepinfo, NULL);
1338     ok(hr == DISP_E_BADPARAMCOUNT, "IDispatch_Invoke should have returned DISP_E_BADPARAMCOUNT instead of 0x%08x\n", hr);
1339     VariantClear(&varresult);
1340
1341     /* tests propget function with large param count */
1342     VariantInit(&vararg[0]);
1343     V_VT(&vararg[0]) = VT_BSTR;
1344     V_BSTR(&vararg[0]) = NULL;
1345     V_VT(&vararg[1]) = VT_I4;
1346     V_I4(&vararg[1]) = 1;
1347     dispparams.cNamedArgs = 0;
1348     dispparams.cArgs = 2;
1349     dispparams.rgdispidNamedArgs = NULL;
1350     dispparams.rgvarg = vararg;
1351     hr = IDispatch_Invoke(pDispatch, DISPID_TM_STATE, &IID_NULL, LOCALE_NEUTRAL, DISPATCH_PROPERTYGET, &dispparams, &varresult, &excepinfo, NULL);
1352     ok(hr == DISP_E_NOTACOLLECTION, "IDispatch_Invoke should have returned DISP_E_NOTACOLLECTION instead of 0x%08x\n", hr);
1353
1354     /* test propput with lcid */
1355
1356     /* the lcid passed to the function is the first lcid in the typelib header.
1357        Since we don't explicitly set an lcid in the idl, it'll default to US English. */
1358     VariantInit(&vararg[0]);
1359     V_VT(&vararg[0]) = VT_I4;
1360     V_I4(&vararg[0]) = 0xcafe;
1361     dispparams.cNamedArgs = 1;
1362     dispparams.rgdispidNamedArgs = &dispidNamed;
1363     dispparams.cArgs = 1;
1364     dispparams.rgvarg = vararg;
1365     VariantInit(&varresult);
1366     hr = IDispatch_Invoke(pDispatch, DISPID_TM_PROP_WITH_LCID, &IID_NULL, 0x40c, DISPATCH_PROPERTYPUT, &dispparams, &varresult, &excepinfo, NULL);
1367     ok_ole_success(hr, ITypeInfo_Invoke);
1368     VariantClear(&varresult);
1369
1370     /* test propget with lcid */
1371     dispparams.cNamedArgs = 0;
1372     dispparams.cArgs = 0;
1373     dispparams.rgvarg = NULL;
1374     dispparams.rgdispidNamedArgs = NULL;
1375     hr = IDispatch_Invoke(pDispatch, DISPID_TM_PROP_WITH_LCID, &IID_NULL, 0x40c, DISPATCH_PROPERTYGET, &dispparams, &varresult, &excepinfo, NULL);
1376     ok_ole_success(hr, ITypeInfo_Invoke);
1377     ok(V_VT(&varresult) == VT_I4, "got %x\n", V_VT(&varresult));
1378     ok(V_I4(&varresult) == 0x409, "got %x\n", V_I4(&varresult));
1379     VariantClear(&varresult);
1380
1381     /* test propget of INT value */
1382     dispparams.cNamedArgs = 0;
1383     dispparams.cArgs = 0;
1384     dispparams.rgvarg = NULL;
1385     dispparams.rgdispidNamedArgs = NULL;
1386     hr = IDispatch_Invoke(pDispatch, DISPID_TM_PROP_INT, &IID_NULL, 0x40c, DISPATCH_PROPERTYGET, &dispparams, &varresult, &excepinfo, NULL);
1387     ok_ole_success(hr, ITypeInfo_Invoke);
1388     ok(V_VT(&varresult) == VT_I4, "got %x\n", V_VT(&varresult));
1389     ok(V_I4(&varresult) == -13, "got %x\n", V_I4(&varresult));
1390     VariantClear(&varresult);
1391
1392     /* test propget of INT value */
1393     dispparams.cNamedArgs = 0;
1394     dispparams.cArgs = 0;
1395     dispparams.rgvarg = NULL;
1396     dispparams.rgdispidNamedArgs = NULL;
1397     hr = IDispatch_Invoke(pDispatch, DISPID_TM_PROP_UINT, &IID_NULL, 0x40c, DISPATCH_PROPERTYGET, &dispparams, &varresult, &excepinfo, NULL);
1398     ok_ole_success(hr, ITypeInfo_Invoke);
1399     ok(V_VT(&varresult) == VT_UI4, "got %x\n", V_VT(&varresult));
1400     ok(V_UI4(&varresult) == 42, "got %x\n", V_UI4(&varresult));
1401     VariantClear(&varresult);
1402
1403     /* test byref marshalling */
1404     uval = 666;
1405     VariantInit(&vararg[0]);
1406     V_VT(&vararg[0]) = VT_UI4|VT_BYREF;
1407     V_UI4REF(&vararg[0]) = &uval;
1408     dispparams.cNamedArgs = 0;
1409     dispparams.cArgs = 1;
1410     dispparams.rgvarg = vararg;
1411     dispparams.rgdispidNamedArgs = NULL;
1412     hr = IDispatch_Invoke(pDispatch, DISPID_TM_BYREF_UINT, &IID_NULL, LOCALE_NEUTRAL, DISPATCH_METHOD, &dispparams, &varresult, &excepinfo, NULL);
1413     ok_ole_success(hr, ITypeInfo_Invoke);
1414     ok(V_VT(&varresult) == VT_EMPTY, "varresult should be VT_EMPTY\n");
1415     ok(V_VT(&vararg[0]) == (VT_UI4|VT_BYREF), "arg VT not unmarshalled correctly: %x\n", V_VT(&vararg[0]));
1416     ok(V_UI4REF(&vararg[0]) == &uval, "Byref pointer not preserved: %p/%p\n", &uval, V_UI4REF(&vararg[0]));
1417     ok(*V_UI4REF(&vararg[0]) == 42, "Expected 42 to be returned instead of %u\n", *V_UI4REF(&vararg[0]));
1418     VariantClear(&varresult);
1419     VariantClear(&vararg[0]);
1420
1421     /* test propput with optional argument. */
1422     VariantInit(&vararg[0]);
1423     V_VT(&vararg[0]) = VT_I4;
1424     V_I4(&vararg[0]) = 0xcafe;
1425     dispparams.cNamedArgs = 1;
1426     dispparams.rgdispidNamedArgs = &dispidNamed;
1427     dispparams.cArgs = 1;
1428     dispparams.rgvarg = vararg;
1429     VariantInit(&varresult);
1430     hr = IDispatch_Invoke(pDispatch, DISPID_TM_PROP_OPT_ARG, &IID_NULL, 0x40c, DISPATCH_PROPERTYPUT, &dispparams, &varresult, &excepinfo, NULL);
1431     ok_ole_success(hr, ITypeInfo_Invoke);
1432     VariantClear(&varresult);
1433
1434     /* test propput with required argument. */
1435     VariantInit(&vararg[0]);
1436     VariantInit(&vararg[1]);
1437     V_VT(&vararg[0]) = VT_I4;
1438     V_I4(&vararg[0]) = 0x1234;
1439     V_VT(&vararg[1]) = VT_I4;
1440     V_I4(&vararg[1]) = 0x5678;
1441     dispparams.cNamedArgs = 1;
1442     dispparams.rgdispidNamedArgs = &dispidNamed;
1443     dispparams.cArgs = 2;
1444     dispparams.rgvarg = vararg;
1445     VariantInit(&varresult);
1446     hr = IDispatch_Invoke(pDispatch, DISPID_TM_PROP_REQ_ARG, &IID_NULL, 0x40c, DISPATCH_PROPERTYPUT, &dispparams, &varresult, &excepinfo, NULL);
1447     ok_ole_success(hr, ITypeInfo_Invoke);
1448     VariantClear(&varresult);
1449
1450     IDispatch_Release(pDispatch);
1451     IWidget_Release(pWidget);
1452
1453     trace("calling end_host_object\n");
1454     end_host_object(tid, thread);
1455 }
1456
1457 static void test_DispCallFunc(void)
1458 {
1459     static const WCHAR szEmpty[] = { 0 };
1460     VARTYPE rgvt[] = { VT_R8, VT_BSTR, VT_BSTR, VT_VARIANT|VT_BYREF };
1461     VARIANTARG vararg[4];
1462     VARIANTARG varref;
1463     VARIANTARG *rgpvarg[4] = { &vararg[0], &vararg[1], &vararg[2], &vararg[3] };
1464     VARIANTARG varresult;
1465     HRESULT hr;
1466     IWidget *pWidget = Widget_Create();
1467     V_VT(&vararg[0]) = VT_R8;
1468     V_R8(&vararg[0]) = 3.141;
1469     V_VT(&vararg[1]) = VT_BSTR;
1470     V_BSTRREF(&vararg[1]) = CoTaskMemAlloc(sizeof(BSTR));
1471     V_VT(&vararg[2]) = VT_BSTR;
1472     V_BSTR(&vararg[2]) = SysAllocString(szEmpty);
1473     V_VT(&vararg[3]) = VT_VARIANT|VT_BYREF;
1474     V_VARIANTREF(&vararg[3]) = &varref;
1475     V_VT(&varref) = VT_ERROR;
1476     V_ERROR(&varref) = DISP_E_PARAMNOTFOUND;
1477     VariantInit(&varresult);
1478     hr = DispCallFunc(pWidget, 9*sizeof(void*), CC_STDCALL, VT_UI4, 4, rgvt, rgpvarg, &varresult);
1479     ok_ole_success(hr, DispCallFunc);
1480     VariantClear(&varresult);
1481     SysFreeString(*V_BSTRREF(&vararg[1]));
1482     CoTaskMemFree(V_BSTRREF(&vararg[1]));
1483     VariantClear(&vararg[2]);
1484     IWidget_Release(pWidget);
1485 }
1486
1487 static void test_StaticWidget(void)
1488 {
1489     ITypeInfo *type_info;
1490     DISPPARAMS dispparams;
1491     VARIANTARG vararg[4];
1492     EXCEPINFO excepinfo;
1493     VARIANT varresult;
1494     HRESULT hr;
1495
1496     type_info = get_type_info(&IID_IStaticWidget);
1497
1498     /* call TestDual */
1499     dispparams.cNamedArgs = 0;
1500     dispparams.cArgs = 1;
1501     dispparams.rgdispidNamedArgs = NULL;
1502     dispparams.rgvarg = vararg;
1503     V_VT(vararg) = VT_DISPATCH;
1504     V_DISPATCH(vararg) = (IDispatch*)&TestDualDisp;
1505     VariantInit(&varresult);
1506     hr = ITypeInfo_Invoke(type_info, &StaticWidget, DISPID_TM_TESTDUAL, DISPATCH_METHOD,
1507             &dispparams, &varresult, &excepinfo, NULL);
1508     ok_ole_success(hr, IDispatch_Invoke);
1509     ok(V_VT(&varresult) == VT_EMPTY, "vt %x\n", V_VT(&varresult));
1510     VariantClear(&varresult);
1511
1512     ITypeInfo_Release(type_info);
1513 }
1514
1515 static void test_libattr(void)
1516 {
1517     ITypeLib *pTypeLib;
1518     HRESULT hr;
1519     TLIBATTR *pattr;
1520
1521     hr = LoadRegTypeLib(&LIBID_TestTypelib, 1, 0, LOCALE_NEUTRAL, &pTypeLib);
1522     ok_ole_success(hr, LoadRegTypeLib);
1523     if (FAILED(hr))
1524         return;
1525
1526     hr = ITypeLib_GetLibAttr(pTypeLib, &pattr);
1527     ok_ole_success(hr, GetLibAttr);
1528     if (SUCCEEDED(hr))
1529     {
1530         ok(pattr->lcid == MAKELANGID(LANG_NEUTRAL, SUBLANG_NEUTRAL), "lcid %x\n", pattr->lcid);
1531
1532         ITypeLib_ReleaseTLibAttr(pTypeLib, pattr);
1533     }
1534
1535     ITypeLib_Release(pTypeLib);
1536 }
1537
1538 START_TEST(tmarshal)
1539 {
1540     HRESULT hr;
1541
1542     CoInitializeEx(NULL, COINIT_APARTMENTTHREADED);
1543
1544     hr = register_current_module_typelib();
1545     ok_ole_success(hr, register_current_module_typelib);
1546
1547     test_typelibmarshal();
1548     test_DispCallFunc();
1549     test_StaticWidget();
1550     test_libattr();
1551
1552     hr = UnRegisterTypeLib(&LIBID_TestTypelib, 1, 0, LOCALE_NEUTRAL,
1553                            sizeof(void*) == 8 ? SYS_WIN64 : SYS_WIN32);
1554     ok_ole_success(hr, UnRegisterTypeLib);
1555
1556     CoUninitialize();
1557 }