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 /* Debugging functions from wine/libs/wine/debug.c */
  48
  49 /* allocate some tmp string space */
  50 /* FIXME: this is not 100% thread-safe */
  51 static char *get_tmp_space( int size )
  52 {
  53     static char *list[32];
  54     static long pos;
  55     char *ret;
  56     int idx;
  57
  58     idx = ++pos % (sizeof(list)/sizeof(list[0]));
  59     if (list[idx])
  60         ret = HeapReAlloc( GetProcessHeap(), 0, list[idx], size );
  61     else
  62         ret = HeapAlloc( GetProcessHeap(), 0, size );
  63     if (ret) list[idx] = ret;
  64     return ret;
  65 }
  66
  67 /* default implementation of wine_dbgstr_wn */
  68 static const char *default_dbgstr_wn( const WCHAR *str, int n )
  69 {
  70     char *dst, *res;
  71
  72     if (!HIWORD(str))
  73     {
  74         if (!str) return "(null)";
  75         res = get_tmp_space( 6 );
  76         sprintf( res, "#%04x", LOWORD(str) );
  77         return res;
  78     }
  79     if (n == -1) n = lstrlenW(str);
  80     if (n < 0) n = 0;
  81     else if (n > 200) n = 200;
  82     dst = res = get_tmp_space( n * 5 + 7 );
  83     *dst++ = 'L';
  84     *dst++ = '"';
  85     while (n-- > 0)
  86     {
  87         WCHAR c = *str++;
  88         switch (c)
  89         {
  90         case '\n': *dst++ = '\\'; *dst++ = 'n'; break;
  91         case '\r': *dst++ = '\\'; *dst++ = 'r'; break;
  92         case '\t': *dst++ = '\\'; *dst++ = 't'; break;
  93         case '"':  *dst++ = '\\'; *dst++ = '"'; break;
  94         case '\\': *dst++ = '\\'; *dst++ = '\\'; break;
  95         default:
  96             if (c >= ' ' && c <= 126)
  97                 *dst++ = (char)c;
  98             else
  99             {
 100                 *dst++ = '\\';
 101                 sprintf(dst,"%04x",c);
 102                 dst+=4;
 103             }
 104         }
 105     }
 106     *dst++ = '"';
 107     if (*str)
 108     {
 109         *dst++ = '.';
 110         *dst++ = '.';
 111         *dst++ = '.';
 112     }
 113     *dst = 0;
 114     return res;
 115 }
 116
 117 const char *wine_dbgstr_wn( const WCHAR *s, int n )
 118 {
 119     return default_dbgstr_wn(s, n);
 120 }
 121
 122 const char *wine_dbgstr_w( const WCHAR *s )
 123 {
 124     return default_dbgstr_wn( s, -1 );
 125 }
 126
 127
 128 #define RELEASEMARSHALDATA WM_USER
 129
 130 struct host_object_data
 131 {
 132     IStream *stream;
 133     IID iid;
 134     IUnknown *object;
 135     MSHLFLAGS marshal_flags;
 136     HANDLE marshal_event;
 137     IMessageFilter *filter;
 138 };
 139
 140 static DWORD CALLBACK host_object_proc(LPVOID p)
 141 {
 142     struct host_object_data *data = p;
 143     HRESULT hr;
 144     MSG msg;
 145
 146     CoInitializeEx(NULL, COINIT_APARTMENTTHREADED);
 147
 148     if (data->filter)
 149     {
 150         IMessageFilter * prev_filter = NULL;
 151         hr = CoRegisterMessageFilter(data->filter, &prev_filter);
 152         if (prev_filter) IMessageFilter_Release(prev_filter);
 153         ok_ole_success(hr, CoRegisterMessageFilter);
 154     }
 155
 156     hr = CoMarshalInterface(data->stream, &data->iid, data->object, MSHCTX_INPROC, NULL, data->marshal_flags);
 157     ok_ole_success(hr, CoMarshalInterface);
 158
 159     /* force the message queue to be created before signaling parent thread */
 160     PeekMessage(&msg, NULL, WM_USER, WM_USER, PM_NOREMOVE);
 161
 162     SetEvent(data->marshal_event);
 163
 164     while (GetMessage(&msg, NULL, 0, 0))
 165     {
 166         if (msg.hwnd == NULL && msg.message == RELEASEMARSHALDATA)
 167         {
 168             trace("releasing marshal data\n");
 169             CoReleaseMarshalData(data->stream);
 170             SetEvent((HANDLE)msg.lParam);
 171         }
 172         else
 173             DispatchMessage(&msg);
 174     }
 175
 176     HeapFree(GetProcessHeap(), 0, data);
 177
 178     CoUninitialize();
 179
 180     return hr;
 181 }
 182
 183 static DWORD start_host_object2(IStream *stream, REFIID riid, IUnknown *object, MSHLFLAGS marshal_flags, IMessageFilter *filter, HANDLE *thread)
 184 {
 185     DWORD tid = 0;
 186     HANDLE marshal_event = CreateEvent(NULL, FALSE, FALSE, NULL);
 187     struct host_object_data *data = HeapAlloc(GetProcessHeap(), 0, sizeof(*data));
 188
 189     data->stream = stream;
 190     data->iid = *riid;
 191     data->object = object;
 192     data->marshal_flags = marshal_flags;
 193     data->marshal_event = marshal_event;
 194     data->filter = filter;
 195
 196     *thread = CreateThread(NULL, 0, host_object_proc, data, 0, &tid);
 197
 198     /* wait for marshaling to complete before returning */
 199     WaitForSingleObject(marshal_event, INFINITE);
 200     CloseHandle(marshal_event);
 201
 202     return tid;
 203 }
 204
 205 static DWORD start_host_object(IStream *stream, REFIID riid, IUnknown *object, MSHLFLAGS marshal_flags, HANDLE *thread)
 206 {
 207     return start_host_object2(stream, riid, object, marshal_flags, NULL, thread);
 208 }
 209
 210 #if 0 /* not used */
 211 /* asks thread to release the marshal data because it has to be done by the
 212  * same thread that marshaled the interface in the first place. */
 213 static void release_host_object(DWORD tid)
 214 {
 215     HANDLE event = CreateEvent(NULL, FALSE, FALSE, NULL);
 216     PostThreadMessage(tid, RELEASEMARSHALDATA, 0, (LPARAM)event);
 217     WaitForSingleObject(event, INFINITE);
 218     CloseHandle(event);
 219 }
 220 #endif
 221
 222 static void end_host_object(DWORD tid, HANDLE thread)
 223 {
 224     BOOL ret = PostThreadMessage(tid, WM_QUIT, 0, 0);
 225     ok(ret, "PostThreadMessage failed with error %d\n", GetLastError());
 226     /* be careful of races - don't return until hosting thread has terminated */
 227     WaitForSingleObject(thread, INFINITE);
 228     CloseHandle(thread);
 229 }
 230
 231 static ItestDual TestDual, TestDualDisp;
 232
 233 static HRESULT WINAPI TestDual_QueryInterface(ItestDual *iface, REFIID riid, void **ppvObject)
 234 {
 235     if (IsEqualIID(riid, &IID_IUnknown) || IsEqualIID(riid, &IID_IDispatch)) {
 236         *ppvObject = &TestDualDisp;
 237         return S_OK;
 238     }else if(IsEqualGUID(riid, &IID_ItestDual)) {
 239         *ppvObject = &TestDual;
 240         return S_OK;
 241     }
 242
 243     *ppvObject = NULL;
 244     return E_NOINTERFACE;
 245 }
 246
 247 static ULONG WINAPI TestDual_AddRef(ItestDual *iface)
 248 {
 249     return 2;
 250 }
 251
 252 static ULONG WINAPI TestDual_Release(ItestDual *iface)
 253 {
 254     return 1;
 255 }
 256
 257 static HRESULT WINAPI TestDual_GetTypeInfoCount(ItestDual *iface, UINT *pctinfo)
 258 {
 259     ok(0, "unexpected call\n");
 260     return E_NOTIMPL;
 261 }
 262
 263 static HRESULT WINAPI TestDual_GetTypeInfo(ItestDual *iface, UINT iTInfo, LCID lcid, ITypeInfo **ppTInfo)
 264 {
 265     ok(0, "unexpected call\n");
 266     return E_NOTIMPL;
 267 }
 268
 269 static HRESULT WINAPI TestDual_GetIDsOfNames(ItestDual *iface, REFIID riid, LPOLESTR *rgszNames,
 270         UINT cNames, LCID lcid, DISPID *rgDispId)
 271 {
 272     ok(0, "unexpected call\n");
 273     return E_NOTIMPL;
 274 }
 275
 276 static HRESULT WINAPI TestDual_Invoke(ItestDual *iface, DISPID dispIdMember, REFIID riid, LCID lcid,
 277         WORD wFlags, DISPPARAMS *pDispParams, VARIANT *pVarResult, EXCEPINFO *pExcepInfo,
 278         UINT *puArgErr)
 279 {
 280     ok(0, "unexpected call\n");
 281     return E_NOTIMPL;
 282 }
 283
 284 static ItestDualVtbl TestDualVtbl = {
 285     TestDual_QueryInterface,
 286     TestDual_AddRef,
 287     TestDual_Release,
 288     TestDual_GetTypeInfoCount,
 289     TestDual_GetTypeInfo,
 290     TestDual_GetIDsOfNames,
 291     TestDual_Invoke
 292 };
 293
 294 static ItestDual TestDual = { &TestDualVtbl };
 295 static ItestDual TestDualDisp = { &TestDualVtbl };
 296
 297 typedef struct Widget
 298 {
 299     const IWidgetVtbl *lpVtbl;
 300     LONG refs;
 301     IUnknown *pDispatchUnknown;
 302 } Widget;
 303
 304 static HRESULT WINAPI Widget_QueryInterface(
 305     IWidget *iface,
 306     /* [in] */ REFIID riid,
 307     /* [iid_is][out] */ void __RPC_FAR *__RPC_FAR *ppvObject)
 308 {
 309     if (IsEqualIID(riid, &IID_IWidget) || IsEqualIID(riid, &IID_IUnknown) || IsEqualIID(riid, &IID_IDispatch))
 310     {
 311         IWidget_AddRef(iface);
 312         *ppvObject = iface;
 313         return S_OK;
 314     }
 315     else
 316     {
 317         *ppvObject = NULL;
 318         return E_NOINTERFACE;
 319     }
 320 }
 321
 322 static ULONG WINAPI Widget_AddRef(
 323     IWidget *iface)
 324 {
 325     Widget *This = (Widget *)iface;
 326
 327     return InterlockedIncrement(&This->refs);
 328 }
 329
 330 static ULONG WINAPI Widget_Release(
 331     IWidget *iface)
 332 {
 333     Widget *This = (Widget *)iface;
 334     ULONG refs = InterlockedDecrement(&This->refs);
 335     if (!refs)
 336     {
 337         IUnknown_Release(This->pDispatchUnknown);
 338         memset(This, 0xcc, sizeof(*This));
 339         HeapFree(GetProcessHeap(), 0, This);
 340         trace("Widget destroyed!\n");
 341     }
 342
 343     return refs;
 344 }
 345
 346 static HRESULT WINAPI Widget_GetTypeInfoCount(
 347     IWidget *iface,
 348     /* [out] */ UINT __RPC_FAR *pctinfo)
 349 {
 350     Widget *This = (Widget *)iface;
 351     IDispatch *pDispatch;
 352     HRESULT hr = IUnknown_QueryInterface(This->pDispatchUnknown, &IID_IDispatch, (void **)&pDispatch);
 353     if (SUCCEEDED(hr))
 354     {
 355         hr = IDispatch_GetTypeInfoCount(pDispatch, pctinfo);
 356         IDispatch_Release(pDispatch);
 357     }
 358     return hr;
 359 }
 360
 361 static HRESULT WINAPI Widget_GetTypeInfo(
 362     IWidget __RPC_FAR * iface,
 363     /* [in] */ UINT iTInfo,
 364     /* [in] */ LCID lcid,
 365     /* [out] */ ITypeInfo __RPC_FAR *__RPC_FAR *ppTInfo)
 366 {
 367     Widget *This = (Widget *)iface;
 368     IDispatch *pDispatch;
 369     HRESULT hr = IUnknown_QueryInterface(This->pDispatchUnknown, &IID_IDispatch, (void **)&pDispatch);
 370     if (SUCCEEDED(hr))
 371     {
 372         hr = IDispatch_GetTypeInfo(pDispatch, iTInfo, lcid, ppTInfo);
 373         IDispatch_Release(pDispatch);
 374     }
 375     return hr;
 376 }
 377
 378 static HRESULT WINAPI Widget_GetIDsOfNames(
 379     IWidget __RPC_FAR * iface,
 380     /* [in] */ REFIID riid,
 381     /* [size_is][in] */ LPOLESTR __RPC_FAR *rgszNames,
 382     /* [in] */ UINT cNames,
 383     /* [in] */ LCID lcid,
 384     /* [size_is][out] */ DISPID __RPC_FAR *rgDispId)
 385 {
 386     Widget *This = (Widget *)iface;
 387     IDispatch *pDispatch;
 388     HRESULT hr = IUnknown_QueryInterface(This->pDispatchUnknown, &IID_IDispatch, (void **)&pDispatch);
 389     if (SUCCEEDED(hr))
 390     {
 391         hr = IDispatch_GetIDsOfNames(pDispatch, riid, rgszNames, cNames, lcid, rgDispId);
 392         IDispatch_Release(pDispatch);
 393     }
 394     return hr;
 395 }
 396
 397 static HRESULT WINAPI Widget_Invoke(
 398     IWidget __RPC_FAR * iface,
 399     /* [in] */ DISPID dispIdMember,
 400     /* [in] */ REFIID riid,
 401     /* [in] */ LCID lcid,
 402     /* [in] */ WORD wFlags,
 403     /* [out][in] */ DISPPARAMS __RPC_FAR *pDispParams,
 404     /* [out] */ VARIANT __RPC_FAR *pVarResult,
 405     /* [out] */ EXCEPINFO __RPC_FAR *pExcepInfo,
 406     /* [out] */ UINT __RPC_FAR *puArgErr)
 407 {
 408     Widget *This = (Widget *)iface;
 409     IDispatch *pDispatch;
 410     HRESULT hr = IUnknown_QueryInterface(This->pDispatchUnknown, &IID_IDispatch, (void **)&pDispatch);
 411     if (SUCCEEDED(hr))
 412     {
 413         hr = IDispatch_Invoke(pDispatch, dispIdMember, riid, lcid, wFlags, pDispParams, pVarResult, pExcepInfo, puArgErr);
 414         IDispatch_Release(pDispatch);
 415     }
 416     return hr;
 417 }
 418
 419 static HRESULT WINAPI Widget_put_Name(
 420     IWidget __RPC_FAR * iface,
 421     /* [in] */ BSTR name)
 422 {
 423     trace("put_Name(%s)\n", wine_dbgstr_w(name));
 424     return S_OK;
 425 }
 426
 427 static HRESULT WINAPI Widget_get_Name(
 428     IWidget __RPC_FAR * iface,
 429     /* [out] */ BSTR __RPC_FAR *name)
 430 {
 431     static const WCHAR szCat[] = { 'C','a','t',0 };
 432     trace("get_Name()\n");
 433     *name = SysAllocString(szCat);
 434     return S_OK;
 435 }
 436
 437 static HRESULT WINAPI Widget_DoSomething(
 438     IWidget __RPC_FAR * iface,
 439     /* [in] */ double number,
 440     /* [out] */ BSTR *str1,
 441     /* [defaultvalue][in] */ BSTR str2,
 442     /* [optional][in] */ VARIANT __RPC_FAR *opt)
 443 {
 444     static const WCHAR szString[] = { 'S','t','r','i','n','g',0 };
 445     trace("DoSomething()\n");
 446
 447     ok(number == 3.141, "number(%f) != 3.141\n", number);
 448     ok(*str2 == '\0', "str2(%s) != \"\"\n", wine_dbgstr_w(str2));
 449     ok(V_VT(opt) == VT_ERROR, "V_VT(opt) should be VT_ERROR instead of 0x%x\n", V_VT(opt));
 450     ok(V_ERROR(opt) == DISP_E_PARAMNOTFOUND, "V_ERROR(opt) should be DISP_E_PARAMNOTFOUND instead of 0x%08x\n", V_ERROR(opt));
 451     *str1 = SysAllocString(szString);
 452
 453     return S_FALSE;
 454 }
 455
 456 static HRESULT WINAPI Widget_get_State(
 457     IWidget __RPC_FAR * iface,
 458     /* [retval][out] */ STATE __RPC_FAR *state)
 459 {
 460     trace("get_State() = STATE_WIDGETIFIED\n");
 461     *state = STATE_WIDGETIFIED;
 462     return S_OK;
 463 }
 464
 465 static HRESULT WINAPI Widget_put_State(
 466     IWidget __RPC_FAR * iface,
 467     /* [in] */ STATE state)
 468 {
 469     trace("put_State(%d)\n", state);
 470     return S_OK;
 471 }
 472
 473 static HRESULT WINAPI Widget_Map(
 474     IWidget * iface,
 475     BSTR bstrId,
 476     BSTR *sValue)
 477 {
 478     trace("Map(%s, %p)\n", wine_dbgstr_w(bstrId), sValue);
 479     *sValue = SysAllocString(bstrId);
 480     return S_OK;
 481 }
 482
 483 static HRESULT WINAPI Widget_SetOleColor(
 484     IWidget * iface,
 485     OLE_COLOR val)
 486 {
 487     trace("SetOleColor(0x%x)\n", val);
 488     return S_OK;
 489 }
 490
 491 static HRESULT WINAPI Widget_GetOleColor(
 492     IWidget * iface,
 493     OLE_COLOR *pVal)
 494 {
 495     trace("GetOleColor() = 0x8000000f\n");
 496     *pVal = 0x8000000f;
 497     return S_FALSE;
 498 }
 499
 500 static HRESULT WINAPI Widget_Clone(
 501     IWidget *iface,
 502     IWidget **ppVal)
 503 {
 504     trace("Clone()\n");
 505     return Widget_QueryInterface(iface, &IID_IWidget, (void **)ppVal);
 506 }
 507
 508 static HRESULT WINAPI Widget_CloneDispatch(
 509     IWidget *iface,
 510     IDispatch **ppVal)
 511 {
 512     trace("CloneDispatch()\n");
 513     return Widget_QueryInterface(iface, &IID_IWidget, (void **)ppVal);
 514 }
 515
 516 static HRESULT WINAPI Widget_CloneCoclass(
 517     IWidget *iface,
 518     ApplicationObject2 **ppVal)
 519 {
 520     trace("CloneCoclass()\n");
 521     return S_OK;
 522 }
 523
 524 static HRESULT WINAPI Widget_Value(
 525     IWidget __RPC_FAR * iface,
 526     VARIANT *value,
 527     VARIANT *retval)
 528 {
 529     trace("Value(%p, %p)\n", value, retval);
 530     ok(V_VT(value) == VT_I2, "V_VT(value) was %d instead of VT_I2\n", V_VT(value));
 531     ok(V_I2(value) == 1, "V_I2(value) was %d instead of 1\n", V_I2(value));
 532     V_VT(retval) = VT_I2;
 533     V_I2(retval) = 1234;
 534     return S_OK;
 535 }
 536
 537 static HRESULT WINAPI Widget_Array(
 538     IWidget * iface,
 539     SAFEARRAY * values)
 540 {
 541     trace("Array(%p)\n", values);
 542     return S_OK;
 543 }
 544
 545 static HRESULT WINAPI Widget_VariantArrayPtr(
 546     IWidget * iface,
 547     SAFEARRAY ** values)
 548 {
 549     trace("VariantArrayPtr(%p)\n", values);
 550     return S_OK;
 551 }
 552
 553 static HRESULT WINAPI Widget_Variant(
 554     IWidget __RPC_FAR * iface,
 555     VARIANT var)
 556 {
 557     trace("Variant()\n");
 558     ok(V_VT(&var) == VT_CY, "V_VT(&var) was %d\n", V_VT(&var));
 559     ok(S(V_CY(&var)).Hi == 0xdababe, "V_CY(&var).Hi was 0x%x\n", S(V_CY(&var)).Hi);
 560     ok(S(V_CY(&var)).Lo == 0xdeadbeef, "V_CY(&var).Lo was 0x%x\n", S(V_CY(&var)).Lo);
 561     return S_OK;
 562 }
 563
 564 static HRESULT WINAPI Widget_VarArg(
 565     IWidget * iface,
 566     int numexpect,
 567     SAFEARRAY * values)
 568 {
 569     LONG lbound, ubound, i;
 570     VARIANT * data;
 571     HRESULT hr;
 572
 573     trace("VarArg(%p)\n", values);
 574
 575     hr = SafeArrayGetLBound(values, 1, &lbound);
 576     ok(hr == S_OK, "SafeArrayGetLBound failed with %x\n", hr);
 577     ok(lbound == 0, "SafeArrayGetLBound returned %d\n", lbound);
 578
 579     hr = SafeArrayGetUBound(values, 1, &ubound);
 580     ok(hr == S_OK, "SafeArrayGetUBound failed with %x\n", hr);
 581     ok(ubound == numexpect-1, "SafeArrayGetUBound returned %d, but expected %d\n", ubound, numexpect-1);
 582
 583     hr = SafeArrayAccessData(values, (LPVOID)&data);
 584     ok(hr == S_OK, "SafeArrayAccessData failed with %x\n", hr);
 585
 586     for (i=0; i<=ubound-lbound; i++)
 587     {
 588         ok(V_VT(&data[i]) == VT_I4, "V_VT(&data[%d]) was %d\n", i, V_VT(&data[i]));
 589         ok(V_I4(&data[i]) == i, "V_I4(&data[%d]) was %d\n", i, V_I4(&data[i]));
 590     }
 591
 592     hr = SafeArrayUnaccessData(values);
 593     ok(hr == S_OK, "SafeArrayUnaccessData failed with %x\n", hr);
 594
 595     return S_OK;
 596 }
 597
 598 static HRESULT WINAPI Widget_StructArgs(
 599     IWidget * iface,
 600     MYSTRUCT byval,
 601     MYSTRUCT *byptr,
 602     MYSTRUCT arr[5])
 603 {
 604     ok(memcmp(&byval, &MYSTRUCT_BYVAL, sizeof(MYSTRUCT))==0, "Struct parameter passed by value corrupted\n");
 605     ok(memcmp(byptr,  &MYSTRUCT_BYPTR, sizeof(MYSTRUCT))==0, "Struct parameter passed by pointer corrupted\n");
 606     ok(memcmp(arr,    MYSTRUCT_ARRAY,  sizeof(MYSTRUCT_ARRAY))==0, "Array of structs corrupted\n");
 607     return S_OK;
 608 }
 609
 610
 611 static HRESULT WINAPI Widget_Error(
 612     IWidget __RPC_FAR * iface)
 613 {
 614     trace("Error()\n");
 615     return E_NOTIMPL;
 616 }
 617
 618 static HRESULT WINAPI Widget_CloneInterface(
 619     IWidget __RPC_FAR * iface,
 620     ISomethingFromDispatch **ppVal)
 621 {
 622     trace("CloneInterface()\n");
 623     *ppVal = 0;
 624     return S_OK;
 625 }
 626
 627 static const struct IWidgetVtbl Widget_VTable =
 628 {
 629     Widget_QueryInterface,
 630     Widget_AddRef,
 631     Widget_Release,
 632     Widget_GetTypeInfoCount,
 633     Widget_GetTypeInfo,
 634     Widget_GetIDsOfNames,
 635     Widget_Invoke,
 636     Widget_put_Name,
 637     Widget_get_Name,
 638     Widget_DoSomething,
 639     Widget_get_State,
 640     Widget_put_State,
 641     Widget_Map,
 642     Widget_SetOleColor,
 643     Widget_GetOleColor,
 644     Widget_Clone,
 645     Widget_CloneDispatch,
 646     Widget_CloneCoclass,
 647     Widget_Value,
 648     Widget_Array,
 649     Widget_VariantArrayPtr,
 650     Widget_Variant,
 651     Widget_VarArg,
 652     Widget_StructArgs,
 653     Widget_Error,
 654     Widget_CloneInterface
 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     const IKindaEnumWidgetVtbl *lpVtbl;
 732     LONG refs;
 733 } KindaEnum;
 734
 735 static HRESULT register_current_module_typelib(void)
 736 {
 737     WCHAR path[MAX_PATH];
 738     CHAR pathA[MAX_PATH];
 739     HRESULT hr;
 740     ITypeLib *typelib;
 741
 742     GetModuleFileNameA(NULL, pathA, MAX_PATH);
 743     MultiByteToWideChar(CP_ACP, 0, pathA, -1, path, MAX_PATH);
 744
 745     hr = LoadTypeLib(path, &typelib);
 746     if (SUCCEEDED(hr))
 747     {
 748         hr = RegisterTypeLib(typelib, path, NULL);
 749         ITypeLib_Release(typelib);
 750     }
 751     return hr;
 752 }
 753
 754 static ITypeInfo *get_type_info(REFIID riid)
 755 {
 756     ITypeInfo *pTypeInfo;
 757     ITypeLib *pTypeLib;
 758     HRESULT hr;
 759
 760     hr = LoadRegTypeLib(&LIBID_TestTypelib, 1, 0, LOCALE_NEUTRAL, &pTypeLib);
 761     ok_ole_success(hr, LoadRegTypeLib);
 762     if (FAILED(hr))
 763         return NULL;
 764
 765     hr = ITypeLib_GetTypeInfoOfGuid(pTypeLib, riid, &pTypeInfo);
 766     ITypeLib_Release(pTypeLib);
 767     ok_ole_success(hr, ITypeLib_GetTypeInfoOfGuid);
 768     if (FAILED(hr))
 769         return NULL;
 770
 771     return pTypeInfo;
 772 }
 773
 774 static IWidget *Widget_Create(void)
 775 {
 776     Widget *This;
 777     ITypeInfo *pTypeInfo;
 778     HRESULT hr = E_FAIL;
 779
 780     pTypeInfo = get_type_info(&IID_IWidget);
 781     if(!pTypeInfo)
 782         return NULL;
 783
 784     This = HeapAlloc(GetProcessHeap(), 0, sizeof(*This));
 785     This->lpVtbl = &Widget_VTable;
 786     This->refs = 1;
 787     This->pDispatchUnknown = NULL;
 788
 789     hr = CreateStdDispatch((IUnknown *)&This->lpVtbl, This, pTypeInfo, &This->pDispatchUnknown);
 790     ok_ole_success(hr, CreateStdDispatch);
 791     ITypeInfo_Release(pTypeInfo);
 792
 793     if (SUCCEEDED(hr))
 794         return (IWidget *)&This->lpVtbl;
 795     else
 796     {
 797         HeapFree(GetProcessHeap(), 0, This);
 798         return NULL;
 799     }
 800 }
 801
 802 static HRESULT WINAPI KindaEnum_QueryInterface(
 803     IKindaEnumWidget *iface,
 804     /* [in] */ REFIID riid,
 805     /* [iid_is][out] */ void __RPC_FAR *__RPC_FAR *ppvObject)
 806 {
 807     if (IsEqualIID(riid, &IID_IKindaEnumWidget) || IsEqualIID(riid, &IID_IUnknown))
 808     {
 809         IKindaEnumWidget_AddRef(iface);
 810         *ppvObject = iface;
 811         return S_OK;
 812     }
 813     else
 814     {
 815         *ppvObject = NULL;
 816         return E_NOINTERFACE;
 817     }
 818 }
 819
 820 static ULONG WINAPI KindaEnum_AddRef(
 821     IKindaEnumWidget *iface)
 822 {
 823     KindaEnum *This = (KindaEnum *)iface;
 824
 825     return InterlockedIncrement(&This->refs);
 826 }
 827
 828 static ULONG WINAPI KindaEnum_Release(
 829     IKindaEnumWidget *iface)
 830 {
 831     KindaEnum *This = (KindaEnum *)iface;
 832     ULONG refs = InterlockedDecrement(&This->refs);
 833     if (!refs)
 834     {
 835         memset(This, 0xcc, sizeof(*This));
 836         HeapFree(GetProcessHeap(), 0, This);
 837         trace("KindaEnumWidget destroyed!\n");
 838     }
 839
 840     return refs;
 841 }
 842
 843 static HRESULT WINAPI KindaEnum_Next(
 844     IKindaEnumWidget *iface,
 845     /* [out] */ IWidget __RPC_FAR *__RPC_FAR *widget)
 846 {
 847     *widget = Widget_Create();
 848     if (*widget)
 849         return S_OK;
 850     else
 851         return E_OUTOFMEMORY;
 852 }
 853
 854 static HRESULT WINAPI KindaEnum_Count(
 855     IKindaEnumWidget *iface,
 856     /* [out] */ ULONG __RPC_FAR *count)
 857 {
 858     return E_NOTIMPL;
 859 }
 860
 861 static HRESULT WINAPI KindaEnum_Reset(
 862     IKindaEnumWidget *iface)
 863 {
 864     return E_NOTIMPL;
 865 }
 866
 867 static HRESULT WINAPI KindaEnum_Clone(
 868     IKindaEnumWidget *iface,
 869     /* [out] */ IKindaEnumWidget __RPC_FAR *__RPC_FAR *ppenum)
 870 {
 871     return E_NOTIMPL;
 872 }
 873
 874 static const IKindaEnumWidgetVtbl KindaEnumWidget_VTable =
 875 {
 876     KindaEnum_QueryInterface,
 877     KindaEnum_AddRef,
 878     KindaEnum_Release,
 879     KindaEnum_Next,
 880     KindaEnum_Count,
 881     KindaEnum_Reset,
 882     KindaEnum_Clone
 883 };
 884
 885 static IKindaEnumWidget *KindaEnumWidget_Create(void)
 886 {
 887     KindaEnum *This;
 888
 889     This = HeapAlloc(GetProcessHeap(), 0, sizeof(*This));
 890     if (!This) return NULL;
 891     This->lpVtbl = &KindaEnumWidget_VTable;
 892     This->refs = 1;
 893     return (IKindaEnumWidget *)This;
 894 }
 895
 896 static HRESULT WINAPI NonOleAutomation_QueryInterface(INonOleAutomation *iface, REFIID riid, void **ppv)
 897 {
 898     if (IsEqualIID(riid, &IID_IUnknown) || IsEqualIID(riid, &IID_INonOleAutomation))
 899     {
 900         *(INonOleAutomation **)ppv = iface;
 901         return S_OK;
 902     }
 903     *ppv = NULL;
 904     return E_NOINTERFACE;
 905 }
 906
 907 static ULONG WINAPI NonOleAutomation_AddRef(INonOleAutomation *iface)
 908 {
 909     return 2;
 910 }
 911
 912 static ULONG WINAPI NonOleAutomation_Release(INonOleAutomation *iface)
 913 {
 914     return 1;
 915 }
 916
 917 static BSTR WINAPI NonOleAutomation_BstrRet(INonOleAutomation *iface)
 918 {
 919     static const WCHAR wszTestString[] = {'T','h','i','s',' ','i','s',' ','a',' ','t','e','s','t',' ','s','t','r','i','n','g',0};
 920     return SysAllocString(wszTestString);
 921 }
 922
 923 static INonOleAutomationVtbl NonOleAutomation_VTable =
 924 {
 925     NonOleAutomation_QueryInterface,
 926     NonOleAutomation_AddRef,
 927     NonOleAutomation_Release,
 928     NonOleAutomation_BstrRet,
 929 };
 930
 931 static INonOleAutomation NonOleAutomation = { &NonOleAutomation_VTable };
 932
 933 static ITypeInfo *NonOleAutomation_GetTypeInfo(void)
 934 {
 935     ITypeLib *pTypeLib;
 936     HRESULT hr = LoadRegTypeLib(&LIBID_TestTypelib, 1, 0, LOCALE_NEUTRAL, &pTypeLib);
 937     ok_ole_success(hr, LoadRegTypeLib);
 938     if (SUCCEEDED(hr))
 939     {
 940         ITypeInfo *pTypeInfo;
 941         hr = ITypeLib_GetTypeInfoOfGuid(pTypeLib, &IID_INonOleAutomation, &pTypeInfo);
 942         ok_ole_success(hr, ITypeLib_GetTypeInfoOfGuid);
 943         return pTypeInfo;
 944     }
 945     return NULL;
 946 }
 947
 948 static void test_typelibmarshal(void)
 949 {
 950     static const WCHAR szCat[] = { 'C','a','t',0 };
 951     static const WCHAR szTestTest[] = { 'T','e','s','t','T','e','s','t',0 };
 952     static WCHAR szSuperman[] = { 'S','u','p','e','r','m','a','n',0 };
 953     HRESULT hr;
 954     IKindaEnumWidget *pKEW = KindaEnumWidget_Create();
 955     IWidget *pWidget;
 956     IStream *pStream;
 957     IDispatch *pDispatch;
 958     static const LARGE_INTEGER ullZero;
 959     EXCEPINFO excepinfo;
 960     VARIANT varresult;
 961     DISPID dispidNamed = DISPID_PROPERTYPUT;
 962     DISPPARAMS dispparams;
 963     VARIANTARG vararg[4];
 964     STATE the_state;
 965     HANDLE thread;
 966     DWORD tid;
 967     BSTR bstr;
 968     ITypeInfo *pTypeInfo;
 969     MYSTRUCT mystruct;
 970     MYSTRUCT mystructArray[5];
 971
 972     ok(pKEW != NULL, "Widget creation failed\n");
 973
 974     hr = CreateStreamOnHGlobal(NULL, TRUE, &pStream);
 975     ok_ole_success(hr, CreateStreamOnHGlobal);
 976     tid = start_host_object(pStream, &IID_IKindaEnumWidget, (IUnknown *)pKEW, MSHLFLAGS_NORMAL, &thread);
 977     IKindaEnumWidget_Release(pKEW);
 978
 979     IStream_Seek(pStream, ullZero, STREAM_SEEK_SET, NULL);
 980     hr = CoUnmarshalInterface(pStream, &IID_IKindaEnumWidget, (void **)&pKEW);
 981     ok_ole_success(hr, CoUnmarshalInterface);
 982     IStream_Release(pStream);
 983
 984     hr = IKindaEnumWidget_Next(pKEW, &pWidget);
 985     ok_ole_success(hr, IKindaEnumWidget_Next);
 986
 987     IKindaEnumWidget_Release(pKEW);
 988
 989     hr = IWidget_QueryInterface(pWidget, &IID_IDispatch, (void **)&pDispatch);
 990
 991     /* call put_Name */
 992     VariantInit(&vararg[0]);
 993     dispparams.cNamedArgs = 1;
 994     dispparams.rgdispidNamedArgs = &dispidNamed;
 995     dispparams.cArgs = 1;
 996     dispparams.rgvarg = vararg;
 997     VariantInit(&varresult);
 998     hr = IDispatch_Invoke(pDispatch, DISPID_TM_NAME, &IID_NULL, LOCALE_NEUTRAL, DISPATCH_PROPERTYPUT, &dispparams, &varresult, &excepinfo, NULL);
 999     ok_ole_success(hr, IDispatch_Invoke);
1000     ok(excepinfo.wCode == 0x0 && excepinfo.scode == S_OK,
1001         "EXCEPINFO differs from expected: wCode = 0x%x, scode = 0x%08x\n",
1002         excepinfo.wCode, excepinfo.scode);
1003     VariantClear(&varresult);
1004
1005     /* call put_Name (direct) */
1006     bstr = SysAllocString(szSuperman);
1007     hr = IWidget_put_Name(pWidget, bstr);
1008     ok_ole_success(hr, IWidget_put_Name);
1009     SysFreeString(bstr);
1010
1011     /* call get_Name */
1012     dispparams.cNamedArgs = 0;
1013     dispparams.rgdispidNamedArgs = NULL;
1014     dispparams.cArgs = 0;
1015     dispparams.rgvarg = NULL;
1016     VariantInit(&varresult);
1017     hr = IDispatch_Invoke(pDispatch, DISPID_TM_NAME, &IID_NULL, LOCALE_NEUTRAL, DISPATCH_PROPERTYGET, &dispparams, &varresult, &excepinfo, NULL);
1018     ok_ole_success(hr, IDispatch_Invoke);
1019     ok(excepinfo.wCode == 0x0 && excepinfo.scode == S_OK,
1020         "EXCEPINFO differs from expected: wCode = 0x%x, scode = 0x%08x\n",
1021         excepinfo.wCode, excepinfo.scode);
1022     trace("Name = %s\n", wine_dbgstr_w(V_BSTR(&varresult)));
1023     VariantClear(&varresult);
1024
1025     /* call get_Name (direct) */
1026     bstr = NULL;
1027     hr = IWidget_get_Name(pWidget, &bstr);
1028     ok_ole_success(hr, IWidget_get_Name);
1029     ok(!lstrcmpW(bstr, szCat), "IWidget_get_Name should have returned string \"Cat\" instead of %s\n", wine_dbgstr_w(bstr));
1030     SysFreeString(bstr);
1031
1032     /* call DoSomething */
1033     VariantInit(&vararg[0]);
1034     VariantInit(&vararg[1]);
1035     V_VT(&vararg[1]) = VT_R8;
1036     V_R8(&vararg[1]) = 3.141;
1037     dispparams.cNamedArgs = 0;
1038     dispparams.cArgs = 2;
1039     dispparams.rgdispidNamedArgs = NULL;
1040     dispparams.rgvarg = vararg;
1041     VariantInit(&varresult);
1042     hr = IDispatch_Invoke(pDispatch, DISPID_TM_DOSOMETHING, &IID_NULL, LOCALE_NEUTRAL, DISPATCH_METHOD, &dispparams, &varresult, &excepinfo, NULL);
1043     ok_ole_success(hr, IDispatch_Invoke);
1044     ok(V_VT(&varresult) == VT_EMPTY, "varresult should be VT_EMPTY\n");
1045     VariantClear(&varresult);
1046
1047     /* call get_State */
1048     dispparams.cNamedArgs = 0;
1049     dispparams.cArgs = 0;
1050     dispparams.rgdispidNamedArgs = NULL;
1051     dispparams.rgvarg = NULL;
1052     hr = IDispatch_Invoke(pDispatch, DISPID_TM_STATE, &IID_NULL, LOCALE_NEUTRAL, DISPATCH_PROPERTYGET, &dispparams, &varresult, &excepinfo, NULL);
1053     ok_ole_success(hr, IDispatch_Invoke);
1054     ok((V_VT(&varresult) == VT_I4) && (V_I4(&varresult) == STATE_WIDGETIFIED), "Return val mismatch\n");
1055
1056     /* call get_State (direct) */
1057     hr = IWidget_get_State(pWidget, &the_state);
1058     ok_ole_success(hr, IWidget_get_state);
1059     ok(the_state == STATE_WIDGETIFIED, "should have returned WIDGET_WIDGETIFIED instead of %d\n", the_state);
1060
1061     /* call put_State */
1062     the_state = STATE_WIDGETIFIED;
1063     VariantInit(&vararg[0]);
1064     V_VT(&vararg[0]) = VT_BYREF|VT_I4;
1065     V_I4REF(&vararg[0]) = (int *)&the_state;
1066     dispparams.cNamedArgs = 1;
1067     dispparams.cArgs = 1;
1068     dispparams.rgdispidNamedArgs = &dispidNamed;
1069     dispparams.rgvarg = vararg;
1070     hr = IDispatch_Invoke(pDispatch, DISPID_TM_STATE, &IID_NULL, LOCALE_NEUTRAL, DISPATCH_PROPERTYPUT, &dispparams, &varresult, &excepinfo, NULL);
1071     ok_ole_success(hr, IDispatch_Invoke);
1072
1073     /* call Map */
1074     bstr = SysAllocString(szTestTest);
1075     VariantInit(&vararg[0]);
1076     V_VT(&vararg[0]) = VT_BYREF|VT_BSTR;
1077     V_BSTRREF(&vararg[0]) = &bstr;
1078     dispparams.cNamedArgs = 0;
1079     dispparams.cArgs = 1;
1080     dispparams.rgdispidNamedArgs = NULL;
1081     dispparams.rgvarg = vararg;
1082     VariantInit(&varresult);
1083     hr = IDispatch_Invoke(pDispatch, DISPID_TM_MAP, &IID_NULL, LOCALE_NEUTRAL, DISPATCH_METHOD, &dispparams, &varresult, &excepinfo, NULL);
1084     ok_ole_success(hr, IDispatch_Invoke);
1085     ok(V_VT(&varresult) == VT_BSTR, "Return value should be of type BSTR instead of %d\n", V_VT(&varresult));
1086     ok(!lstrcmpW(V_BSTR(&varresult), szTestTest), "Return value should have been \"TestTest\" instead of %s\n", wine_dbgstr_w(V_BSTR(&varresult)));
1087     VariantClear(&varresult);
1088
1089     /* call SetOleColor with large negative VT_I4 param */
1090     VariantInit(&vararg[0]);
1091     V_VT(&vararg[0]) = VT_I4;
1092     V_I4(&vararg[0]) = 0x80000005;
1093     dispparams.cNamedArgs = 0;
1094     dispparams.cArgs = 1;
1095     dispparams.rgdispidNamedArgs = NULL;
1096     dispparams.rgvarg = vararg;
1097     hr = IDispatch_Invoke(pDispatch, DISPID_TM_SETOLECOLOR, &IID_NULL, LOCALE_NEUTRAL, DISPATCH_METHOD, &dispparams, NULL, &excepinfo, NULL);
1098     ok_ole_success(hr, IDispatch_Invoke);
1099
1100     /* call GetOleColor */
1101     dispparams.cNamedArgs = 0;
1102     dispparams.cArgs = 0;
1103     dispparams.rgdispidNamedArgs = NULL;
1104     dispparams.rgvarg = NULL;
1105     VariantInit(&varresult);
1106     hr = IDispatch_Invoke(pDispatch, DISPID_TM_GETOLECOLOR, &IID_NULL, LOCALE_NEUTRAL, DISPATCH_METHOD, &dispparams, &varresult, &excepinfo, NULL);
1107     ok_ole_success(hr, IDispatch_Invoke);
1108     VariantClear(&varresult);
1109
1110     /* call StructArgs (direct) */
1111     mystruct = MYSTRUCT_BYPTR;
1112     memcpy(mystructArray, MYSTRUCT_ARRAY, sizeof(mystructArray));
1113     hr = IWidget_StructArgs(pWidget, MYSTRUCT_BYVAL, &mystruct, mystructArray);
1114     todo_wine {
1115     ok_ole_success(hr, IWidget_StructArgs);
1116     }
1117
1118     /* call Clone */
1119     dispparams.cNamedArgs = 0;
1120     dispparams.cArgs = 0;
1121     dispparams.rgdispidNamedArgs = NULL;
1122     dispparams.rgvarg = NULL;
1123     VariantInit(&varresult);
1124     hr = IDispatch_Invoke(pDispatch, DISPID_TM_CLONE, &IID_NULL, LOCALE_NEUTRAL, DISPATCH_PROPERTYGET, &dispparams, &varresult, &excepinfo, NULL);
1125     ok_ole_success(hr, IDispatch_Invoke);
1126     ok(V_VT(&varresult) == VT_DISPATCH, "vt %x\n", V_VT(&varresult));
1127     VariantClear(&varresult);
1128
1129     /* call CloneInterface */
1130     dispparams.cNamedArgs = 0;
1131     dispparams.cArgs = 0;
1132     dispparams.rgdispidNamedArgs = NULL;
1133     dispparams.rgvarg = NULL;
1134     VariantInit(&varresult);
1135     hr = IDispatch_Invoke(pDispatch, DISPID_TM_CLONEINTERFACE, &IID_NULL, LOCALE_NEUTRAL, DISPATCH_PROPERTYGET, &dispparams, &varresult, &excepinfo, NULL);
1136     ok_ole_success(hr, IDispatch_Invoke);
1137     ok(V_VT(&varresult) == VT_DISPATCH, "vt %x\n", V_VT(&varresult));
1138     VariantClear(&varresult);
1139
1140     /* call CloneDispatch with automatic value getting */
1141     V_VT(&vararg[0]) = VT_I2;
1142     V_I2(&vararg[0]) = 1;
1143     dispparams.cNamedArgs = 0;
1144     dispparams.rgdispidNamedArgs = NULL;
1145     dispparams.cArgs = 1;
1146     dispparams.rgvarg = vararg;
1147     VariantInit(&varresult);
1148     hr = IDispatch_Invoke(pDispatch, DISPID_TM_CLONEDISPATCH, &IID_NULL, LOCALE_NEUTRAL, DISPATCH_PROPERTYGET, &dispparams, &varresult, &excepinfo, NULL);
1149     ok_ole_success(hr, IDispatch_Invoke);
1150
1151     ok(excepinfo.wCode == 0x0 && excepinfo.scode == S_OK,
1152         "EXCEPINFO differs from expected: wCode = 0x%x, scode = 0x%08x\n",
1153         excepinfo.wCode, excepinfo.scode);
1154
1155     ok(V_VT(&varresult) == VT_I2, "V_VT(&varresult) was %d instead of VT_I2\n", V_VT(&varresult));
1156     ok(V_I2(&varresult) == 1234, "V_I2(&varresult) was %d instead of 1234\n", V_I2(&varresult));
1157     VariantClear(&varresult);
1158
1159     /* call CloneCoclass */
1160     dispparams.cNamedArgs = 0;
1161     dispparams.cArgs = 0;
1162     dispparams.rgdispidNamedArgs = NULL;
1163     dispparams.rgvarg = NULL;
1164     VariantInit(&varresult);
1165     hr = IDispatch_Invoke(pDispatch, DISPID_TM_CLONECOCLASS, &IID_NULL, LOCALE_NEUTRAL, DISPATCH_PROPERTYGET, &dispparams, &varresult, &excepinfo, NULL);
1166     ok_ole_success(hr, IDispatch_Invoke);
1167
1168     ok(excepinfo.wCode == 0x0 && excepinfo.scode == S_OK,
1169        "EXCEPINFO differs from expected: wCode = 0x%x, scode = 0x%08x\n",
1170        excepinfo.wCode, excepinfo.scode);
1171
1172     ok(V_VT(&varresult) == VT_DISPATCH, "V_VT(&varresult) was %d instead of VT_DISPATCH\n", V_VT(&varresult));
1173     ok(!V_DISPATCH(&varresult), "V_DISPATCH(&varresult) should be NULL instead of %p\n", V_DISPATCH(&varresult));
1174     VariantClear(&varresult);
1175
1176     /* call Value with a VT_VARIANT|VT_BYREF type */
1177     V_VT(&vararg[0]) = VT_VARIANT|VT_BYREF;
1178     V_VARIANTREF(&vararg[0]) = &vararg[1];
1179     V_VT(&vararg[1]) = VT_I2;
1180     V_I2(&vararg[1]) = 1;
1181     dispparams.cNamedArgs = 0;
1182     dispparams.rgdispidNamedArgs = NULL;
1183     dispparams.cArgs = 1;
1184     dispparams.rgvarg = vararg;
1185     VariantInit(&varresult);
1186     hr = IDispatch_Invoke(pDispatch, DISPID_VALUE, &IID_NULL, LOCALE_NEUTRAL, DISPATCH_PROPERTYGET, &dispparams, &varresult, &excepinfo, NULL);
1187     ok_ole_success(hr, IDispatch_Invoke);
1188
1189     ok(excepinfo.wCode == 0x0 && excepinfo.scode == S_OK,
1190         "EXCEPINFO differs from expected: wCode = 0x%x, scode = 0x%08x\n",
1191         excepinfo.wCode, excepinfo.scode);
1192
1193     ok(V_VT(&varresult) == VT_I2, "V_VT(&varresult) was %d instead of VT_I2\n", V_VT(&varresult));
1194     ok(V_I2(&varresult) == 1234, "V_I2(&varresult) was %d instead of 1234\n", V_I2(&varresult));
1195     VariantClear(&varresult);
1196
1197     /* call Variant - exercises variant copying in ITypeInfo::Invoke and
1198      * handling of void return types */
1199     /* use a big type to ensure that the variant was properly copied into the
1200      * destination function's args */
1201     V_VT(&vararg[0]) = VT_CY;
1202     S(V_CY(&vararg[0])).Hi = 0xdababe;
1203     S(V_CY(&vararg[0])).Lo = 0xdeadbeef;
1204     dispparams.cNamedArgs = 0;
1205     dispparams.cArgs = 1;
1206     dispparams.rgdispidNamedArgs = NULL;
1207     dispparams.rgvarg = vararg;
1208     VariantInit(&varresult);
1209     hr = IDispatch_Invoke(pDispatch, DISPID_TM_VARIANT, &IID_NULL, LOCALE_NEUTRAL, DISPATCH_METHOD, &dispparams, NULL, NULL, NULL);
1210     ok_ole_success(hr, IDispatch_Invoke);
1211     VariantClear(&varresult);
1212
1213     /* call VarArg */
1214     VariantInit(&vararg[3]);
1215     V_VT(&vararg[3]) = VT_I4;
1216     V_I4(&vararg[3]) = 3;
1217     VariantInit(&vararg[2]);
1218     V_VT(&vararg[2]) = VT_I4;
1219     V_I4(&vararg[2]) = 0;
1220     VariantInit(&vararg[1]);
1221     V_VT(&vararg[1]) = VT_I4;
1222     V_I4(&vararg[1]) = 1;
1223     VariantInit(&vararg[0]);
1224     V_VT(&vararg[0]) = VT_I4;
1225     V_I4(&vararg[0]) = 2;
1226     dispparams.cNamedArgs = 0;
1227     dispparams.cArgs = 4;
1228     dispparams.rgdispidNamedArgs = NULL;
1229     dispparams.rgvarg = vararg;
1230     hr = IDispatch_Invoke(pDispatch, DISPID_TM_VARARG, &IID_NULL, LOCALE_NEUTRAL, DISPATCH_METHOD, &dispparams, NULL, NULL, NULL);
1231     ok_ole_success(hr, ITypeInfo_Invoke);
1232
1233     /* call VarArg, even one (non-optional, non-safearray) named argument is not allowed */
1234     dispidNamed = 0;
1235     dispparams.cNamedArgs = 1;
1236     dispparams.rgdispidNamedArgs = &dispidNamed;
1237     hr = IDispatch_Invoke(pDispatch, DISPID_TM_VARARG, &IID_NULL, LOCALE_NEUTRAL, DISPATCH_METHOD, &dispparams, NULL, NULL, NULL);
1238     ok(hr == DISP_E_NONAMEDARGS, "IDispatch_Invoke should have returned DISP_E_NONAMEDARGS instead of 0x%08x\n", hr);
1239     dispidNamed = DISPID_PROPERTYPUT;
1240
1241     /* call Error */
1242     dispparams.cNamedArgs = 0;
1243     dispparams.cArgs = 0;
1244     dispparams.rgdispidNamedArgs = NULL;
1245     dispparams.rgvarg = NULL;
1246     VariantInit(&varresult);
1247     hr = IDispatch_Invoke(pDispatch, DISPID_TM_ERROR, &IID_NULL, LOCALE_NEUTRAL, DISPATCH_METHOD, &dispparams, NULL, &excepinfo, NULL);
1248     ok(hr == DISP_E_EXCEPTION, "IDispatch_Invoke should have returned DISP_E_EXCEPTION instead of 0x%08x\n", hr);
1249     ok(excepinfo.wCode == 0x0 && excepinfo.scode == E_NOTIMPL,
1250         "EXCEPINFO differs from expected: wCode = 0x%x, scode = 0x%08x\n",
1251         excepinfo.wCode, excepinfo.scode);
1252     VariantClear(&varresult);
1253
1254     /* call BstrRet */
1255     pTypeInfo = NonOleAutomation_GetTypeInfo();
1256     dispparams.cNamedArgs = 0;
1257     dispparams.cArgs = 0;
1258     dispparams.rgdispidNamedArgs = NULL;
1259     dispparams.rgvarg = NULL;
1260     VariantInit(&varresult);
1261     hr = ITypeInfo_Invoke(pTypeInfo, &NonOleAutomation, DISPID_NOA_BSTRRET, DISPATCH_METHOD, &dispparams, &varresult, &excepinfo, NULL);
1262     ok_ole_success(hr, ITypeInfo_Invoke);
1263     ok(V_VT(&varresult) == VT_BSTR, "V_VT(&varresult) should be VT_BSTR instead of %d\n", V_VT(&varresult));
1264     ok(V_BSTR(&varresult) != NULL, "V_BSTR(&varresult) should not be NULL\n");
1265
1266     VariantClear(&varresult);
1267     ITypeInfo_Release(pTypeInfo);
1268
1269     /* tests call put_Name without named arg */
1270     VariantInit(&vararg[0]);
1271     dispparams.cNamedArgs = 0;
1272     dispparams.rgdispidNamedArgs = NULL;
1273     dispparams.cArgs = 1;
1274     dispparams.rgvarg = vararg;
1275     VariantInit(&varresult);
1276     hr = IDispatch_Invoke(pDispatch, DISPID_TM_NAME, &IID_NULL, LOCALE_NEUTRAL, DISPATCH_PROPERTYPUT, &dispparams, &varresult, &excepinfo, NULL);
1277     ok(hr == DISP_E_PARAMNOTFOUND, "IDispatch_Invoke should have returned DISP_E_PARAMNOTFOUND instead of 0x%08x\n", hr);
1278     VariantClear(&varresult);
1279
1280     /* tests param type that cannot be coerced */
1281     VariantInit(&vararg[0]);
1282     V_VT(&vararg[0]) = VT_UNKNOWN;
1283     V_UNKNOWN(&vararg[0]) = NULL;
1284     dispparams.cNamedArgs = 1;
1285     dispparams.rgdispidNamedArgs = &dispidNamed;
1286     dispparams.cArgs = 1;
1287     dispparams.rgvarg = vararg;
1288     VariantInit(&varresult);
1289 #if 0 /* NULL unknown not currently marshaled correctly */
1290     hr = IDispatch_Invoke(pDispatch, DISPID_TM_NAME, &IID_NULL, LOCALE_NEUTRAL, DISPATCH_PROPERTYPUT, &dispparams, &varresult, &excepinfo, NULL);
1291     ok(hr == DISP_E_TYPEMISMATCH, "IDispatch_Invoke should have returned DISP_E_TYPEMISMATCH instead of 0x%08x\n", hr);
1292 #endif
1293     VariantClear(&varresult);
1294
1295     /* tests bad param type */
1296     VariantInit(&vararg[0]);
1297     V_VT(&vararg[0]) = VT_CLSID;
1298     V_BYREF(&vararg[0]) = NULL;
1299     dispparams.cNamedArgs = 1;
1300     dispparams.rgdispidNamedArgs = &dispidNamed;
1301     dispparams.cArgs = 1;
1302     dispparams.rgvarg = vararg;
1303     VariantInit(&varresult);
1304     hr = IDispatch_Invoke(pDispatch, DISPID_TM_NAME, &IID_NULL, LOCALE_NEUTRAL, DISPATCH_PROPERTYPUT, &dispparams, &varresult, &excepinfo, NULL);
1305     ok(hr == DISP_E_BADVARTYPE, "IDispatch_Invoke should have returned DISP_E_BADVARTYPE instead of 0x%08x\n", hr);
1306     VariantClear(&varresult);
1307
1308     /* tests too small param count */
1309     dispparams.cNamedArgs = 0;
1310     dispparams.rgdispidNamedArgs = NULL;
1311     dispparams.cArgs = 0;
1312     dispparams.rgvarg = NULL;
1313     VariantInit(&varresult);
1314     hr = IDispatch_Invoke(pDispatch, DISPID_TM_DOSOMETHING, &IID_NULL, LOCALE_NEUTRAL, DISPATCH_METHOD, &dispparams, &varresult, &excepinfo, NULL);
1315     ok(hr == DISP_E_BADPARAMCOUNT, "IDispatch_Invoke should have returned DISP_E_BADPARAMCOUNT instead of 0x%08x\n", hr);
1316     VariantClear(&varresult);
1317
1318     /* tests propget function with large param count */
1319     VariantInit(&vararg[0]);
1320     V_VT(&vararg[0]) = VT_BSTR;
1321     V_BSTR(&vararg[0]) = NULL;
1322     V_VT(&vararg[1]) = VT_I4;
1323     V_I4(&vararg[1]) = 1;
1324     dispparams.cNamedArgs = 0;
1325     dispparams.cArgs = 2;
1326     dispparams.rgdispidNamedArgs = NULL;
1327     dispparams.rgvarg = vararg;
1328     hr = IDispatch_Invoke(pDispatch, DISPID_TM_STATE, &IID_NULL, LOCALE_NEUTRAL, DISPATCH_PROPERTYGET, &dispparams, &varresult, &excepinfo, NULL);
1329     ok(hr == DISP_E_NOTACOLLECTION, "IDispatch_Invoke should have returned DISP_E_NOTACOLLECTION instead of 0x%08x\n", hr);
1330
1331     IDispatch_Release(pDispatch);
1332     IWidget_Release(pWidget);
1333
1334     trace("calling end_host_object\n");
1335     end_host_object(tid, thread);
1336 }
1337
1338 static void test_DispCallFunc(void)
1339 {
1340     static const WCHAR szEmpty[] = { 0 };
1341     VARTYPE rgvt[] = { VT_R8, VT_BSTR, VT_BSTR, VT_VARIANT|VT_BYREF };
1342     VARIANTARG vararg[4];
1343     VARIANTARG varref;
1344     VARIANTARG *rgpvarg[4] = { &vararg[0], &vararg[1], &vararg[2], &vararg[3] };
1345     VARIANTARG varresult;
1346     HRESULT hr;
1347     IWidget *pWidget = Widget_Create();
1348     V_VT(&vararg[0]) = VT_R8;
1349     V_R8(&vararg[0]) = 3.141;
1350     V_VT(&vararg[1]) = VT_BSTR;
1351     V_BSTR(&vararg[1]) = SysAllocString(szEmpty);
1352     V_VT(&vararg[2]) = VT_BSTR;
1353     V_BSTR(&vararg[2]) = SysAllocString(szEmpty);
1354     V_VT(&vararg[3]) = VT_VARIANT|VT_BYREF;
1355     V_VARIANTREF(&vararg[3]) = &varref;
1356     V_VT(&varref) = VT_ERROR;
1357     V_ERROR(&varref) = DISP_E_PARAMNOTFOUND;
1358     VariantInit(&varresult);
1359     hr = DispCallFunc(pWidget, 9*sizeof(void*), CC_STDCALL, VT_UI4, 4, rgvt, rgpvarg, &varresult);
1360     ok_ole_success(hr, DispCallFunc);
1361     VariantClear(&varresult);
1362     VariantClear(&vararg[1]);
1363     VariantClear(&vararg[2]);
1364     IWidget_Release(pWidget);
1365 }
1366
1367 static void test_StaticWidget(void)
1368 {
1369     ITypeInfo *type_info;
1370     DISPPARAMS dispparams;
1371     VARIANTARG vararg[4];
1372     EXCEPINFO excepinfo;
1373     VARIANT varresult;
1374     HRESULT hr;
1375
1376     type_info = get_type_info(&IID_IStaticWidget);
1377
1378     /* call TestDual */
1379     dispparams.cNamedArgs = 0;
1380     dispparams.cArgs = 1;
1381     dispparams.rgdispidNamedArgs = NULL;
1382     dispparams.rgvarg = vararg;
1383     V_VT(vararg) = VT_DISPATCH;
1384     V_DISPATCH(vararg) = (IDispatch*)&TestDualDisp;
1385     VariantInit(&varresult);
1386     hr = ITypeInfo_Invoke(type_info, &StaticWidget, DISPID_TM_TESTDUAL, DISPATCH_METHOD,
1387             &dispparams, &varresult, &excepinfo, NULL);
1388     ok_ole_success(hr, IDispatch_Invoke);
1389     ok(V_VT(&varresult) == VT_EMPTY, "vt %x\n", V_VT(&varresult));
1390     VariantClear(&varresult);
1391
1392     ITypeInfo_Release(type_info);
1393 }
1394
1395 START_TEST(tmarshal)
1396 {
1397     HRESULT hr;
1398
1399     CoInitializeEx(NULL, COINIT_APARTMENTTHREADED);
1400
1401     hr = register_current_module_typelib();
1402     ok_ole_success(hr, register_current_module_typelib);
1403
1404     test_typelibmarshal();
1405     test_DispCallFunc();
1406     test_StaticWidget();
1407
1408     hr = UnRegisterTypeLib(&LIBID_TestTypelib, 1, 0, LOCALE_NEUTRAL,
1409                            sizeof(void*) == 8 ? SYS_WIN64 : SYS_WIN32);
1410     ok_ole_success(hr, UnRegisterTypeLib);
1411
1412     CoUninitialize();
1413 }