oleaut32/tests: Checking null terminator in return value of SysReAllocStringLen.
[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     const IWidgetVtbl *lpVtbl;
220     LONG refs;
221     IUnknown *pDispatchUnknown;
222 } Widget;
223
224 static HRESULT WINAPI Widget_QueryInterface(
225     IWidget *iface,
226     /* [in] */ REFIID riid,
227     /* [iid_is][out] */ void __RPC_FAR *__RPC_FAR *ppvObject)
228 {
229     if (IsEqualIID(riid, &IID_IWidget) || IsEqualIID(riid, &IID_IUnknown) || IsEqualIID(riid, &IID_IDispatch))
230     {
231         IWidget_AddRef(iface);
232         *ppvObject = iface;
233         return S_OK;
234     }
235     else
236     {
237         *ppvObject = NULL;
238         return E_NOINTERFACE;
239     }
240 }
241
242 static ULONG WINAPI Widget_AddRef(
243     IWidget *iface)
244 {
245     Widget *This = (Widget *)iface;
246
247     return InterlockedIncrement(&This->refs);
248 }
249
250 static ULONG WINAPI Widget_Release(
251     IWidget *iface)
252 {
253     Widget *This = (Widget *)iface;
254     ULONG refs = InterlockedDecrement(&This->refs);
255     if (!refs)
256     {
257         IUnknown_Release(This->pDispatchUnknown);
258         memset(This, 0xcc, sizeof(*This));
259         HeapFree(GetProcessHeap(), 0, This);
260         trace("Widget destroyed!\n");
261     }
262
263     return refs;
264 }
265
266 static HRESULT WINAPI Widget_GetTypeInfoCount(
267     IWidget *iface,
268     /* [out] */ UINT __RPC_FAR *pctinfo)
269 {
270     Widget *This = (Widget *)iface;
271     IDispatch *pDispatch;
272     HRESULT hr = IUnknown_QueryInterface(This->pDispatchUnknown, &IID_IDispatch, (void **)&pDispatch);
273     if (SUCCEEDED(hr))
274     {
275         hr = IDispatch_GetTypeInfoCount(pDispatch, pctinfo);
276         IDispatch_Release(pDispatch);
277     }
278     return hr;
279 }
280
281 static HRESULT WINAPI Widget_GetTypeInfo(
282     IWidget __RPC_FAR * iface,
283     /* [in] */ UINT iTInfo,
284     /* [in] */ LCID lcid,
285     /* [out] */ ITypeInfo __RPC_FAR *__RPC_FAR *ppTInfo)
286 {
287     Widget *This = (Widget *)iface;
288     IDispatch *pDispatch;
289     HRESULT hr = IUnknown_QueryInterface(This->pDispatchUnknown, &IID_IDispatch, (void **)&pDispatch);
290     if (SUCCEEDED(hr))
291     {
292         hr = IDispatch_GetTypeInfo(pDispatch, iTInfo, lcid, ppTInfo);
293         IDispatch_Release(pDispatch);
294     }
295     return hr;
296 }
297
298 static HRESULT WINAPI Widget_GetIDsOfNames(
299     IWidget __RPC_FAR * iface,
300     /* [in] */ REFIID riid,
301     /* [size_is][in] */ LPOLESTR __RPC_FAR *rgszNames,
302     /* [in] */ UINT cNames,
303     /* [in] */ LCID lcid,
304     /* [size_is][out] */ DISPID __RPC_FAR *rgDispId)
305 {
306     Widget *This = (Widget *)iface;
307     IDispatch *pDispatch;
308     HRESULT hr = IUnknown_QueryInterface(This->pDispatchUnknown, &IID_IDispatch, (void **)&pDispatch);
309     if (SUCCEEDED(hr))
310     {
311         hr = IDispatch_GetIDsOfNames(pDispatch, riid, rgszNames, cNames, lcid, rgDispId);
312         IDispatch_Release(pDispatch);
313     }
314     return hr;
315 }
316
317 static HRESULT WINAPI Widget_Invoke(
318     IWidget __RPC_FAR * iface,
319     /* [in] */ DISPID dispIdMember,
320     /* [in] */ REFIID riid,
321     /* [in] */ LCID lcid,
322     /* [in] */ WORD wFlags,
323     /* [out][in] */ DISPPARAMS __RPC_FAR *pDispParams,
324     /* [out] */ VARIANT __RPC_FAR *pVarResult,
325     /* [out] */ EXCEPINFO __RPC_FAR *pExcepInfo,
326     /* [out] */ UINT __RPC_FAR *puArgErr)
327 {
328     Widget *This = (Widget *)iface;
329     IDispatch *pDispatch;
330     HRESULT hr = IUnknown_QueryInterface(This->pDispatchUnknown, &IID_IDispatch, (void **)&pDispatch);
331     if (SUCCEEDED(hr))
332     {
333         hr = IDispatch_Invoke(pDispatch, dispIdMember, riid, lcid, wFlags, pDispParams, pVarResult, pExcepInfo, puArgErr);
334         IDispatch_Release(pDispatch);
335     }
336     return hr;
337 }
338
339 static HRESULT WINAPI Widget_put_Name(
340     IWidget __RPC_FAR * iface,
341     /* [in] */ BSTR name)
342 {
343     trace("put_Name(%s)\n", wine_dbgstr_w(name));
344     return S_OK;
345 }
346
347 static HRESULT WINAPI Widget_get_Name(
348     IWidget __RPC_FAR * iface,
349     /* [out] */ BSTR __RPC_FAR *name)
350 {
351     static const WCHAR szCat[] = { 'C','a','t',0 };
352     trace("get_Name()\n");
353     *name = SysAllocString(szCat);
354     return S_OK;
355 }
356
357 static HRESULT WINAPI Widget_DoSomething(
358     IWidget __RPC_FAR * iface,
359     /* [in] */ double number,
360     /* [out] */ BSTR *str1,
361     /* [defaultvalue][in] */ BSTR str2,
362     /* [optional][in] */ VARIANT __RPC_FAR *opt)
363 {
364     static const WCHAR szString[] = { 'S','t','r','i','n','g',0 };
365     trace("DoSomething()\n");
366
367     ok(number == 3.141, "number(%f) != 3.141\n", number);
368     ok(*str2 == '\0', "str2(%s) != \"\"\n", wine_dbgstr_w(str2));
369     ok(V_VT(opt) == VT_ERROR, "V_VT(opt) should be VT_ERROR instead of 0x%x\n", V_VT(opt));
370     ok(V_ERROR(opt) == DISP_E_PARAMNOTFOUND, "V_ERROR(opt) should be DISP_E_PARAMNOTFOUND instead of 0x%08x\n", V_ERROR(opt));
371     *str1 = SysAllocString(szString);
372
373     return S_FALSE;
374 }
375
376 static HRESULT WINAPI Widget_get_State(
377     IWidget __RPC_FAR * iface,
378     /* [retval][out] */ STATE __RPC_FAR *state)
379 {
380     trace("get_State() = STATE_WIDGETIFIED\n");
381     *state = STATE_WIDGETIFIED;
382     return S_OK;
383 }
384
385 static HRESULT WINAPI Widget_put_State(
386     IWidget __RPC_FAR * iface,
387     /* [in] */ STATE state)
388 {
389     trace("put_State(%d)\n", state);
390     return S_OK;
391 }
392
393 static HRESULT WINAPI Widget_Map(
394     IWidget * iface,
395     BSTR bstrId,
396     BSTR *sValue)
397 {
398     trace("Map(%s, %p)\n", wine_dbgstr_w(bstrId), sValue);
399     *sValue = SysAllocString(bstrId);
400     return S_OK;
401 }
402
403 static HRESULT WINAPI Widget_SetOleColor(
404     IWidget * iface,
405     OLE_COLOR val)
406 {
407     trace("SetOleColor(0x%x)\n", val);
408     return S_OK;
409 }
410
411 static HRESULT WINAPI Widget_GetOleColor(
412     IWidget * iface,
413     OLE_COLOR *pVal)
414 {
415     trace("GetOleColor() = 0x8000000f\n");
416     *pVal = 0x8000000f;
417     return S_FALSE;
418 }
419
420 static HRESULT WINAPI Widget_Clone(
421     IWidget *iface,
422     IWidget **ppVal)
423 {
424     trace("Clone()\n");
425     return Widget_QueryInterface(iface, &IID_IWidget, (void **)ppVal);
426 }
427
428 static HRESULT WINAPI Widget_CloneDispatch(
429     IWidget *iface,
430     IDispatch **ppVal)
431 {
432     trace("CloneDispatch()\n");
433     return Widget_QueryInterface(iface, &IID_IWidget, (void **)ppVal);
434 }
435
436 static HRESULT WINAPI Widget_CloneCoclass(
437     IWidget *iface,
438     ApplicationObject2 **ppVal)
439 {
440     trace("CloneCoclass()\n");
441     return S_OK;
442 }
443
444 static HRESULT WINAPI Widget_Value(
445     IWidget __RPC_FAR * iface,
446     VARIANT *value,
447     VARIANT *retval)
448 {
449     trace("Value(%p, %p)\n", value, retval);
450     ok(V_VT(value) == VT_I2, "V_VT(value) was %d instead of VT_I2\n", V_VT(value));
451     ok(V_I2(value) == 1, "V_I2(value) was %d instead of 1\n", V_I2(value));
452     V_VT(retval) = VT_I2;
453     V_I2(retval) = 1234;
454     return S_OK;
455 }
456
457 static HRESULT WINAPI Widget_Array(
458     IWidget * iface,
459     SAFEARRAY * values)
460 {
461     trace("Array(%p)\n", values);
462     return S_OK;
463 }
464
465 static HRESULT WINAPI Widget_VariantArrayPtr(
466     IWidget * iface,
467     SAFEARRAY ** values)
468 {
469     trace("VariantArrayPtr(%p)\n", values);
470     return S_OK;
471 }
472
473 static HRESULT WINAPI Widget_Variant(
474     IWidget __RPC_FAR * iface,
475     VARIANT var)
476 {
477     trace("Variant()\n");
478     ok(V_VT(&var) == VT_CY, "V_VT(&var) was %d\n", V_VT(&var));
479     ok(S(V_CY(&var)).Hi == 0xdababe, "V_CY(&var).Hi was 0x%x\n", S(V_CY(&var)).Hi);
480     ok(S(V_CY(&var)).Lo == 0xdeadbeef, "V_CY(&var).Lo was 0x%x\n", S(V_CY(&var)).Lo);
481     return S_OK;
482 }
483
484 static HRESULT WINAPI Widget_VarArg(
485     IWidget * iface,
486     int numexpect,
487     SAFEARRAY * values)
488 {
489     LONG lbound, ubound, i;
490     VARIANT * data;
491     HRESULT hr;
492
493     trace("VarArg(%p)\n", values);
494
495     hr = SafeArrayGetLBound(values, 1, &lbound);
496     ok(hr == S_OK, "SafeArrayGetLBound failed with %x\n", hr);
497     ok(lbound == 0, "SafeArrayGetLBound returned %d\n", lbound);
498
499     hr = SafeArrayGetUBound(values, 1, &ubound);
500     ok(hr == S_OK, "SafeArrayGetUBound failed with %x\n", hr);
501     ok(ubound == numexpect-1, "SafeArrayGetUBound returned %d, but expected %d\n", ubound, numexpect-1);
502
503     hr = SafeArrayAccessData(values, (LPVOID)&data);
504     ok(hr == S_OK, "SafeArrayAccessData failed with %x\n", hr);
505
506     for (i=0; i<=ubound-lbound; i++)
507     {
508         ok(V_VT(&data[i]) == VT_I4, "V_VT(&data[%d]) was %d\n", i, V_VT(&data[i]));
509         ok(V_I4(&data[i]) == i, "V_I4(&data[%d]) was %d\n", i, V_I4(&data[i]));
510     }
511
512     hr = SafeArrayUnaccessData(values);
513     ok(hr == S_OK, "SafeArrayUnaccessData failed with %x\n", hr);
514
515     return S_OK;
516 }
517
518 static HRESULT WINAPI Widget_StructArgs(
519     IWidget * iface,
520     MYSTRUCT byval,
521     MYSTRUCT *byptr,
522     MYSTRUCT arr[5])
523 {
524     ok(memcmp(&byval, &MYSTRUCT_BYVAL, sizeof(MYSTRUCT))==0, "Struct parameter passed by value corrupted\n");
525     ok(memcmp(byptr,  &MYSTRUCT_BYPTR, sizeof(MYSTRUCT))==0, "Struct parameter passed by pointer corrupted\n");
526     ok(memcmp(arr,    MYSTRUCT_ARRAY,  sizeof(MYSTRUCT_ARRAY))==0, "Array of structs corrupted\n");
527     return S_OK;
528 }
529
530
531 static HRESULT WINAPI Widget_Error(
532     IWidget __RPC_FAR * iface)
533 {
534     trace("Error()\n");
535     return E_NOTIMPL;
536 }
537
538 static HRESULT WINAPI Widget_CloneInterface(
539     IWidget __RPC_FAR * iface,
540     ISomethingFromDispatch **ppVal)
541 {
542     trace("CloneInterface()\n");
543     *ppVal = 0;
544     return S_OK;
545 }
546
547 static HRESULT WINAPI Widget_put_prop_with_lcid(
548     IWidget* iface, LONG lcid, INT i)
549 {
550     trace("put_prop_with_lcid(%08x, %x)\n", lcid, i);
551     ok(lcid == MAKELANGID(LANG_ENGLISH, SUBLANG_ENGLISH_US), "got lcid %08x\n", lcid);
552     ok(i == 0xcafe, "got %08x\n", i);
553     return S_OK;
554 }
555
556 static HRESULT WINAPI Widget_get_prop_with_lcid(
557     IWidget* iface, LONG lcid, INT *i)
558 {
559     trace("get_prop_with_lcid(%08x, %p)\n", lcid, i);
560     ok(lcid == MAKELANGID(LANG_ENGLISH, SUBLANG_ENGLISH_US), "got lcid %08x\n", lcid);
561     *i = lcid;
562     return S_OK;
563 }
564
565 static HRESULT WINAPI Widget_get_prop_int(
566     IWidget* iface, INT *i)
567 {
568     trace("get_prop_int(%p)\n", i);
569     *i = -13;
570     return S_OK;
571 }
572
573 static HRESULT WINAPI Widget_get_prop_uint(
574     IWidget* iface, UINT *i)
575 {
576     trace("get_prop_uint(%p)\n", i);
577     *i = 42;
578     return S_OK;
579 }
580
581 static const struct IWidgetVtbl Widget_VTable =
582 {
583     Widget_QueryInterface,
584     Widget_AddRef,
585     Widget_Release,
586     Widget_GetTypeInfoCount,
587     Widget_GetTypeInfo,
588     Widget_GetIDsOfNames,
589     Widget_Invoke,
590     Widget_put_Name,
591     Widget_get_Name,
592     Widget_DoSomething,
593     Widget_get_State,
594     Widget_put_State,
595     Widget_Map,
596     Widget_SetOleColor,
597     Widget_GetOleColor,
598     Widget_Clone,
599     Widget_CloneDispatch,
600     Widget_CloneCoclass,
601     Widget_Value,
602     Widget_Array,
603     Widget_VariantArrayPtr,
604     Widget_Variant,
605     Widget_VarArg,
606     Widget_StructArgs,
607     Widget_Error,
608     Widget_CloneInterface,
609     Widget_put_prop_with_lcid,
610     Widget_get_prop_with_lcid,
611     Widget_get_prop_int,
612     Widget_get_prop_uint
613 };
614
615 static HRESULT WINAPI StaticWidget_QueryInterface(IStaticWidget *iface, REFIID riid, void **ppvObject)
616 {
617     if (IsEqualIID(riid, &IID_IStaticWidget) || IsEqualIID(riid, &IID_IUnknown) || IsEqualIID(riid, &IID_IDispatch))
618     {
619         IStaticWidget_AddRef(iface);
620         *ppvObject = iface;
621         return S_OK;
622     }
623
624     *ppvObject = NULL;
625     return E_NOINTERFACE;
626 }
627
628 static ULONG WINAPI StaticWidget_AddRef(IStaticWidget *iface)
629 {
630     return 2;
631 }
632
633 static ULONG WINAPI StaticWidget_Release(IStaticWidget *iface)
634 {
635     return 1;
636 }
637
638 static HRESULT WINAPI StaticWidget_GetTypeInfoCount(IStaticWidget *iface, UINT *pctinfo)
639 {
640     ok(0, "unexpected call\n");
641     return E_NOTIMPL;
642 }
643
644 static HRESULT WINAPI StaticWidget_GetTypeInfo(IStaticWidget *iface, UINT iTInfo, LCID lcid,
645         ITypeInfo **ppTInfo)
646 {
647     ok(0, "unexpected call\n");
648     return E_NOTIMPL;
649 }
650
651 static HRESULT WINAPI StaticWidget_GetIDsOfNames(IStaticWidget *iface, REFIID riid, LPOLESTR *rgszNames,
652         UINT cNames, LCID lcid, DISPID *rgDispId)
653 {
654     ok(0, "unexpected call\n");
655     return E_NOTIMPL;
656 }
657
658 static HRESULT WINAPI StaticWidget_Invoke(IStaticWidget *iface, DISPID dispIdMember, REFIID riid,
659         LCID lcid, WORD wFlags, DISPPARAMS *pDispParams, VARIANT *pVarResult, EXCEPINFO *pExcepInfo,
660          UINT *puArgErr)
661 {
662     ok(0, "unexpected call\n");
663     return E_NOTIMPL;
664 }
665
666 static HRESULT WINAPI StaticWidget_TestDual(IStaticWidget *iface, ItestDual *p)
667 {
668     trace("TestDual()\n");
669     todo_wine
670     ok(p == &TestDual, "wrong ItestDual\n");
671     return S_OK;
672 }
673
674 static const IStaticWidgetVtbl StaticWidgetVtbl = {
675     StaticWidget_QueryInterface,
676     StaticWidget_AddRef,
677     StaticWidget_Release,
678     StaticWidget_GetTypeInfoCount,
679     StaticWidget_GetTypeInfo,
680     StaticWidget_GetIDsOfNames,
681     StaticWidget_Invoke,
682     StaticWidget_TestDual
683 };
684
685 static IStaticWidget StaticWidget = { &StaticWidgetVtbl };
686
687 typedef struct KindaEnum
688 {
689     const IKindaEnumWidgetVtbl *lpVtbl;
690     LONG refs;
691 } KindaEnum;
692
693 static HRESULT register_current_module_typelib(void)
694 {
695     WCHAR path[MAX_PATH];
696     CHAR pathA[MAX_PATH];
697     HRESULT hr;
698     ITypeLib *typelib;
699
700     GetModuleFileNameA(NULL, pathA, MAX_PATH);
701     MultiByteToWideChar(CP_ACP, 0, pathA, -1, path, MAX_PATH);
702
703     hr = LoadTypeLib(path, &typelib);
704     if (SUCCEEDED(hr))
705     {
706         hr = RegisterTypeLib(typelib, path, NULL);
707         ITypeLib_Release(typelib);
708     }
709     return hr;
710 }
711
712 static ITypeInfo *get_type_info(REFIID riid)
713 {
714     ITypeInfo *pTypeInfo;
715     ITypeLib *pTypeLib;
716     HRESULT hr;
717
718     hr = LoadRegTypeLib(&LIBID_TestTypelib, 1, 0, LOCALE_NEUTRAL, &pTypeLib);
719     ok_ole_success(hr, LoadRegTypeLib);
720     if (FAILED(hr))
721         return NULL;
722
723     hr = ITypeLib_GetTypeInfoOfGuid(pTypeLib, riid, &pTypeInfo);
724     ITypeLib_Release(pTypeLib);
725     ok_ole_success(hr, ITypeLib_GetTypeInfoOfGuid);
726     if (FAILED(hr))
727         return NULL;
728
729     return pTypeInfo;
730 }
731
732 static IWidget *Widget_Create(void)
733 {
734     Widget *This;
735     ITypeInfo *pTypeInfo;
736     HRESULT hr = E_FAIL;
737
738     pTypeInfo = get_type_info(&IID_IWidget);
739     if(!pTypeInfo)
740         return NULL;
741
742     This = HeapAlloc(GetProcessHeap(), 0, sizeof(*This));
743     This->lpVtbl = &Widget_VTable;
744     This->refs = 1;
745     This->pDispatchUnknown = NULL;
746
747     hr = CreateStdDispatch((IUnknown *)&This->lpVtbl, This, pTypeInfo, &This->pDispatchUnknown);
748     ok_ole_success(hr, CreateStdDispatch);
749     ITypeInfo_Release(pTypeInfo);
750
751     if (SUCCEEDED(hr))
752         return (IWidget *)&This->lpVtbl;
753     else
754     {
755         HeapFree(GetProcessHeap(), 0, This);
756         return NULL;
757     }
758 }
759
760 static HRESULT WINAPI KindaEnum_QueryInterface(
761     IKindaEnumWidget *iface,
762     /* [in] */ REFIID riid,
763     /* [iid_is][out] */ void __RPC_FAR *__RPC_FAR *ppvObject)
764 {
765     if (IsEqualIID(riid, &IID_IKindaEnumWidget) || IsEqualIID(riid, &IID_IUnknown))
766     {
767         IKindaEnumWidget_AddRef(iface);
768         *ppvObject = iface;
769         return S_OK;
770     }
771     else
772     {
773         *ppvObject = NULL;
774         return E_NOINTERFACE;
775     }
776 }
777
778 static ULONG WINAPI KindaEnum_AddRef(
779     IKindaEnumWidget *iface)
780 {
781     KindaEnum *This = (KindaEnum *)iface;
782
783     return InterlockedIncrement(&This->refs);
784 }
785
786 static ULONG WINAPI KindaEnum_Release(
787     IKindaEnumWidget *iface)
788 {
789     KindaEnum *This = (KindaEnum *)iface;
790     ULONG refs = InterlockedDecrement(&This->refs);
791     if (!refs)
792     {
793         memset(This, 0xcc, sizeof(*This));
794         HeapFree(GetProcessHeap(), 0, This);
795         trace("KindaEnumWidget destroyed!\n");
796     }
797
798     return refs;
799 }
800
801 static HRESULT WINAPI KindaEnum_Next(
802     IKindaEnumWidget *iface,
803     /* [out] */ IWidget __RPC_FAR *__RPC_FAR *widget)
804 {
805     *widget = Widget_Create();
806     if (*widget)
807         return S_OK;
808     else
809         return E_OUTOFMEMORY;
810 }
811
812 static HRESULT WINAPI KindaEnum_Count(
813     IKindaEnumWidget *iface,
814     /* [out] */ ULONG __RPC_FAR *count)
815 {
816     return E_NOTIMPL;
817 }
818
819 static HRESULT WINAPI KindaEnum_Reset(
820     IKindaEnumWidget *iface)
821 {
822     return E_NOTIMPL;
823 }
824
825 static HRESULT WINAPI KindaEnum_Clone(
826     IKindaEnumWidget *iface,
827     /* [out] */ IKindaEnumWidget __RPC_FAR *__RPC_FAR *ppenum)
828 {
829     return E_NOTIMPL;
830 }
831
832 static const IKindaEnumWidgetVtbl KindaEnumWidget_VTable =
833 {
834     KindaEnum_QueryInterface,
835     KindaEnum_AddRef,
836     KindaEnum_Release,
837     KindaEnum_Next,
838     KindaEnum_Count,
839     KindaEnum_Reset,
840     KindaEnum_Clone
841 };
842
843 static IKindaEnumWidget *KindaEnumWidget_Create(void)
844 {
845     KindaEnum *This;
846
847     This = HeapAlloc(GetProcessHeap(), 0, sizeof(*This));
848     if (!This) return NULL;
849     This->lpVtbl = &KindaEnumWidget_VTable;
850     This->refs = 1;
851     return (IKindaEnumWidget *)This;
852 }
853
854 static HRESULT WINAPI NonOleAutomation_QueryInterface(INonOleAutomation *iface, REFIID riid, void **ppv)
855 {
856     if (IsEqualIID(riid, &IID_IUnknown) || IsEqualIID(riid, &IID_INonOleAutomation))
857     {
858         *(INonOleAutomation **)ppv = iface;
859         return S_OK;
860     }
861     *ppv = NULL;
862     return E_NOINTERFACE;
863 }
864
865 static ULONG WINAPI NonOleAutomation_AddRef(INonOleAutomation *iface)
866 {
867     return 2;
868 }
869
870 static ULONG WINAPI NonOleAutomation_Release(INonOleAutomation *iface)
871 {
872     return 1;
873 }
874
875 static BSTR WINAPI NonOleAutomation_BstrRet(INonOleAutomation *iface)
876 {
877     static const WCHAR wszTestString[] = {'T','h','i','s',' ','i','s',' ','a',' ','t','e','s','t',' ','s','t','r','i','n','g',0};
878     return SysAllocString(wszTestString);
879 }
880
881 static INonOleAutomationVtbl NonOleAutomation_VTable =
882 {
883     NonOleAutomation_QueryInterface,
884     NonOleAutomation_AddRef,
885     NonOleAutomation_Release,
886     NonOleAutomation_BstrRet,
887 };
888
889 static INonOleAutomation NonOleAutomation = { &NonOleAutomation_VTable };
890
891 static ITypeInfo *NonOleAutomation_GetTypeInfo(void)
892 {
893     ITypeLib *pTypeLib;
894     HRESULT hr = LoadRegTypeLib(&LIBID_TestTypelib, 1, 0, LOCALE_NEUTRAL, &pTypeLib);
895     ok_ole_success(hr, LoadRegTypeLib);
896     if (SUCCEEDED(hr))
897     {
898         ITypeInfo *pTypeInfo;
899         hr = ITypeLib_GetTypeInfoOfGuid(pTypeLib, &IID_INonOleAutomation, &pTypeInfo);
900         ok_ole_success(hr, ITypeLib_GetTypeInfoOfGuid);
901         return pTypeInfo;
902     }
903     return NULL;
904 }
905
906 static void test_typelibmarshal(void)
907 {
908     static const WCHAR szCat[] = { 'C','a','t',0 };
909     static const WCHAR szTestTest[] = { 'T','e','s','t','T','e','s','t',0 };
910     static WCHAR szSuperman[] = { 'S','u','p','e','r','m','a','n',0 };
911     HRESULT hr;
912     IKindaEnumWidget *pKEW = KindaEnumWidget_Create();
913     IWidget *pWidget;
914     IStream *pStream;
915     IDispatch *pDispatch;
916     static const LARGE_INTEGER ullZero;
917     EXCEPINFO excepinfo;
918     VARIANT varresult;
919     DISPID dispidNamed = DISPID_PROPERTYPUT;
920     DISPPARAMS dispparams;
921     VARIANTARG vararg[4];
922     STATE the_state;
923     HANDLE thread;
924     DWORD tid;
925     BSTR bstr;
926     ITypeInfo *pTypeInfo;
927     MYSTRUCT mystruct;
928     MYSTRUCT mystructArray[5];
929
930     ok(pKEW != NULL, "Widget creation failed\n");
931
932     hr = CreateStreamOnHGlobal(NULL, TRUE, &pStream);
933     ok_ole_success(hr, CreateStreamOnHGlobal);
934     tid = start_host_object(pStream, &IID_IKindaEnumWidget, (IUnknown *)pKEW, MSHLFLAGS_NORMAL, &thread);
935     IKindaEnumWidget_Release(pKEW);
936
937     IStream_Seek(pStream, ullZero, STREAM_SEEK_SET, NULL);
938     hr = CoUnmarshalInterface(pStream, &IID_IKindaEnumWidget, (void **)&pKEW);
939     ok_ole_success(hr, CoUnmarshalInterface);
940     IStream_Release(pStream);
941
942     hr = IKindaEnumWidget_Next(pKEW, &pWidget);
943     ok_ole_success(hr, IKindaEnumWidget_Next);
944
945     IKindaEnumWidget_Release(pKEW);
946
947     hr = IWidget_QueryInterface(pWidget, &IID_IDispatch, (void **)&pDispatch);
948
949     /* call put_Name */
950     VariantInit(&vararg[0]);
951     dispparams.cNamedArgs = 1;
952     dispparams.rgdispidNamedArgs = &dispidNamed;
953     dispparams.cArgs = 1;
954     dispparams.rgvarg = vararg;
955     VariantInit(&varresult);
956     hr = IDispatch_Invoke(pDispatch, DISPID_TM_NAME, &IID_NULL, LOCALE_NEUTRAL, DISPATCH_PROPERTYPUT, &dispparams, &varresult, &excepinfo, NULL);
957     ok_ole_success(hr, IDispatch_Invoke);
958     ok(excepinfo.wCode == 0x0 && excepinfo.scode == S_OK,
959         "EXCEPINFO differs from expected: wCode = 0x%x, scode = 0x%08x\n",
960         excepinfo.wCode, excepinfo.scode);
961     VariantClear(&varresult);
962
963     /* call put_Name (direct) */
964     bstr = SysAllocString(szSuperman);
965     hr = IWidget_put_Name(pWidget, bstr);
966     ok_ole_success(hr, IWidget_put_Name);
967     SysFreeString(bstr);
968
969     /* call get_Name */
970     dispparams.cNamedArgs = 0;
971     dispparams.rgdispidNamedArgs = NULL;
972     dispparams.cArgs = 0;
973     dispparams.rgvarg = NULL;
974     VariantInit(&varresult);
975     hr = IDispatch_Invoke(pDispatch, DISPID_TM_NAME, &IID_NULL, LOCALE_NEUTRAL, DISPATCH_PROPERTYGET, &dispparams, &varresult, &excepinfo, NULL);
976     ok_ole_success(hr, IDispatch_Invoke);
977     ok(excepinfo.wCode == 0x0 && excepinfo.scode == S_OK,
978         "EXCEPINFO differs from expected: wCode = 0x%x, scode = 0x%08x\n",
979         excepinfo.wCode, excepinfo.scode);
980     trace("Name = %s\n", wine_dbgstr_w(V_BSTR(&varresult)));
981     VariantClear(&varresult);
982
983     /* call get_Name (direct) */
984     bstr = NULL;
985     hr = IWidget_get_Name(pWidget, &bstr);
986     ok_ole_success(hr, IWidget_get_Name);
987     ok(!lstrcmpW(bstr, szCat), "IWidget_get_Name should have returned string \"Cat\" instead of %s\n", wine_dbgstr_w(bstr));
988     SysFreeString(bstr);
989
990     /* call DoSomething */
991     VariantInit(&vararg[0]);
992     VariantInit(&vararg[1]);
993     V_VT(&vararg[1]) = VT_R8;
994     V_R8(&vararg[1]) = 3.141;
995     dispparams.cNamedArgs = 0;
996     dispparams.cArgs = 2;
997     dispparams.rgdispidNamedArgs = NULL;
998     dispparams.rgvarg = vararg;
999     VariantInit(&varresult);
1000     hr = IDispatch_Invoke(pDispatch, DISPID_TM_DOSOMETHING, &IID_NULL, LOCALE_NEUTRAL, DISPATCH_METHOD, &dispparams, &varresult, &excepinfo, NULL);
1001     ok_ole_success(hr, IDispatch_Invoke);
1002     ok(V_VT(&varresult) == VT_EMPTY, "varresult should be VT_EMPTY\n");
1003     VariantClear(&varresult);
1004
1005     /* call get_State */
1006     dispparams.cNamedArgs = 0;
1007     dispparams.cArgs = 0;
1008     dispparams.rgdispidNamedArgs = NULL;
1009     dispparams.rgvarg = NULL;
1010     hr = IDispatch_Invoke(pDispatch, DISPID_TM_STATE, &IID_NULL, LOCALE_NEUTRAL, DISPATCH_PROPERTYGET, &dispparams, &varresult, &excepinfo, NULL);
1011     ok_ole_success(hr, IDispatch_Invoke);
1012     ok((V_VT(&varresult) == VT_I4) && (V_I4(&varresult) == STATE_WIDGETIFIED), "Return val mismatch\n");
1013
1014     /* call get_State (direct) */
1015     hr = IWidget_get_State(pWidget, &the_state);
1016     ok_ole_success(hr, IWidget_get_state);
1017     ok(the_state == STATE_WIDGETIFIED, "should have returned WIDGET_WIDGETIFIED instead of %d\n", the_state);
1018
1019     /* call put_State */
1020     the_state = STATE_WIDGETIFIED;
1021     VariantInit(&vararg[0]);
1022     V_VT(&vararg[0]) = VT_BYREF|VT_I4;
1023     V_I4REF(&vararg[0]) = (int *)&the_state;
1024     dispparams.cNamedArgs = 1;
1025     dispparams.cArgs = 1;
1026     dispparams.rgdispidNamedArgs = &dispidNamed;
1027     dispparams.rgvarg = vararg;
1028     hr = IDispatch_Invoke(pDispatch, DISPID_TM_STATE, &IID_NULL, LOCALE_NEUTRAL, DISPATCH_PROPERTYPUT, &dispparams, &varresult, &excepinfo, NULL);
1029     ok_ole_success(hr, IDispatch_Invoke);
1030
1031     /* call Map */
1032     bstr = SysAllocString(szTestTest);
1033     VariantInit(&vararg[0]);
1034     V_VT(&vararg[0]) = VT_BYREF|VT_BSTR;
1035     V_BSTRREF(&vararg[0]) = &bstr;
1036     dispparams.cNamedArgs = 0;
1037     dispparams.cArgs = 1;
1038     dispparams.rgdispidNamedArgs = NULL;
1039     dispparams.rgvarg = vararg;
1040     VariantInit(&varresult);
1041     hr = IDispatch_Invoke(pDispatch, DISPID_TM_MAP, &IID_NULL, LOCALE_NEUTRAL, DISPATCH_METHOD, &dispparams, &varresult, &excepinfo, NULL);
1042     ok_ole_success(hr, IDispatch_Invoke);
1043     ok(V_VT(&varresult) == VT_BSTR, "Return value should be of type BSTR instead of %d\n", V_VT(&varresult));
1044     ok(!lstrcmpW(V_BSTR(&varresult), szTestTest), "Return value should have been \"TestTest\" instead of %s\n", wine_dbgstr_w(V_BSTR(&varresult)));
1045     VariantClear(&varresult);
1046
1047     /* call SetOleColor with large negative VT_I4 param */
1048     VariantInit(&vararg[0]);
1049     V_VT(&vararg[0]) = VT_I4;
1050     V_I4(&vararg[0]) = 0x80000005;
1051     dispparams.cNamedArgs = 0;
1052     dispparams.cArgs = 1;
1053     dispparams.rgdispidNamedArgs = NULL;
1054     dispparams.rgvarg = vararg;
1055     hr = IDispatch_Invoke(pDispatch, DISPID_TM_SETOLECOLOR, &IID_NULL, LOCALE_NEUTRAL, DISPATCH_METHOD, &dispparams, NULL, &excepinfo, NULL);
1056     ok_ole_success(hr, IDispatch_Invoke);
1057
1058     /* call GetOleColor */
1059     dispparams.cNamedArgs = 0;
1060     dispparams.cArgs = 0;
1061     dispparams.rgdispidNamedArgs = NULL;
1062     dispparams.rgvarg = NULL;
1063     VariantInit(&varresult);
1064     hr = IDispatch_Invoke(pDispatch, DISPID_TM_GETOLECOLOR, &IID_NULL, LOCALE_NEUTRAL, DISPATCH_METHOD, &dispparams, &varresult, &excepinfo, NULL);
1065     ok_ole_success(hr, IDispatch_Invoke);
1066     VariantClear(&varresult);
1067
1068     /* call StructArgs (direct) */
1069     mystruct = MYSTRUCT_BYPTR;
1070     memcpy(mystructArray, MYSTRUCT_ARRAY, sizeof(mystructArray));
1071     hr = IWidget_StructArgs(pWidget, MYSTRUCT_BYVAL, &mystruct, mystructArray);
1072     todo_wine {
1073     ok_ole_success(hr, IWidget_StructArgs);
1074     }
1075
1076     /* call Clone */
1077     dispparams.cNamedArgs = 0;
1078     dispparams.cArgs = 0;
1079     dispparams.rgdispidNamedArgs = NULL;
1080     dispparams.rgvarg = NULL;
1081     VariantInit(&varresult);
1082     hr = IDispatch_Invoke(pDispatch, DISPID_TM_CLONE, &IID_NULL, LOCALE_NEUTRAL, DISPATCH_PROPERTYGET, &dispparams, &varresult, &excepinfo, NULL);
1083     ok_ole_success(hr, IDispatch_Invoke);
1084     ok(V_VT(&varresult) == VT_DISPATCH, "vt %x\n", V_VT(&varresult));
1085     VariantClear(&varresult);
1086
1087     /* call CloneInterface */
1088     dispparams.cNamedArgs = 0;
1089     dispparams.cArgs = 0;
1090     dispparams.rgdispidNamedArgs = NULL;
1091     dispparams.rgvarg = NULL;
1092     VariantInit(&varresult);
1093     hr = IDispatch_Invoke(pDispatch, DISPID_TM_CLONEINTERFACE, &IID_NULL, LOCALE_NEUTRAL, DISPATCH_PROPERTYGET, &dispparams, &varresult, &excepinfo, NULL);
1094     ok_ole_success(hr, IDispatch_Invoke);
1095     ok(V_VT(&varresult) == VT_DISPATCH, "vt %x\n", V_VT(&varresult));
1096     VariantClear(&varresult);
1097
1098     /* call CloneDispatch with automatic value getting */
1099     V_VT(&vararg[0]) = VT_I2;
1100     V_I2(&vararg[0]) = 1;
1101     dispparams.cNamedArgs = 0;
1102     dispparams.rgdispidNamedArgs = NULL;
1103     dispparams.cArgs = 1;
1104     dispparams.rgvarg = vararg;
1105     VariantInit(&varresult);
1106     hr = IDispatch_Invoke(pDispatch, DISPID_TM_CLONEDISPATCH, &IID_NULL, LOCALE_NEUTRAL, DISPATCH_PROPERTYGET, &dispparams, &varresult, &excepinfo, NULL);
1107     ok_ole_success(hr, IDispatch_Invoke);
1108
1109     ok(excepinfo.wCode == 0x0 && excepinfo.scode == S_OK,
1110         "EXCEPINFO differs from expected: wCode = 0x%x, scode = 0x%08x\n",
1111         excepinfo.wCode, excepinfo.scode);
1112
1113     ok(V_VT(&varresult) == VT_I2, "V_VT(&varresult) was %d instead of VT_I2\n", V_VT(&varresult));
1114     ok(V_I2(&varresult) == 1234, "V_I2(&varresult) was %d instead of 1234\n", V_I2(&varresult));
1115     VariantClear(&varresult);
1116
1117     /* call CloneCoclass */
1118     dispparams.cNamedArgs = 0;
1119     dispparams.cArgs = 0;
1120     dispparams.rgdispidNamedArgs = NULL;
1121     dispparams.rgvarg = NULL;
1122     VariantInit(&varresult);
1123     hr = IDispatch_Invoke(pDispatch, DISPID_TM_CLONECOCLASS, &IID_NULL, LOCALE_NEUTRAL, DISPATCH_PROPERTYGET, &dispparams, &varresult, &excepinfo, NULL);
1124     ok_ole_success(hr, IDispatch_Invoke);
1125
1126     ok(excepinfo.wCode == 0x0 && excepinfo.scode == S_OK,
1127        "EXCEPINFO differs from expected: wCode = 0x%x, scode = 0x%08x\n",
1128        excepinfo.wCode, excepinfo.scode);
1129
1130     ok(V_VT(&varresult) == VT_DISPATCH, "V_VT(&varresult) was %d instead of VT_DISPATCH\n", V_VT(&varresult));
1131     ok(!V_DISPATCH(&varresult), "V_DISPATCH(&varresult) should be NULL instead of %p\n", V_DISPATCH(&varresult));
1132     VariantClear(&varresult);
1133
1134     /* call Value with a VT_VARIANT|VT_BYREF type */
1135     V_VT(&vararg[0]) = VT_VARIANT|VT_BYREF;
1136     V_VARIANTREF(&vararg[0]) = &vararg[1];
1137     V_VT(&vararg[1]) = VT_I2;
1138     V_I2(&vararg[1]) = 1;
1139     dispparams.cNamedArgs = 0;
1140     dispparams.rgdispidNamedArgs = NULL;
1141     dispparams.cArgs = 1;
1142     dispparams.rgvarg = vararg;
1143     VariantInit(&varresult);
1144     hr = IDispatch_Invoke(pDispatch, DISPID_VALUE, &IID_NULL, LOCALE_NEUTRAL, DISPATCH_PROPERTYGET, &dispparams, &varresult, &excepinfo, NULL);
1145     ok_ole_success(hr, IDispatch_Invoke);
1146
1147     ok(excepinfo.wCode == 0x0 && excepinfo.scode == S_OK,
1148         "EXCEPINFO differs from expected: wCode = 0x%x, scode = 0x%08x\n",
1149         excepinfo.wCode, excepinfo.scode);
1150
1151     ok(V_VT(&varresult) == VT_I2, "V_VT(&varresult) was %d instead of VT_I2\n", V_VT(&varresult));
1152     ok(V_I2(&varresult) == 1234, "V_I2(&varresult) was %d instead of 1234\n", V_I2(&varresult));
1153     VariantClear(&varresult);
1154
1155     /* call Variant - exercises variant copying in ITypeInfo::Invoke and
1156      * handling of void return types */
1157     /* use a big type to ensure that the variant was properly copied into the
1158      * destination function's args */
1159     V_VT(&vararg[0]) = VT_CY;
1160     S(V_CY(&vararg[0])).Hi = 0xdababe;
1161     S(V_CY(&vararg[0])).Lo = 0xdeadbeef;
1162     dispparams.cNamedArgs = 0;
1163     dispparams.cArgs = 1;
1164     dispparams.rgdispidNamedArgs = NULL;
1165     dispparams.rgvarg = vararg;
1166     VariantInit(&varresult);
1167     hr = IDispatch_Invoke(pDispatch, DISPID_TM_VARIANT, &IID_NULL, LOCALE_NEUTRAL, DISPATCH_METHOD, &dispparams, NULL, NULL, NULL);
1168     ok_ole_success(hr, IDispatch_Invoke);
1169     VariantClear(&varresult);
1170
1171     /* call VarArg */
1172     VariantInit(&vararg[3]);
1173     V_VT(&vararg[3]) = VT_I4;
1174     V_I4(&vararg[3]) = 3;
1175     VariantInit(&vararg[2]);
1176     V_VT(&vararg[2]) = VT_I4;
1177     V_I4(&vararg[2]) = 0;
1178     VariantInit(&vararg[1]);
1179     V_VT(&vararg[1]) = VT_I4;
1180     V_I4(&vararg[1]) = 1;
1181     VariantInit(&vararg[0]);
1182     V_VT(&vararg[0]) = VT_I4;
1183     V_I4(&vararg[0]) = 2;
1184     dispparams.cNamedArgs = 0;
1185     dispparams.cArgs = 4;
1186     dispparams.rgdispidNamedArgs = NULL;
1187     dispparams.rgvarg = vararg;
1188     hr = IDispatch_Invoke(pDispatch, DISPID_TM_VARARG, &IID_NULL, LOCALE_NEUTRAL, DISPATCH_METHOD, &dispparams, NULL, NULL, NULL);
1189     ok_ole_success(hr, ITypeInfo_Invoke);
1190
1191     /* call VarArg, even one (non-optional, non-safearray) named argument is not allowed */
1192     dispidNamed = 0;
1193     dispparams.cNamedArgs = 1;
1194     dispparams.rgdispidNamedArgs = &dispidNamed;
1195     hr = IDispatch_Invoke(pDispatch, DISPID_TM_VARARG, &IID_NULL, LOCALE_NEUTRAL, DISPATCH_METHOD, &dispparams, NULL, NULL, NULL);
1196     ok(hr == DISP_E_NONAMEDARGS, "IDispatch_Invoke should have returned DISP_E_NONAMEDARGS instead of 0x%08x\n", hr);
1197     dispidNamed = DISPID_PROPERTYPUT;
1198
1199     /* call Error */
1200     dispparams.cNamedArgs = 0;
1201     dispparams.cArgs = 0;
1202     dispparams.rgdispidNamedArgs = NULL;
1203     dispparams.rgvarg = NULL;
1204     VariantInit(&varresult);
1205     hr = IDispatch_Invoke(pDispatch, DISPID_TM_ERROR, &IID_NULL, LOCALE_NEUTRAL, DISPATCH_METHOD, &dispparams, NULL, &excepinfo, NULL);
1206     ok(hr == DISP_E_EXCEPTION, "IDispatch_Invoke should have returned DISP_E_EXCEPTION instead of 0x%08x\n", hr);
1207     ok(excepinfo.wCode == 0x0 && excepinfo.scode == E_NOTIMPL,
1208         "EXCEPINFO differs from expected: wCode = 0x%x, scode = 0x%08x\n",
1209         excepinfo.wCode, excepinfo.scode);
1210     VariantClear(&varresult);
1211
1212     /* call BstrRet */
1213     pTypeInfo = NonOleAutomation_GetTypeInfo();
1214     dispparams.cNamedArgs = 0;
1215     dispparams.cArgs = 0;
1216     dispparams.rgdispidNamedArgs = NULL;
1217     dispparams.rgvarg = NULL;
1218     VariantInit(&varresult);
1219     hr = ITypeInfo_Invoke(pTypeInfo, &NonOleAutomation, DISPID_NOA_BSTRRET, DISPATCH_METHOD, &dispparams, &varresult, &excepinfo, NULL);
1220     ok_ole_success(hr, ITypeInfo_Invoke);
1221     ok(V_VT(&varresult) == VT_BSTR, "V_VT(&varresult) should be VT_BSTR instead of %d\n", V_VT(&varresult));
1222     ok(V_BSTR(&varresult) != NULL, "V_BSTR(&varresult) should not be NULL\n");
1223
1224     VariantClear(&varresult);
1225     ITypeInfo_Release(pTypeInfo);
1226
1227     /* tests call put_Name without named arg */
1228     VariantInit(&vararg[0]);
1229     dispparams.cNamedArgs = 0;
1230     dispparams.rgdispidNamedArgs = NULL;
1231     dispparams.cArgs = 1;
1232     dispparams.rgvarg = vararg;
1233     VariantInit(&varresult);
1234     hr = IDispatch_Invoke(pDispatch, DISPID_TM_NAME, &IID_NULL, LOCALE_NEUTRAL, DISPATCH_PROPERTYPUT, &dispparams, &varresult, &excepinfo, NULL);
1235     ok(hr == DISP_E_PARAMNOTFOUND, "IDispatch_Invoke should have returned DISP_E_PARAMNOTFOUND instead of 0x%08x\n", hr);
1236     VariantClear(&varresult);
1237
1238     /* tests param type that cannot be coerced */
1239     VariantInit(&vararg[0]);
1240     V_VT(&vararg[0]) = VT_UNKNOWN;
1241     V_UNKNOWN(&vararg[0]) = NULL;
1242     dispparams.cNamedArgs = 1;
1243     dispparams.rgdispidNamedArgs = &dispidNamed;
1244     dispparams.cArgs = 1;
1245     dispparams.rgvarg = vararg;
1246     VariantInit(&varresult);
1247 #if 0 /* NULL unknown not currently marshaled correctly */
1248     hr = IDispatch_Invoke(pDispatch, DISPID_TM_NAME, &IID_NULL, LOCALE_NEUTRAL, DISPATCH_PROPERTYPUT, &dispparams, &varresult, &excepinfo, NULL);
1249     ok(hr == DISP_E_TYPEMISMATCH, "IDispatch_Invoke should have returned DISP_E_TYPEMISMATCH instead of 0x%08x\n", hr);
1250 #endif
1251     VariantClear(&varresult);
1252
1253     /* tests bad param type */
1254     VariantInit(&vararg[0]);
1255     V_VT(&vararg[0]) = VT_CLSID;
1256     V_BYREF(&vararg[0]) = NULL;
1257     dispparams.cNamedArgs = 1;
1258     dispparams.rgdispidNamedArgs = &dispidNamed;
1259     dispparams.cArgs = 1;
1260     dispparams.rgvarg = vararg;
1261     VariantInit(&varresult);
1262     hr = IDispatch_Invoke(pDispatch, DISPID_TM_NAME, &IID_NULL, LOCALE_NEUTRAL, DISPATCH_PROPERTYPUT, &dispparams, &varresult, &excepinfo, NULL);
1263     ok(hr == DISP_E_BADVARTYPE, "IDispatch_Invoke should have returned DISP_E_BADVARTYPE instead of 0x%08x\n", hr);
1264     VariantClear(&varresult);
1265
1266     /* tests too small param count */
1267     dispparams.cNamedArgs = 0;
1268     dispparams.rgdispidNamedArgs = NULL;
1269     dispparams.cArgs = 0;
1270     dispparams.rgvarg = NULL;
1271     VariantInit(&varresult);
1272     hr = IDispatch_Invoke(pDispatch, DISPID_TM_DOSOMETHING, &IID_NULL, LOCALE_NEUTRAL, DISPATCH_METHOD, &dispparams, &varresult, &excepinfo, NULL);
1273     ok(hr == DISP_E_BADPARAMCOUNT, "IDispatch_Invoke should have returned DISP_E_BADPARAMCOUNT instead of 0x%08x\n", hr);
1274     VariantClear(&varresult);
1275
1276     /* tests propget function with large param count */
1277     VariantInit(&vararg[0]);
1278     V_VT(&vararg[0]) = VT_BSTR;
1279     V_BSTR(&vararg[0]) = NULL;
1280     V_VT(&vararg[1]) = VT_I4;
1281     V_I4(&vararg[1]) = 1;
1282     dispparams.cNamedArgs = 0;
1283     dispparams.cArgs = 2;
1284     dispparams.rgdispidNamedArgs = NULL;
1285     dispparams.rgvarg = vararg;
1286     hr = IDispatch_Invoke(pDispatch, DISPID_TM_STATE, &IID_NULL, LOCALE_NEUTRAL, DISPATCH_PROPERTYGET, &dispparams, &varresult, &excepinfo, NULL);
1287     ok(hr == DISP_E_NOTACOLLECTION, "IDispatch_Invoke should have returned DISP_E_NOTACOLLECTION instead of 0x%08x\n", hr);
1288
1289     /* test propput with lcid */
1290
1291     /* the lcid passed to the function is the first lcid in the typelib header.
1292        Since we don't explicitly set an lcid in the idl, it'll default to US English. */
1293     VariantInit(&vararg[0]);
1294     V_VT(&vararg[0]) = VT_I4;
1295     V_I4(&vararg[0]) = 0xcafe;
1296     dispparams.cNamedArgs = 1;
1297     dispparams.rgdispidNamedArgs = &dispidNamed;
1298     dispparams.cArgs = 1;
1299     dispparams.rgvarg = vararg;
1300     VariantInit(&varresult);
1301     hr = IDispatch_Invoke(pDispatch, DISPID_TM_PROP_WITH_LCID, &IID_NULL, 0x40c, DISPATCH_PROPERTYPUT, &dispparams, &varresult, &excepinfo, NULL);
1302     ok_ole_success(hr, ITypeInfo_Invoke);
1303     VariantClear(&varresult);
1304
1305     /* test propget with lcid */
1306     dispparams.cNamedArgs = 0;
1307     dispparams.cArgs = 0;
1308     dispparams.rgvarg = NULL;
1309     dispparams.rgdispidNamedArgs = NULL;
1310     hr = IDispatch_Invoke(pDispatch, DISPID_TM_PROP_WITH_LCID, &IID_NULL, 0x40c, DISPATCH_PROPERTYGET, &dispparams, &varresult, &excepinfo, NULL);
1311     ok_ole_success(hr, ITypeInfo_Invoke);
1312     ok(V_VT(&varresult) == VT_I4, "got %x\n", V_VT(&varresult));
1313     ok(V_I4(&varresult) == 0x409, "got %x\n", V_I4(&varresult));
1314     VariantClear(&varresult);
1315
1316     /* test propget of INT value */
1317     dispparams.cNamedArgs = 0;
1318     dispparams.cArgs = 0;
1319     dispparams.rgvarg = NULL;
1320     dispparams.rgdispidNamedArgs = NULL;
1321     hr = IDispatch_Invoke(pDispatch, DISPID_TM_PROP_INT, &IID_NULL, 0x40c, DISPATCH_PROPERTYGET, &dispparams, &varresult, &excepinfo, NULL);
1322     ok_ole_success(hr, ITypeInfo_Invoke);
1323     ok(V_VT(&varresult) == VT_I4, "got %x\n", V_VT(&varresult));
1324     ok(V_I4(&varresult) == -13, "got %x\n", V_I4(&varresult));
1325     VariantClear(&varresult);
1326
1327     /* test propget of INT value */
1328     dispparams.cNamedArgs = 0;
1329     dispparams.cArgs = 0;
1330     dispparams.rgvarg = NULL;
1331     dispparams.rgdispidNamedArgs = NULL;
1332     hr = IDispatch_Invoke(pDispatch, DISPID_TM_PROP_UINT, &IID_NULL, 0x40c, DISPATCH_PROPERTYGET, &dispparams, &varresult, &excepinfo, NULL);
1333     ok_ole_success(hr, ITypeInfo_Invoke);
1334     ok(V_VT(&varresult) == VT_UI4, "got %x\n", V_VT(&varresult));
1335     ok(V_UI4(&varresult) == 42, "got %x\n", V_UI4(&varresult));
1336     VariantClear(&varresult);
1337
1338     IDispatch_Release(pDispatch);
1339     IWidget_Release(pWidget);
1340
1341     trace("calling end_host_object\n");
1342     end_host_object(tid, thread);
1343 }
1344
1345 static void test_DispCallFunc(void)
1346 {
1347     static const WCHAR szEmpty[] = { 0 };
1348     VARTYPE rgvt[] = { VT_R8, VT_BSTR, VT_BSTR, VT_VARIANT|VT_BYREF };
1349     VARIANTARG vararg[4];
1350     VARIANTARG varref;
1351     VARIANTARG *rgpvarg[4] = { &vararg[0], &vararg[1], &vararg[2], &vararg[3] };
1352     VARIANTARG varresult;
1353     HRESULT hr;
1354     IWidget *pWidget = Widget_Create();
1355     V_VT(&vararg[0]) = VT_R8;
1356     V_R8(&vararg[0]) = 3.141;
1357     V_VT(&vararg[1]) = VT_BSTR;
1358     V_BSTR(&vararg[1]) = SysAllocString(szEmpty);
1359     V_VT(&vararg[2]) = VT_BSTR;
1360     V_BSTR(&vararg[2]) = SysAllocString(szEmpty);
1361     V_VT(&vararg[3]) = VT_VARIANT|VT_BYREF;
1362     V_VARIANTREF(&vararg[3]) = &varref;
1363     V_VT(&varref) = VT_ERROR;
1364     V_ERROR(&varref) = DISP_E_PARAMNOTFOUND;
1365     VariantInit(&varresult);
1366     hr = DispCallFunc(pWidget, 9*sizeof(void*), CC_STDCALL, VT_UI4, 4, rgvt, rgpvarg, &varresult);
1367     ok_ole_success(hr, DispCallFunc);
1368     VariantClear(&varresult);
1369     VariantClear(&vararg[1]);
1370     VariantClear(&vararg[2]);
1371     IWidget_Release(pWidget);
1372 }
1373
1374 static void test_StaticWidget(void)
1375 {
1376     ITypeInfo *type_info;
1377     DISPPARAMS dispparams;
1378     VARIANTARG vararg[4];
1379     EXCEPINFO excepinfo;
1380     VARIANT varresult;
1381     HRESULT hr;
1382
1383     type_info = get_type_info(&IID_IStaticWidget);
1384
1385     /* call TestDual */
1386     dispparams.cNamedArgs = 0;
1387     dispparams.cArgs = 1;
1388     dispparams.rgdispidNamedArgs = NULL;
1389     dispparams.rgvarg = vararg;
1390     V_VT(vararg) = VT_DISPATCH;
1391     V_DISPATCH(vararg) = (IDispatch*)&TestDualDisp;
1392     VariantInit(&varresult);
1393     hr = ITypeInfo_Invoke(type_info, &StaticWidget, DISPID_TM_TESTDUAL, DISPATCH_METHOD,
1394             &dispparams, &varresult, &excepinfo, NULL);
1395     ok_ole_success(hr, IDispatch_Invoke);
1396     ok(V_VT(&varresult) == VT_EMPTY, "vt %x\n", V_VT(&varresult));
1397     VariantClear(&varresult);
1398
1399     ITypeInfo_Release(type_info);
1400 }
1401
1402 static void test_libattr(void)
1403 {
1404     ITypeLib *pTypeLib;
1405     HRESULT hr;
1406     TLIBATTR *pattr;
1407
1408     hr = LoadRegTypeLib(&LIBID_TestTypelib, 1, 0, LOCALE_NEUTRAL, &pTypeLib);
1409     ok_ole_success(hr, LoadRegTypeLib);
1410     if (FAILED(hr))
1411         return;
1412
1413     hr = ITypeLib_GetLibAttr(pTypeLib, &pattr);
1414     ok_ole_success(hr, GetLibAttr);
1415     if (SUCCEEDED(hr))
1416     {
1417         ok(pattr->lcid == MAKELANGID(LANG_NEUTRAL, SUBLANG_NEUTRAL), "lcid %x\n", pattr->lcid);
1418
1419         ITypeLib_ReleaseTLibAttr(pTypeLib, pattr);
1420     }
1421
1422     ITypeLib_Release(pTypeLib);
1423 }
1424
1425 START_TEST(tmarshal)
1426 {
1427     HRESULT hr;
1428
1429     CoInitializeEx(NULL, COINIT_APARTMENTTHREADED);
1430
1431     hr = register_current_module_typelib();
1432     ok_ole_success(hr, register_current_module_typelib);
1433
1434     test_typelibmarshal();
1435     test_DispCallFunc();
1436     test_StaticWidget();
1437     test_libattr();
1438
1439     hr = UnRegisterTypeLib(&LIBID_TestTypelib, 1, 0, LOCALE_NEUTRAL,
1440                            sizeof(void*) == 8 ? SYS_WIN64 : SYS_WIN32);
1441     ok_ole_success(hr, UnRegisterTypeLib);
1442
1443     CoUninitialize();
1444 }