2 * Copyright (C) 2005-2006 Robert Shearman for CodeWeavers
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.
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.
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
26 #include <wine/test.h>
29 #include "tmarshal_dispids.h"
31 #define ok_ole_success(hr, func) ok(hr == S_OK, #func " failed with error 0x%08lx\n", (unsigned long int)hr)
33 /* Debugging functions from wine/libs/wine/debug.c */
35 /* allocate some tmp string space */
36 /* FIXME: this is not 100% thread-safe */
37 static char *get_tmp_space( int size )
39 static char *list[32];
44 idx = ++pos % (sizeof(list)/sizeof(list[0]));
45 if ((ret = realloc( list[idx], size ))) list[idx] = ret;
49 /* default implementation of wine_dbgstr_wn */
50 static const char *default_dbgstr_wn( const WCHAR *str, int n )
56 if (!str) return "(null)";
57 res = get_tmp_space( 6 );
58 sprintf( res, "#%04x", LOWORD(str) );
61 if (n == -1) n = lstrlenW(str);
63 else if (n > 200) n = 200;
64 dst = res = get_tmp_space( n * 5 + 7 );
72 case '\n': *dst++ = '\\'; *dst++ = 'n'; break;
73 case '\r': *dst++ = '\\'; *dst++ = 'r'; break;
74 case '\t': *dst++ = '\\'; *dst++ = 't'; break;
75 case '"': *dst++ = '\\'; *dst++ = '"'; break;
76 case '\\': *dst++ = '\\'; *dst++ = '\\'; break;
78 if (c >= ' ' && c <= 126)
83 sprintf(dst,"%04x",c);
99 const char *wine_dbgstr_wn( const WCHAR *s, int n )
101 return default_dbgstr_wn(s, n);
104 const char *wine_dbgstr_w( const WCHAR *s )
106 return default_dbgstr_wn( s, -1 );
110 #define RELEASEMARSHALDATA WM_USER
112 struct host_object_data
117 MSHLFLAGS marshal_flags;
118 HANDLE marshal_event;
119 IMessageFilter *filter;
122 static DWORD CALLBACK host_object_proc(LPVOID p)
124 struct host_object_data *data = (struct host_object_data *)p;
128 CoInitializeEx(NULL, COINIT_APARTMENTTHREADED);
132 IMessageFilter * prev_filter = NULL;
133 hr = CoRegisterMessageFilter(data->filter, &prev_filter);
134 if (prev_filter) IMessageFilter_Release(prev_filter);
135 ok_ole_success(hr, CoRegisterMessageFilter);
138 hr = CoMarshalInterface(data->stream, &data->iid, data->object, MSHCTX_INPROC, NULL, data->marshal_flags);
139 ok_ole_success(hr, CoMarshalInterface);
141 /* force the message queue to be created before signaling parent thread */
142 PeekMessage(&msg, NULL, WM_USER, WM_USER, PM_NOREMOVE);
144 SetEvent(data->marshal_event);
146 while (GetMessage(&msg, NULL, 0, 0))
148 if (msg.hwnd == NULL && msg.message == RELEASEMARSHALDATA)
150 trace("releasing marshal data\n");
151 CoReleaseMarshalData(data->stream);
152 SetEvent((HANDLE)msg.lParam);
155 DispatchMessage(&msg);
158 HeapFree(GetProcessHeap(), 0, data);
165 static DWORD start_host_object2(IStream *stream, REFIID riid, IUnknown *object, MSHLFLAGS marshal_flags, IMessageFilter *filter, HANDLE *thread)
168 HANDLE marshal_event = CreateEvent(NULL, FALSE, FALSE, NULL);
169 struct host_object_data *data = HeapAlloc(GetProcessHeap(), 0, sizeof(*data));
171 data->stream = stream;
173 data->object = object;
174 data->marshal_flags = marshal_flags;
175 data->marshal_event = marshal_event;
176 data->filter = filter;
178 *thread = CreateThread(NULL, 0, host_object_proc, data, 0, &tid);
180 /* wait for marshaling to complete before returning */
181 WaitForSingleObject(marshal_event, INFINITE);
182 CloseHandle(marshal_event);
187 static DWORD start_host_object(IStream *stream, REFIID riid, IUnknown *object, MSHLFLAGS marshal_flags, HANDLE *thread)
189 return start_host_object2(stream, riid, object, marshal_flags, NULL, thread);
193 /* asks thread to release the marshal data because it has to be done by the
194 * same thread that marshaled the interface in the first place. */
195 static void release_host_object(DWORD tid)
197 HANDLE event = CreateEvent(NULL, FALSE, FALSE, NULL);
198 PostThreadMessage(tid, RELEASEMARSHALDATA, 0, (LPARAM)event);
199 WaitForSingleObject(event, INFINITE);
204 static void end_host_object(DWORD tid, HANDLE thread)
206 BOOL ret = PostThreadMessage(tid, WM_QUIT, 0, 0);
207 ok(ret, "PostThreadMessage failed with error %d\n", GetLastError());
208 /* be careful of races - don't return until hosting thread has terminated */
209 WaitForSingleObject(thread, INFINITE);
213 typedef struct Widget
215 const IWidgetVtbl *lpVtbl;
217 IUnknown *pDispatchUnknown;
220 HRESULT WINAPI Widget_QueryInterface(
222 /* [in] */ REFIID riid,
223 /* [iid_is][out] */ void __RPC_FAR *__RPC_FAR *ppvObject)
225 if (IsEqualIID(riid, &IID_IWidget) || IsEqualIID(riid, &IID_IUnknown) || IsEqualIID(riid, &IID_IDispatch))
227 IWidget_AddRef(iface);
234 return E_NOINTERFACE;
238 ULONG WINAPI Widget_AddRef(
241 Widget *This = (Widget *)iface;
243 return InterlockedIncrement(&This->refs);
246 ULONG WINAPI Widget_Release(
249 Widget *This = (Widget *)iface;
250 ULONG refs = InterlockedDecrement(&This->refs);
253 IUnknown_Release(This->pDispatchUnknown);
254 memset(This, 0xcc, sizeof(*This));
255 HeapFree(GetProcessHeap(), 0, This);
256 trace("Widget destroyed!\n");
262 HRESULT WINAPI Widget_GetTypeInfoCount(
264 /* [out] */ UINT __RPC_FAR *pctinfo)
266 Widget *This = (Widget *)iface;
267 IDispatch *pDispatch;
268 HRESULT hr = IUnknown_QueryInterface(This->pDispatchUnknown, &IID_IDispatch, (void **)&pDispatch);
271 hr = IDispatch_GetTypeInfoCount(pDispatch, pctinfo);
272 IDispatch_Release(pDispatch);
277 HRESULT WINAPI Widget_GetTypeInfo(
278 IWidget __RPC_FAR * iface,
279 /* [in] */ UINT iTInfo,
280 /* [in] */ LCID lcid,
281 /* [out] */ ITypeInfo __RPC_FAR *__RPC_FAR *ppTInfo)
283 Widget *This = (Widget *)iface;
284 IDispatch *pDispatch;
285 HRESULT hr = IUnknown_QueryInterface(This->pDispatchUnknown, &IID_IDispatch, (void **)&pDispatch);
288 hr = IDispatch_GetTypeInfo(pDispatch, iTInfo, lcid, ppTInfo);
289 IDispatch_Release(pDispatch);
294 HRESULT WINAPI Widget_GetIDsOfNames(
295 IWidget __RPC_FAR * iface,
296 /* [in] */ REFIID riid,
297 /* [size_is][in] */ LPOLESTR __RPC_FAR *rgszNames,
298 /* [in] */ UINT cNames,
299 /* [in] */ LCID lcid,
300 /* [size_is][out] */ DISPID __RPC_FAR *rgDispId)
302 Widget *This = (Widget *)iface;
303 IDispatch *pDispatch;
304 HRESULT hr = IUnknown_QueryInterface(This->pDispatchUnknown, &IID_IDispatch, (void **)&pDispatch);
307 hr = IDispatch_GetIDsOfNames(pDispatch, riid, rgszNames, cNames, lcid, rgDispId);
308 IDispatch_Release(pDispatch);
313 HRESULT WINAPI Widget_Invoke(
314 IWidget __RPC_FAR * iface,
315 /* [in] */ DISPID dispIdMember,
316 /* [in] */ REFIID riid,
317 /* [in] */ LCID lcid,
318 /* [in] */ WORD wFlags,
319 /* [out][in] */ DISPPARAMS __RPC_FAR *pDispParams,
320 /* [out] */ VARIANT __RPC_FAR *pVarResult,
321 /* [out] */ EXCEPINFO __RPC_FAR *pExcepInfo,
322 /* [out] */ UINT __RPC_FAR *puArgErr)
324 Widget *This = (Widget *)iface;
325 IDispatch *pDispatch;
326 HRESULT hr = IUnknown_QueryInterface(This->pDispatchUnknown, &IID_IDispatch, (void **)&pDispatch);
329 hr = IDispatch_Invoke(pDispatch, dispIdMember, riid, lcid, wFlags, pDispParams, pVarResult, pExcepInfo, puArgErr);
330 IDispatch_Release(pDispatch);
335 HRESULT WINAPI Widget_put_Name(
336 IWidget __RPC_FAR * iface,
337 /* [in] */ BSTR name)
339 trace("put_Name(%s)\n", wine_dbgstr_w(name));
343 HRESULT WINAPI Widget_get_Name(
344 IWidget __RPC_FAR * iface,
345 /* [out] */ BSTR __RPC_FAR *name)
347 static const WCHAR szCat[] = { 'C','a','t',0 };
348 trace("get_Name()\n");
349 *name = SysAllocString(szCat);
353 HRESULT WINAPI Widget_DoSomething(
354 IWidget __RPC_FAR * iface,
355 /* [in] */ double number,
356 /* [out] */ BSTR *str1,
357 /* [defaultvalue][in] */ BSTR str2,
358 /* [optional][in] */ VARIANT __RPC_FAR *opt)
360 static const WCHAR szString[] = { 'S','t','r','i','n','g',0 };
361 trace("DoSomething()\n");
363 ok(number == 3.141, "number(%f) != 3.141\n", number);
364 ok(*str2 == '\0', "str2(%s) != \"\"\n", wine_dbgstr_w(str2));
365 ok(V_VT(opt) == VT_ERROR, "V_VT(opt) should be VT_ERROR instead of 0x%x\n", V_VT(opt));
366 ok(V_ERROR(opt) == DISP_E_PARAMNOTFOUND, "V_ERROR(opt) should be DISP_E_PARAMNOTFOUND instead of 0x%08x\n", V_ERROR(opt));
367 *str1 = SysAllocString(szString);
372 HRESULT WINAPI Widget_get_State(
373 IWidget __RPC_FAR * iface,
374 /* [retval][out] */ STATE __RPC_FAR *state)
376 trace("get_State() = STATE_WIDGETIFIED\n");
377 *state = STATE_WIDGETIFIED;
381 HRESULT WINAPI Widget_put_State(
382 IWidget __RPC_FAR * iface,
383 /* [in] */ STATE state)
385 trace("put_State(%d)\n", state);
389 HRESULT WINAPI Widget_Map(
394 trace("Map(%s, %p)\n", wine_dbgstr_w(bstrId), sValue);
395 *sValue = SysAllocString(bstrId);
399 HRESULT WINAPI Widget_SetOleColor(
403 trace("SetOleColor(0x%x)\n", val);
407 HRESULT WINAPI Widget_GetOleColor(
411 trace("GetOleColor() = 0x8000000f\n");
416 HRESULT WINAPI Widget_Clone(
421 return Widget_QueryInterface(iface, &IID_IWidget, (void **)ppVal);
424 HRESULT WINAPI Widget_CloneDispatch(
428 trace("CloneDispatch()\n");
429 return Widget_QueryInterface(iface, &IID_IWidget, (void **)ppVal);
432 HRESULT WINAPI Widget_CloneCoclass(
434 ApplicationObject2 **ppVal)
436 trace("CloneCoclass()\n");
440 HRESULT WINAPI Widget_Value(
441 IWidget __RPC_FAR * iface,
445 trace("Value(%p, %p)\n", value, retval);
446 ok(V_VT(value) == VT_I2, "V_VT(value) was %d instead of VT_I2\n", V_VT(value));
447 ok(V_I2(value) == 1, "V_I2(value) was %d instead of 1\n", V_I2(value));
448 V_VT(retval) = VT_I2;
453 HRESULT WINAPI Widget_Array(
457 trace("Array(%p)\n", values);
461 HRESULT WINAPI Widget_VariantArrayPtr(
465 trace("VariantArrayPtr(%p)\n", values);
469 void WINAPI Widget_Variant(
470 IWidget __RPC_FAR * iface,
473 trace("Variant()\n");
474 ok(V_VT(&var) == VT_CY, "V_VT(&var) was %d\n", V_VT(&var));
475 ok(S(V_CY(&var)).Hi == 0xdababe, "V_CY(&var).Hi was 0x%x\n", S(V_CY(&var)).Hi);
476 ok(S(V_CY(&var)).Lo == 0xdeadbeef, "V_CY(&var).Lo was 0x%x\n", S(V_CY(&var)).Lo);
479 HRESULT WINAPI Widget_Error(
480 IWidget __RPC_FAR * iface)
486 static const struct IWidgetVtbl Widget_VTable =
488 Widget_QueryInterface,
491 Widget_GetTypeInfoCount,
493 Widget_GetIDsOfNames,
504 Widget_CloneDispatch,
508 Widget_VariantArrayPtr,
514 typedef struct KindaEnum
516 const IKindaEnumWidgetVtbl *lpVtbl;
520 static HRESULT register_current_module_typelib(void)
522 WCHAR path[MAX_PATH];
526 GetModuleFileNameW(NULL, path, MAX_PATH);
528 hr = LoadTypeLib(path, &typelib);
531 hr = RegisterTypeLib(typelib, path, NULL);
532 ITypeLib_Release(typelib);
537 static IWidget *Widget_Create(void)
539 Widget *This = (Widget *)HeapAlloc(GetProcessHeap(), 0, sizeof(*This));
543 This->lpVtbl = &Widget_VTable;
546 hr = LoadRegTypeLib(&LIBID_TestTypelib, 1, 0, LOCALE_NEUTRAL, &pTypeLib);
547 ok_ole_success(hr, LoadRegTypeLib);
550 ITypeInfo *pTypeInfo;
551 hr = ITypeLib_GetTypeInfoOfGuid(pTypeLib, &IID_IWidget, &pTypeInfo);
552 ok_ole_success(hr, ITypeLib_GetTypeInfoOfGuid);
555 This->pDispatchUnknown = NULL;
556 hr = CreateStdDispatch((IUnknown *)&This->lpVtbl, This, pTypeInfo, &This->pDispatchUnknown);
557 ok_ole_success(hr, CreateStdDispatch);
558 ITypeInfo_Release(pTypeInfo);
563 return (IWidget *)&This->lpVtbl;
566 HeapFree(GetProcessHeap(), 0, This);
571 HRESULT WINAPI KindaEnum_QueryInterface(
572 IKindaEnumWidget *iface,
573 /* [in] */ REFIID riid,
574 /* [iid_is][out] */ void __RPC_FAR *__RPC_FAR *ppvObject)
576 if (IsEqualIID(riid, &IID_IKindaEnumWidget) || IsEqualIID(riid, &IID_IUnknown))
578 IKindaEnumWidget_AddRef(iface);
585 return E_NOINTERFACE;
589 ULONG WINAPI KindaEnum_AddRef(
590 IKindaEnumWidget *iface)
592 KindaEnum *This = (KindaEnum *)iface;
594 return InterlockedIncrement(&This->refs);
597 ULONG WINAPI KindaEnum_Release(
598 IKindaEnumWidget *iface)
600 KindaEnum *This = (KindaEnum *)iface;
601 ULONG refs = InterlockedDecrement(&This->refs);
604 memset(This, 0xcc, sizeof(*This));
605 HeapFree(GetProcessHeap(), 0, This);
606 trace("KindaEnumWidget destroyed!\n");
612 HRESULT WINAPI KindaEnum_Next(
613 IKindaEnumWidget *iface,
614 /* [out] */ IWidget __RPC_FAR *__RPC_FAR *widget)
616 *widget = Widget_Create();
620 return E_OUTOFMEMORY;
623 HRESULT WINAPI KindaEnum_Count(
624 IKindaEnumWidget *iface,
625 /* [out] */ unsigned long __RPC_FAR *count)
630 HRESULT WINAPI KindaEnum_Reset(
631 IKindaEnumWidget *iface)
636 HRESULT WINAPI KindaEnum_Clone(
637 IKindaEnumWidget *iface,
638 /* [out] */ IKindaEnumWidget __RPC_FAR *__RPC_FAR *ppenum)
643 static const IKindaEnumWidgetVtbl KindaEnumWidget_VTable =
645 KindaEnum_QueryInterface,
654 static IKindaEnumWidget *KindaEnumWidget_Create(void)
658 This = (KindaEnum *)HeapAlloc(GetProcessHeap(), 0, sizeof(*This));
659 if (!This) return NULL;
660 This->lpVtbl = &KindaEnumWidget_VTable;
662 return (IKindaEnumWidget *)This;
665 static HRESULT WINAPI NonOleAutomation_QueryInterface(INonOleAutomation *iface, REFIID riid, void **ppv)
667 if (IsEqualIID(riid, &IID_IUnknown) || IsEqualIID(riid, &IID_INonOleAutomation))
669 *(INonOleAutomation **)ppv = iface;
673 return E_NOINTERFACE;
676 static ULONG WINAPI NonOleAutomation_AddRef(INonOleAutomation *iface)
681 static ULONG WINAPI NonOleAutomation_Release(INonOleAutomation *iface)
686 static BSTR WINAPI NonOleAutomation_BstrRet(INonOleAutomation *iface)
688 static const WCHAR wszTestString[] = {'T','h','i','s',' ','i','s',' ','a',' ','t','e','s','t',' ','s','t','r','i','n','g',0};
689 return SysAllocString(wszTestString);
692 static INonOleAutomationVtbl NonOleAutomation_VTable =
694 NonOleAutomation_QueryInterface,
695 NonOleAutomation_AddRef,
696 NonOleAutomation_Release,
697 NonOleAutomation_BstrRet,
700 static INonOleAutomation NonOleAutomation = { &NonOleAutomation_VTable };
702 static ITypeInfo *NonOleAutomation_GetTypeInfo(void)
705 HRESULT hr = LoadRegTypeLib(&LIBID_TestTypelib, 1, 0, LOCALE_NEUTRAL, &pTypeLib);
706 ok_ole_success(hr, LoadRegTypeLib);
709 ITypeInfo *pTypeInfo;
710 hr = ITypeLib_GetTypeInfoOfGuid(pTypeLib, &IID_INonOleAutomation, &pTypeInfo);
711 ok_ole_success(hr, ITypeLib_GetTypeInfoOfGuid);
717 static void test_typelibmarshal(void)
719 static const WCHAR szCat[] = { 'C','a','t',0 };
720 static const WCHAR szTestTest[] = { 'T','e','s','t','T','e','s','t',0 };
721 static WCHAR szSuperman[] = { 'S','u','p','e','r','m','a','n',0 };
723 IKindaEnumWidget *pKEW = KindaEnumWidget_Create();
726 IDispatch *pDispatch;
727 static const LARGE_INTEGER ullZero;
730 DISPID dispidNamed = DISPID_PROPERTYPUT;
731 DISPPARAMS dispparams;
732 VARIANTARG vararg[4];
737 ITypeInfo *pTypeInfo;
739 ok(pKEW != NULL, "Widget creation failed\n");
741 hr = CreateStreamOnHGlobal(NULL, TRUE, &pStream);
742 ok_ole_success(hr, CreateStreamOnHGlobal);
743 tid = start_host_object(pStream, &IID_IKindaEnumWidget, (IUnknown *)pKEW, MSHLFLAGS_NORMAL, &thread);
745 IStream_Seek(pStream, ullZero, STREAM_SEEK_SET, NULL);
746 hr = CoUnmarshalInterface(pStream, &IID_IKindaEnumWidget, (void **)&pKEW);
747 ok_ole_success(hr, CoUnmarshalInterface);
748 IStream_Release(pStream);
750 hr = IKindaEnumWidget_Next(pKEW, &pWidget);
751 ok_ole_success(hr, IKindaEnumWidget_Next);
753 IKindaEnumWidget_Release(pKEW);
755 hr = IWidget_QueryInterface(pWidget, &IID_IDispatch, (void **)&pDispatch);
758 VariantInit(&vararg[0]);
759 dispparams.cNamedArgs = 1;
760 dispparams.rgdispidNamedArgs = &dispidNamed;
761 dispparams.cArgs = 1;
762 dispparams.rgvarg = vararg;
763 VariantInit(&varresult);
764 hr = IDispatch_Invoke(pDispatch, DISPID_TM_NAME, &IID_NULL, LOCALE_NEUTRAL, DISPATCH_PROPERTYPUT, &dispparams, &varresult, &excepinfo, NULL);
765 ok_ole_success(hr, IDispatch_Invoke);
766 ok(excepinfo.wCode == 0x0 && excepinfo.scode == S_OK,
767 "EXCEPINFO differs from expected: wCode = 0x%x, scode = 0x%08x\n",
768 excepinfo.wCode, excepinfo.scode);
769 VariantClear(&varresult);
771 /* call put_Name (direct) */
772 bstr = SysAllocString(szSuperman);
773 hr = IWidget_put_Name(pWidget, bstr);
774 ok_ole_success(hr, IWidget_put_Name);
778 dispparams.cNamedArgs = 0;
779 dispparams.rgdispidNamedArgs = NULL;
780 dispparams.cArgs = 0;
781 dispparams.rgvarg = NULL;
782 VariantInit(&varresult);
783 hr = IDispatch_Invoke(pDispatch, DISPID_TM_NAME, &IID_NULL, LOCALE_NEUTRAL, DISPATCH_PROPERTYGET, &dispparams, &varresult, &excepinfo, NULL);
784 ok_ole_success(hr, IDispatch_Invoke);
785 ok(excepinfo.wCode == 0x0 && excepinfo.scode == S_OK,
786 "EXCEPINFO differs from expected: wCode = 0x%x, scode = 0x%08x\n",
787 excepinfo.wCode, excepinfo.scode);
788 trace("Name = %s\n", wine_dbgstr_w(V_BSTR(&varresult)));
789 VariantClear(&varresult);
791 /* call get_Name (direct) */
793 hr = IWidget_get_Name(pWidget, &bstr);
794 ok_ole_success(hr, IWidget_get_Name);
795 ok(!lstrcmpW(bstr, szCat), "IWidget_get_Name should have returned string \"Cat\" instead of %s\n", wine_dbgstr_w(bstr));
798 /* call DoSomething */
799 VariantInit(&vararg[0]);
800 VariantInit(&vararg[1]);
801 V_VT(&vararg[1]) = VT_R8;
802 V_R8(&vararg[1]) = 3.141;
803 dispparams.cNamedArgs = 0;
804 dispparams.cArgs = 2;
805 dispparams.rgdispidNamedArgs = NULL;
806 dispparams.rgvarg = vararg;
807 VariantInit(&varresult);
808 hr = IDispatch_Invoke(pDispatch, DISPID_TM_DOSOMETHING, &IID_NULL, LOCALE_NEUTRAL, DISPATCH_METHOD, &dispparams, &varresult, &excepinfo, NULL);
809 ok_ole_success(hr, IDispatch_Invoke);
810 ok(V_VT(&varresult) == VT_EMPTY, "varresult should be VT_EMPTY\n");
811 VariantClear(&varresult);
814 dispparams.cNamedArgs = 0;
815 dispparams.cArgs = 0;
816 dispparams.rgdispidNamedArgs = NULL;
817 dispparams.rgvarg = NULL;
818 hr = IDispatch_Invoke(pDispatch, DISPID_TM_STATE, &IID_NULL, LOCALE_NEUTRAL, DISPATCH_PROPERTYGET, &dispparams, &varresult, &excepinfo, NULL);
819 ok_ole_success(hr, IDispatch_Invoke);
820 ok((V_VT(&varresult) == VT_I4) && (V_I4(&varresult) == STATE_WIDGETIFIED), "Return val mismatch\n");
822 /* call get_State (direct) */
823 hr = IWidget_get_State(pWidget, &the_state);
824 ok_ole_success(hr, IWidget_get_state);
825 ok(the_state == STATE_WIDGETIFIED, "should have returned WIDGET_WIDGETIFIED instead of %d\n", the_state);
828 the_state = STATE_WIDGETIFIED;
829 VariantInit(&vararg[0]);
830 V_VT(&vararg[0]) = VT_BYREF|VT_I4;
831 V_I4REF(&vararg[0]) = (int *)&the_state;
832 dispparams.cNamedArgs = 1;
833 dispparams.cArgs = 1;
834 dispparams.rgdispidNamedArgs = &dispidNamed;
835 dispparams.rgvarg = vararg;
836 hr = IDispatch_Invoke(pDispatch, DISPID_TM_STATE, &IID_NULL, LOCALE_NEUTRAL, DISPATCH_PROPERTYPUT, &dispparams, &varresult, &excepinfo, NULL);
837 ok_ole_success(hr, IDispatch_Invoke);
840 bstr = SysAllocString(szTestTest);
841 VariantInit(&vararg[0]);
842 V_VT(&vararg[0]) = VT_BYREF|VT_BSTR;
843 V_BSTRREF(&vararg[0]) = &bstr;
844 dispparams.cNamedArgs = 0;
845 dispparams.cArgs = 1;
846 dispparams.rgdispidNamedArgs = NULL;
847 dispparams.rgvarg = vararg;
848 VariantInit(&varresult);
849 hr = IDispatch_Invoke(pDispatch, DISPID_TM_MAP, &IID_NULL, LOCALE_NEUTRAL, DISPATCH_METHOD, &dispparams, &varresult, &excepinfo, NULL);
850 ok_ole_success(hr, IDispatch_Invoke);
851 ok(V_VT(&varresult) == VT_BSTR, "Return value should be of type BSTR instead of %d\n", V_VT(&varresult));
852 ok(!lstrcmpW(V_BSTR(&varresult), szTestTest), "Return value should have been \"TestTest\" instead of %s\n", wine_dbgstr_w(V_BSTR(&varresult)));
853 VariantClear(&varresult);
855 /* call SetOleColor with large negative VT_I4 param */
856 VariantInit(&vararg[0]);
857 V_VT(&vararg[0]) = VT_I4;
858 V_I4(&vararg[0]) = 0x80000005;
859 dispparams.cNamedArgs = 0;
860 dispparams.cArgs = 1;
861 dispparams.rgdispidNamedArgs = NULL;
862 dispparams.rgvarg = vararg;
863 hr = IDispatch_Invoke(pDispatch, DISPID_TM_SETOLECOLOR, &IID_NULL, LOCALE_NEUTRAL, DISPATCH_METHOD, &dispparams, NULL, &excepinfo, NULL);
864 ok_ole_success(hr, IDispatch_Invoke);
866 /* call GetOleColor */
867 dispparams.cNamedArgs = 0;
868 dispparams.cArgs = 0;
869 dispparams.rgdispidNamedArgs = NULL;
870 dispparams.rgvarg = NULL;
871 VariantInit(&varresult);
872 hr = IDispatch_Invoke(pDispatch, DISPID_TM_GETOLECOLOR, &IID_NULL, LOCALE_NEUTRAL, DISPATCH_METHOD, &dispparams, &varresult, &excepinfo, NULL);
873 ok_ole_success(hr, IDispatch_Invoke);
874 VariantClear(&varresult);
877 dispparams.cNamedArgs = 0;
878 dispparams.cArgs = 0;
879 dispparams.rgdispidNamedArgs = NULL;
880 dispparams.rgvarg = NULL;
881 VariantInit(&varresult);
882 hr = IDispatch_Invoke(pDispatch, DISPID_TM_CLONE, &IID_NULL, LOCALE_NEUTRAL, DISPATCH_PROPERTYGET, &dispparams, &varresult, &excepinfo, NULL);
883 ok_ole_success(hr, IDispatch_Invoke);
884 VariantClear(&varresult);
886 /* call CloneDispatch with automatic value getting */
887 V_VT(&vararg[0]) = VT_I2;
888 V_I2(&vararg[0]) = 1;
889 dispparams.cNamedArgs = 0;
890 dispparams.rgdispidNamedArgs = NULL;
891 dispparams.cArgs = 1;
892 dispparams.rgvarg = vararg;
893 VariantInit(&varresult);
894 hr = IDispatch_Invoke(pDispatch, DISPID_TM_CLONEDISPATCH, &IID_NULL, LOCALE_NEUTRAL, DISPATCH_PROPERTYGET, &dispparams, &varresult, &excepinfo, NULL);
895 ok_ole_success(hr, IDispatch_Invoke);
897 ok(excepinfo.wCode == 0x0 && excepinfo.scode == S_OK,
898 "EXCEPINFO differs from expected: wCode = 0x%x, scode = 0x%08x\n",
899 excepinfo.wCode, excepinfo.scode);
901 ok(V_VT(&varresult) == VT_I2, "V_VT(&varresult) was %d instead of VT_I2\n", V_VT(&varresult));
902 ok(V_I2(&varresult) == 1234, "V_I2(&varresult) was %d instead of 1234\n", V_I2(&varresult));
903 VariantClear(&varresult);
905 /* call CloneCoclass */
906 dispparams.cNamedArgs = 0;
907 dispparams.cArgs = 0;
908 dispparams.rgdispidNamedArgs = NULL;
909 dispparams.rgvarg = NULL;
910 VariantInit(&varresult);
911 hr = IDispatch_Invoke(pDispatch, DISPID_TM_CLONECOCLASS, &IID_NULL, LOCALE_NEUTRAL, DISPATCH_PROPERTYGET, &dispparams, &varresult, &excepinfo, NULL);
912 ok_ole_success(hr, IDispatch_Invoke);
914 ok(excepinfo.wCode == 0x0 && excepinfo.scode == S_OK,
915 "EXCEPINFO differs from expected: wCode = 0x%x, scode = 0x%08x\n",
916 excepinfo.wCode, excepinfo.scode);
918 ok(V_VT(&varresult) == VT_DISPATCH, "V_VT(&varresult) was %d instead of VT_DISPATCH\n", V_VT(&varresult));
919 ok(!V_DISPATCH(&varresult), "V_DISPATCH(&varresult) should be NULL instead of %p\n", V_DISPATCH(&varresult));
920 VariantClear(&varresult);
922 /* call Value with a VT_VARIANT|VT_BYREF type */
923 V_VT(&vararg[0]) = VT_VARIANT|VT_BYREF;
924 V_VARIANTREF(&vararg[0]) = &vararg[1];
925 V_VT(&vararg[1]) = VT_I2;
926 V_I2(&vararg[1]) = 1;
927 dispparams.cNamedArgs = 0;
928 dispparams.rgdispidNamedArgs = NULL;
929 dispparams.cArgs = 1;
930 dispparams.rgvarg = vararg;
931 VariantInit(&varresult);
932 hr = IDispatch_Invoke(pDispatch, DISPID_VALUE, &IID_NULL, LOCALE_NEUTRAL, DISPATCH_PROPERTYGET, &dispparams, &varresult, &excepinfo, NULL);
933 ok_ole_success(hr, IDispatch_Invoke);
935 ok(excepinfo.wCode == 0x0 && excepinfo.scode == S_OK,
936 "EXCEPINFO differs from expected: wCode = 0x%x, scode = 0x%08x\n",
937 excepinfo.wCode, excepinfo.scode);
939 ok(V_VT(&varresult) == VT_I2, "V_VT(&varresult) was %d instead of VT_I2\n", V_VT(&varresult));
940 ok(V_I2(&varresult) == 1234, "V_I2(&varresult) was %d instead of 1234\n", V_I2(&varresult));
941 VariantClear(&varresult);
943 /* call Variant - exercises variant copying in ITypeInfo::Invoke and
944 * handling of void return types */
945 /* use a big type to ensure that the variant was properly copied into the
946 * destination function's args */
947 V_VT(&vararg[0]) = VT_CY;
948 S(V_CY(&vararg[0])).Hi = 0xdababe;
949 S(V_CY(&vararg[0])).Lo = 0xdeadbeef;
950 dispparams.cNamedArgs = 0;
951 dispparams.cArgs = 1;
952 dispparams.rgdispidNamedArgs = NULL;
953 dispparams.rgvarg = vararg;
954 VariantInit(&varresult);
955 hr = IDispatch_Invoke(pDispatch, DISPID_TM_VARIANT, &IID_NULL, LOCALE_NEUTRAL, DISPATCH_METHOD, &dispparams, NULL, NULL, NULL);
956 ok_ole_success(hr, IDispatch_Invoke);
957 VariantClear(&varresult);
960 dispparams.cNamedArgs = 0;
961 dispparams.cArgs = 0;
962 dispparams.rgdispidNamedArgs = NULL;
963 dispparams.rgvarg = NULL;
964 VariantInit(&varresult);
965 hr = IDispatch_Invoke(pDispatch, DISPID_TM_ERROR, &IID_NULL, LOCALE_NEUTRAL, DISPATCH_METHOD, &dispparams, NULL, &excepinfo, NULL);
966 ok(hr == DISP_E_EXCEPTION, "IDispatch_Invoke should have returned DISP_E_EXCEPTION instead of 0x%08x\n", hr);
967 ok(excepinfo.wCode == 0x0 && excepinfo.scode == E_NOTIMPL,
968 "EXCEPINFO differs from expected: wCode = 0x%x, scode = 0x%08x\n",
969 excepinfo.wCode, excepinfo.scode);
970 VariantClear(&varresult);
973 pTypeInfo = NonOleAutomation_GetTypeInfo();
974 dispparams.cNamedArgs = 0;
975 dispparams.cArgs = 0;
976 dispparams.rgdispidNamedArgs = NULL;
977 dispparams.rgvarg = NULL;
978 VariantInit(&varresult);
979 hr = ITypeInfo_Invoke(pTypeInfo, &NonOleAutomation, DISPID_NOA_BSTRRET, DISPATCH_METHOD, &dispparams, &varresult, &excepinfo, NULL);
980 ok_ole_success(hr, ITypeInfo_Invoke);
981 ok(V_VT(&varresult) == VT_BSTR, "V_VT(&varresult) should be VT_BSTR instead of %d\n", V_VT(&varresult));
982 ok(V_BSTR(&varresult) != NULL, "V_BSTR(&varresult) should not be NULL\n");
984 VariantClear(&varresult);
985 ITypeInfo_Release(pTypeInfo);
987 /* tests call put_Name without named arg */
988 VariantInit(&vararg[0]);
989 dispparams.cNamedArgs = 0;
990 dispparams.rgdispidNamedArgs = NULL;
991 dispparams.cArgs = 1;
992 dispparams.rgvarg = vararg;
993 VariantInit(&varresult);
994 hr = IDispatch_Invoke(pDispatch, DISPID_TM_NAME, &IID_NULL, LOCALE_NEUTRAL, DISPATCH_PROPERTYPUT, &dispparams, &varresult, &excepinfo, NULL);
995 ok(hr == DISP_E_PARAMNOTFOUND, "IDispatch_Invoke should have returned DISP_E_PARAMNOTFOUND instead of 0x%08x\n", hr);
996 VariantClear(&varresult);
998 /* tests param type that cannot be coerced */
999 VariantInit(&vararg[0]);
1000 V_VT(&vararg[0]) = VT_UNKNOWN;
1001 V_UNKNOWN(&vararg[0]) = NULL;
1002 dispparams.cNamedArgs = 1;
1003 dispparams.rgdispidNamedArgs = &dispidNamed;
1004 dispparams.cArgs = 1;
1005 dispparams.rgvarg = vararg;
1006 VariantInit(&varresult);
1007 #if 0 /* NULL unknown not currently marshaled correctly */
1008 hr = IDispatch_Invoke(pDispatch, DISPID_TM_NAME, &IID_NULL, LOCALE_NEUTRAL, DISPATCH_PROPERTYPUT, &dispparams, &varresult, &excepinfo, NULL);
1009 ok(hr == DISP_E_TYPEMISMATCH, "IDispatch_Invoke should have returned DISP_E_TYPEMISMATCH instead of 0x%08x\n", hr);
1011 VariantClear(&varresult);
1013 /* tests bad param type */
1014 VariantInit(&vararg[0]);
1015 V_VT(&vararg[0]) = VT_CLSID;
1016 V_BYREF(&vararg[0]) = NULL;
1017 dispparams.cNamedArgs = 1;
1018 dispparams.rgdispidNamedArgs = &dispidNamed;
1019 dispparams.cArgs = 1;
1020 dispparams.rgvarg = vararg;
1021 VariantInit(&varresult);
1022 hr = IDispatch_Invoke(pDispatch, DISPID_TM_NAME, &IID_NULL, LOCALE_NEUTRAL, DISPATCH_PROPERTYPUT, &dispparams, &varresult, &excepinfo, NULL);
1023 ok(hr == DISP_E_BADVARTYPE, "IDispatch_Invoke should have returned DISP_E_BADVARTYPE instead of 0x%08x\n", hr);
1024 VariantClear(&varresult);
1026 /* tests too small param count */
1027 dispparams.cNamedArgs = 0;
1028 dispparams.rgdispidNamedArgs = NULL;
1029 dispparams.cArgs = 0;
1030 dispparams.rgvarg = NULL;
1031 VariantInit(&varresult);
1032 hr = IDispatch_Invoke(pDispatch, DISPID_TM_DOSOMETHING, &IID_NULL, LOCALE_NEUTRAL, DISPATCH_METHOD, &dispparams, &varresult, &excepinfo, NULL);
1033 ok(hr == DISP_E_BADPARAMCOUNT, "IDispatch_Invoke should have returned DISP_E_BADPARAMCOUNT instead of 0x%08x\n", hr);
1034 VariantClear(&varresult);
1036 /* tests propget function with large param count */
1037 VariantInit(&vararg[0]);
1038 V_VT(&vararg[0]) = VT_BSTR;
1039 V_BSTR(&vararg[0]) = NULL;
1040 V_VT(&vararg[1]) = VT_I4;
1041 V_I4(&vararg[1]) = 1;
1042 dispparams.cNamedArgs = 0;
1043 dispparams.cArgs = 2;
1044 dispparams.rgdispidNamedArgs = NULL;
1045 dispparams.rgvarg = vararg;
1046 hr = IDispatch_Invoke(pDispatch, DISPID_TM_STATE, &IID_NULL, LOCALE_NEUTRAL, DISPATCH_PROPERTYGET, &dispparams, &varresult, &excepinfo, NULL);
1047 ok(hr == DISP_E_NOTACOLLECTION, "IDispatch_Invoke should have returned DISP_E_NOTACOLLECTION instead of 0x%08x\n", hr);
1049 IDispatch_Release(pDispatch);
1050 IWidget_Release(pWidget);
1052 trace("calling end_host_object\n");
1053 end_host_object(tid, thread);
1056 static void test_DispCallFunc(void)
1058 static const WCHAR szEmpty[] = { 0 };
1059 VARTYPE rgvt[] = { VT_R8, VT_BSTR, VT_BSTR, VT_VARIANT|VT_BYREF };
1060 VARIANTARG vararg[4];
1062 VARIANTARG *rgpvarg[4] = { &vararg[0], &vararg[1], &vararg[2], &vararg[3] };
1063 VARIANTARG varresult;
1065 IWidget *pWidget = Widget_Create();
1066 V_VT(&vararg[0]) = VT_R8;
1067 V_R8(&vararg[0]) = 3.141;
1068 V_VT(&vararg[1]) = VT_BSTR;
1069 V_BSTR(&vararg[1]) = SysAllocString(szEmpty);
1070 V_VT(&vararg[2]) = VT_BSTR;
1071 V_BSTR(&vararg[2]) = SysAllocString(szEmpty);
1072 V_VT(&vararg[3]) = VT_VARIANT|VT_BYREF;
1073 V_VARIANTREF(&vararg[3]) = &varref;
1074 V_VT(&varref) = VT_ERROR;
1075 V_ERROR(&varref) = DISP_E_PARAMNOTFOUND;
1076 VariantInit(&varresult);
1077 hr = DispCallFunc(pWidget, 36, CC_STDCALL, VT_UI4, 4, rgvt, rgpvarg, &varresult);
1078 ok_ole_success(hr, DispCallFunc);
1079 VariantClear(&varresult);
1082 START_TEST(tmarshal)
1086 CoInitializeEx(NULL, COINIT_APARTMENTTHREADED);
1088 hr = register_current_module_typelib();
1089 ok_ole_success(hr, register_current_module_typelib);
1091 test_typelibmarshal();
1092 test_DispCallFunc();