dlls/rpcrt4/tests/cstub.c

   1 /*
   2  * Unit test suite for cstubs
   3  *
   4  * Copyright 2006 Huw Davies
   5  *
   6  * This library is free software; you can redistribute it and/or
   7  * modify it under the terms of the GNU Lesser General Public
   8  * License as published by the Free Software Foundation; either
   9  * version 2.1 of the License, or (at your option) any later version.
  10  *
  11  * This library is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14  * Lesser General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU Lesser General Public
  17  * License along with this library; if not, write to the Free Software
  18  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
  19  */
  20
  21 #include <stdarg.h>
  22
  23 #define PROXY_DELEGATION
  24 #define COBJMACROS
  25
  26 #include "wine/test.h"
  27 #include <windef.h>
  28 #include <winbase.h>
  29 #include <winnt.h>
  30 #include <winerror.h>
  31
  32
  33 #include "initguid.h"
  34 #include "rpc.h"
  35 #include "rpcdce.h"
  36 #include "rpcproxy.h"
  37
  38 static CStdPSFactoryBuffer PSFactoryBuffer;
  39
  40 CSTDSTUBBUFFERRELEASE(&PSFactoryBuffer)
  41 CSTDSTUBBUFFER2RELEASE(&PSFactoryBuffer)
  42
  43 static GUID IID_if1 = {0x12345678, 1234, 5678, {12,34,56,78,90,0xab,0xcd,0xef}};
  44 static GUID IID_if2 = {0x12345679, 1234, 5678, {12,34,56,78,90,0xab,0xcd,0xef}};
  45 static GUID IID_if3 = {0x1234567a, 1234, 5678, {12,34,56,78,90,0xab,0xcd,0xef}};
  46 static GUID IID_if4 = {0x1234567b, 1234, 5678, {12,34,56,78,90,0xab,0xcd,0xef}};
  47
  48 static int my_alloc_called;
  49 static int my_free_called;
  50
  51 static void * CALLBACK my_alloc(SIZE_T size)
  52 {
  53     my_alloc_called++;
  54     return NdrOleAllocate(size);
  55 }
  56
  57 static void CALLBACK my_free(void *ptr)
  58 {
  59     my_free_called++;
  60     NdrOleFree(ptr);
  61 }
  62
  63 typedef struct _MIDL_PROC_FORMAT_STRING
  64 {
  65     short          Pad;
  66     unsigned char  Format[ 2 ];
  67 } MIDL_PROC_FORMAT_STRING;
  68
  69 typedef struct _MIDL_TYPE_FORMAT_STRING
  70 {
  71     short          Pad;
  72     unsigned char  Format[ 2 ];
  73 } MIDL_TYPE_FORMAT_STRING;
  74
  75
  76 static const MIDL_PROC_FORMAT_STRING __MIDL_ProcFormatString =
  77 {
  78     0,
  79     {
  80         0, 0
  81     }
  82 };
  83
  84 static const MIDL_TYPE_FORMAT_STRING __MIDL_TypeFormatString =
  85 {
  86     0,
  87     {
  88         0, 0
  89     }
  90 };
  91
  92 static const MIDL_STUB_DESC Object_StubDesc =
  93     {
  94     NULL,
  95     my_alloc,
  96     my_free,
  97     { 0 },
  98     0,
  99     0,
 100     0,
 101     0,
 102     __MIDL_TypeFormatString.Format,
 103     1, /* -error bounds_check flag */
 104     0x20000, /* Ndr library version */
 105     0,
 106     0x50100a4, /* MIDL Version 5.1.164 */
 107     0,
 108     NULL,
 109     0,  /* notify & notify_flag routine table */
 110     1,  /* Flags */
 111     0,  /* Reserved3 */
 112     0,  /* Reserved4 */
 113     0   /* Reserved5 */
 114     };
 115
 116 static HRESULT WINAPI if1_fn1_Proxy(void *This)
 117 {
 118     return S_OK;
 119 }
 120
 121 static void __RPC_STUB if1_fn1_Stub(
 122     IRpcStubBuffer *This,
 123     IRpcChannelBuffer *_pRpcChannelBuffer,
 124     PRPC_MESSAGE _pRpcMessage,
 125     DWORD *_pdwStubPhase)
 126 {
 127     trace("fn1 stub\n");
 128 }
 129
 130 static HRESULT WINAPI if1_fn2_Proxy(void *This)
 131 {
 132     return S_OK;
 133 }
 134
 135 static void __RPC_STUB if1_fn2_Stub(
 136     IRpcStubBuffer *This,
 137     IRpcChannelBuffer *_pRpcChannelBuffer,
 138     PRPC_MESSAGE _pRpcMessage,
 139     DWORD *_pdwStubPhase)
 140 {
 141     trace("fn2 stub\n");
 142 }
 143
 144 static CINTERFACE_PROXY_VTABLE(5) if1_proxy_vtbl =
 145 {
 146     { &IID_if1 },
 147     {   IUnknown_QueryInterface_Proxy,
 148         IUnknown_AddRef_Proxy,
 149         IUnknown_Release_Proxy ,
 150         if1_fn1_Proxy,
 151         if1_fn2_Proxy
 152     }
 153 };
 154
 155
 156 static const unsigned short if1_FormatStringOffsetTable[] =
 157     {
 158     0,
 159     0
 160     };
 161
 162 static const MIDL_SERVER_INFO if1_server_info =
 163     {
 164     &Object_StubDesc,
 165     0,
 166     __MIDL_ProcFormatString.Format,
 167     &if1_FormatStringOffsetTable[-3],
 168     0,
 169     0,
 170     0,
 171     0};
 172
 173
 174 static const PRPC_STUB_FUNCTION if1_table[] =
 175 {
 176     if1_fn1_Stub,
 177     if1_fn2_Stub
 178 };
 179
 180 static CInterfaceStubVtbl if1_stub_vtbl =
 181 {
 182     {
 183         &IID_if1,
 184         &if1_server_info,
 185         5,
 186         &if1_table[-3]
 187     },
 188     { CStdStubBuffer_METHODS }
 189 };
 190
 191 static CINTERFACE_PROXY_VTABLE(13) if2_proxy_vtbl =
 192 {
 193     { &IID_if2 },
 194     {   IUnknown_QueryInterface_Proxy,
 195         IUnknown_AddRef_Proxy,
 196         IUnknown_Release_Proxy ,
 197         0,
 198         0,
 199         0,
 200         0,
 201         0,
 202         0,
 203         0,
 204         0,
 205         0,
 206         0
 207     }
 208 };
 209
 210 static const unsigned short if2_FormatStringOffsetTable[] =
 211     {
 212     (unsigned short) -1,
 213     (unsigned short) -1,
 214     (unsigned short) -1,
 215     (unsigned short) -1,
 216     (unsigned short) -1,
 217     (unsigned short) -1,
 218     (unsigned short) -1,
 219     (unsigned short) -1,
 220     (unsigned short) -1,
 221     (unsigned short) -1,
 222     0
 223     };
 224
 225 static const MIDL_SERVER_INFO if2_server_info =
 226     {
 227     &Object_StubDesc,
 228     0,
 229     __MIDL_ProcFormatString.Format,
 230     &if2_FormatStringOffsetTable[-3],
 231     0,
 232     0,
 233     0,
 234     0};
 235
 236
 237 static const PRPC_STUB_FUNCTION if2_table[] =
 238 {
 239     STUB_FORWARDING_FUNCTION,
 240     STUB_FORWARDING_FUNCTION,
 241     STUB_FORWARDING_FUNCTION,
 242     STUB_FORWARDING_FUNCTION,
 243     STUB_FORWARDING_FUNCTION,
 244     STUB_FORWARDING_FUNCTION,
 245     STUB_FORWARDING_FUNCTION,
 246     STUB_FORWARDING_FUNCTION,
 247     STUB_FORWARDING_FUNCTION,
 248     STUB_FORWARDING_FUNCTION
 249 };
 250
 251 static CInterfaceStubVtbl if2_stub_vtbl =
 252 {
 253     {
 254         &IID_if2,
 255         &if2_server_info,
 256         13,
 257         &if2_table[-3]
 258     },
 259     { CStdStubBuffer_DELEGATING_METHODS }
 260 };
 261
 262 static CINTERFACE_PROXY_VTABLE(4) if3_proxy_vtbl =
 263 {
 264     { &IID_if3 },
 265     {   IUnknown_QueryInterface_Proxy,
 266         IUnknown_AddRef_Proxy,
 267         IUnknown_Release_Proxy ,
 268         if1_fn1_Proxy
 269     }
 270 };
 271
 272
 273 static const unsigned short if3_FormatStringOffsetTable[] =
 274     {
 275     0,
 276     0
 277     };
 278
 279 static const MIDL_SERVER_INFO if3_server_info =
 280     {
 281     &Object_StubDesc,
 282     0,
 283     __MIDL_ProcFormatString.Format,
 284     &if3_FormatStringOffsetTable[-3],
 285     0,
 286     0,
 287     0,
 288     0};
 289
 290
 291 static const PRPC_STUB_FUNCTION if3_table[] =
 292 {
 293     if1_fn1_Stub
 294 };
 295
 296 static CInterfaceStubVtbl if3_stub_vtbl =
 297 {
 298     {
 299         &IID_if3,
 300         &if3_server_info,
 301         4,
 302         &if1_table[-3]
 303     },
 304     { 0, 0, 0, 0, 0, 0, 0, 0, 0, 0 }
 305 };
 306
 307 static CINTERFACE_PROXY_VTABLE(7) if4_proxy_vtbl =
 308 {
 309     { &IID_if4 },
 310     {   IUnknown_QueryInterface_Proxy,
 311         IUnknown_AddRef_Proxy,
 312         IUnknown_Release_Proxy ,
 313         0,
 314         0,
 315         0,
 316         0
 317     }
 318 };
 319
 320 static const unsigned short if4_FormatStringOffsetTable[] =
 321     {
 322     (unsigned short) -1,
 323     (unsigned short) -1,
 324     (unsigned short) -1,
 325     (unsigned short) -1,
 326     0
 327     };
 328
 329 static const MIDL_SERVER_INFO if4_server_info =
 330     {
 331     &Object_StubDesc,
 332     0,
 333     __MIDL_ProcFormatString.Format,
 334     &if4_FormatStringOffsetTable[-3],
 335     0,
 336     0,
 337     0,
 338     0};
 339
 340
 341 static const PRPC_STUB_FUNCTION if4_table[] =
 342 {
 343     STUB_FORWARDING_FUNCTION,
 344     STUB_FORWARDING_FUNCTION,
 345     STUB_FORWARDING_FUNCTION,
 346     STUB_FORWARDING_FUNCTION,
 347 };
 348
 349 static CInterfaceStubVtbl if4_stub_vtbl =
 350 {
 351     {
 352         &IID_if4,
 353         &if4_server_info,
 354         7,
 355         &if2_table[-3]
 356     },
 357     { 0, 0, 0, 0, 0, 0, 0, 0, 0, 0 }
 358 };
 359
 360 static const CInterfaceProxyVtbl *cstub_ProxyVtblList[] =
 361 {
 362     (const CInterfaceProxyVtbl *) &if1_proxy_vtbl,
 363     (const CInterfaceProxyVtbl *) &if2_proxy_vtbl,
 364     (const CInterfaceProxyVtbl *) &if3_proxy_vtbl,
 365     (const CInterfaceProxyVtbl *) &if4_proxy_vtbl,
 366     NULL
 367 };
 368
 369 static const CInterfaceStubVtbl *cstub_StubVtblList[] =
 370 {
 371     &if1_stub_vtbl,
 372     &if2_stub_vtbl,
 373     &if3_stub_vtbl,
 374     &if4_stub_vtbl,
 375     NULL
 376 };
 377
 378 static PCInterfaceName const if_name_list[] =
 379 {
 380     "if1",
 381     "if2",
 382     "if3",
 383     "if4",
 384     NULL
 385 };
 386
 387 static const IID *base_iid_list[] =
 388 {
 389     NULL,
 390     &IID_ITypeLib,
 391     NULL,
 392     &IID_IDispatch,
 393     NULL
 394 };
 395
 396 #define cstub_CHECK_IID(n)     IID_GENERIC_CHECK_IID( cstub, pIID, n)
 397
 398 static int __stdcall iid_lookup( const IID * pIID, int * pIndex )
 399 {
 400     IID_BS_LOOKUP_SETUP
 401
 402     IID_BS_LOOKUP_INITIAL_TEST( cstub, 4, 4 )
 403     IID_BS_LOOKUP_NEXT_TEST( cstub, 2 )
 404     IID_BS_LOOKUP_NEXT_TEST( cstub, 1 )
 405     IID_BS_LOOKUP_RETURN_RESULT( cstub, 4, *pIndex )
 406
 407 }
 408
 409
 410 static const ExtendedProxyFileInfo my_proxy_file_info =
 411 {
 412     (const PCInterfaceProxyVtblList *) &cstub_ProxyVtblList,
 413     (const PCInterfaceStubVtblList *) &cstub_StubVtblList,
 414     (const PCInterfaceName *) &if_name_list,
 415     (const IID **) &base_iid_list,
 416     &iid_lookup,
 417     4,
 418     1,
 419     NULL,
 420     0,
 421     0,
 422     0
 423 };
 424
 425 static const ProxyFileInfo *proxy_file_list[] = {
 426     &my_proxy_file_info,
 427     NULL
 428 };
 429
 430
 431 static IPSFactoryBuffer *test_NdrDllGetClassObject(void)
 432 {
 433     IPSFactoryBuffer *ppsf = NULL;
 434     const CLSID PSDispatch = {0x20420, 0, 0, {0xc0, 0, 0, 0, 0, 0, 0, 0x46}};
 435     const CLSID CLSID_Unknown = {0x45678, 0x1234, 0x6666, {0xff, 0x67, 0x45, 0x98, 0x76, 0x12, 0x34, 0x56}};
 436     HRESULT r;
 437     HMODULE hmod = GetModuleHandleA("rpcrt4.dll");
 438     void *CStd_QueryInterface = GetProcAddress(hmod, "CStdStubBuffer_QueryInterface");
 439     void *CStd_AddRef = GetProcAddress(hmod, "CStdStubBuffer_AddRef");
 440     void *CStd_Release = GetProcAddress(hmod, "NdrCStdStubBuffer_Release");
 441     void *CStd_Connect = GetProcAddress(hmod, "CStdStubBuffer_Connect");
 442     void *CStd_Disconnect = GetProcAddress(hmod, "CStdStubBuffer_Disconnect");
 443     void *CStd_Invoke = GetProcAddress(hmod, "CStdStubBuffer_Invoke");
 444     void *CStd_IsIIDSupported = GetProcAddress(hmod, "CStdStubBuffer_IsIIDSupported");
 445     void *CStd_CountRefs = GetProcAddress(hmod, "CStdStubBuffer_CountRefs");
 446     void *CStd_DebugServerQueryInterface = GetProcAddress(hmod, "CStdStubBuffer_DebugServerQueryInterface");
 447     void *CStd_DebugServerRelease = GetProcAddress(hmod, "CStdStubBuffer_DebugServerRelease");
 448
 449     r = NdrDllGetClassObject(&PSDispatch, &IID_IPSFactoryBuffer, (void**)&ppsf, proxy_file_list,
 450                              &CLSID_Unknown, &PSFactoryBuffer);
 451     ok(r == CLASS_E_CLASSNOTAVAILABLE, "NdrDllGetClassObject with unknown clsid should have returned CLASS_E_CLASSNOTAVAILABLE instead of 0x%x\n", r);
 452     ok(ppsf == NULL, "NdrDllGetClassObject should have set ppsf to NULL on failure\n");
 453
 454     r = NdrDllGetClassObject(&PSDispatch, &IID_IPSFactoryBuffer, (void**)&ppsf, proxy_file_list,
 455                          &PSDispatch, &PSFactoryBuffer);
 456
 457     ok(r == S_OK, "ret %08x\n", r);
 458     ok(ppsf != NULL, "ppsf == NULL\n");
 459
 460     ok(PSFactoryBuffer.pProxyFileList == proxy_file_list, "pfl not the same\n");
 461     ok(PSFactoryBuffer.pProxyFileList[0]->pStubVtblList == (PCInterfaceStubVtblList *) &cstub_StubVtblList, "stub vtbllist not the same\n");
 462
 463     /* if1 is non-delegating, if2 is delegating, if3 is non-delegating
 464        but I've zero'ed the vtbl entries, similarly if4 is delegating
 465        with zero'ed vtbl entries */
 466
 467 #define VTBL_TEST_NOT_CHANGE_TO(name, i)                                  \
 468     ok(PSFactoryBuffer.pProxyFileList[0]->pStubVtblList[i]->Vtbl.name != CStd_##name, #name "vtbl %d updated %p %p\n", \
 469        i, PSFactoryBuffer.pProxyFileList[0]->pStubVtblList[i]->Vtbl.name, CStd_##name );
 470 #define VTBL_TEST_CHANGE_TO(name, i)                                  \
 471     ok(PSFactoryBuffer.pProxyFileList[0]->pStubVtblList[i]->Vtbl.name == CStd_##name, #name "vtbl %d not updated %p %p\n", \
 472        i, PSFactoryBuffer.pProxyFileList[0]->pStubVtblList[i]->Vtbl.name, CStd_##name );
 473 #define VTBL_TEST_ZERO(name, i)                                  \
 474     ok(PSFactoryBuffer.pProxyFileList[0]->pStubVtblList[i]->Vtbl.name == NULL, #name "vtbl %d not null %p\n", \
 475        i, PSFactoryBuffer.pProxyFileList[0]->pStubVtblList[i]->Vtbl.name );
 476     VTBL_TEST_NOT_CHANGE_TO(QueryInterface, 0);
 477     VTBL_TEST_NOT_CHANGE_TO(AddRef, 0);
 478     VTBL_TEST_NOT_CHANGE_TO(Release, 0);
 479     VTBL_TEST_NOT_CHANGE_TO(Connect, 0);
 480     VTBL_TEST_NOT_CHANGE_TO(Disconnect, 0);
 481     VTBL_TEST_NOT_CHANGE_TO(Invoke, 0);
 482     VTBL_TEST_NOT_CHANGE_TO(IsIIDSupported, 0);
 483     VTBL_TEST_NOT_CHANGE_TO(CountRefs, 0);
 484     VTBL_TEST_NOT_CHANGE_TO(DebugServerQueryInterface, 0);
 485     VTBL_TEST_NOT_CHANGE_TO(DebugServerRelease, 0);
 486
 487     VTBL_TEST_CHANGE_TO(QueryInterface, 1);
 488     VTBL_TEST_CHANGE_TO(AddRef, 1);
 489     VTBL_TEST_NOT_CHANGE_TO(Release, 1);
 490     VTBL_TEST_NOT_CHANGE_TO(Connect, 1);
 491     VTBL_TEST_NOT_CHANGE_TO(Disconnect, 1);
 492     VTBL_TEST_CHANGE_TO(Invoke, 1);
 493     VTBL_TEST_CHANGE_TO(IsIIDSupported, 1);
 494     VTBL_TEST_NOT_CHANGE_TO(CountRefs, 1);
 495     VTBL_TEST_CHANGE_TO(DebugServerQueryInterface, 1);
 496     VTBL_TEST_CHANGE_TO(DebugServerRelease, 1);
 497
 498     VTBL_TEST_CHANGE_TO(QueryInterface, 2);
 499     VTBL_TEST_CHANGE_TO(AddRef, 2);
 500     VTBL_TEST_ZERO(Release, 2);
 501     VTBL_TEST_CHANGE_TO(Connect, 2);
 502     VTBL_TEST_CHANGE_TO(Disconnect, 2);
 503     VTBL_TEST_CHANGE_TO(Invoke, 2);
 504     VTBL_TEST_CHANGE_TO(IsIIDSupported, 2);
 505     VTBL_TEST_CHANGE_TO(CountRefs, 2);
 506     VTBL_TEST_CHANGE_TO(DebugServerQueryInterface, 2);
 507     VTBL_TEST_CHANGE_TO(DebugServerRelease, 2);
 508
 509     VTBL_TEST_CHANGE_TO(QueryInterface, 3);
 510     VTBL_TEST_CHANGE_TO(AddRef, 3);
 511     VTBL_TEST_ZERO(Release, 3);
 512     VTBL_TEST_NOT_CHANGE_TO(Connect, 3);
 513     VTBL_TEST_NOT_CHANGE_TO(Disconnect, 3);
 514     VTBL_TEST_CHANGE_TO(Invoke, 3);
 515     VTBL_TEST_CHANGE_TO(IsIIDSupported, 3);
 516     VTBL_TEST_NOT_CHANGE_TO(CountRefs, 3);
 517     VTBL_TEST_CHANGE_TO(DebugServerQueryInterface, 3);
 518     VTBL_TEST_CHANGE_TO(DebugServerRelease, 3);
 519
 520
 521
 522 #undef VTBL_TEST_NOT_CHANGE_TO
 523 #undef VTBL_TEST_CHANGE_TO
 524 #undef VTBL_TEST_ZERO
 525
 526     ok(PSFactoryBuffer.RefCount == 1, "ref count %d\n", PSFactoryBuffer.RefCount);
 527     IPSFactoryBuffer_Release(ppsf);
 528
 529     r = NdrDllGetClassObject(&IID_if3, &IID_IPSFactoryBuffer, (void**)&ppsf, proxy_file_list,
 530                              NULL, &PSFactoryBuffer);
 531     ok(r == S_OK, "ret %08x\n", r);
 532     ok(ppsf != NULL, "ppsf == NULL\n");
 533
 534     return ppsf;
 535 }
 536
 537 static int base_buffer_invoke_called;
 538 static HRESULT WINAPI base_buffer_Invoke(IRpcStubBuffer *This, RPCOLEMESSAGE *msg, IRpcChannelBuffer *channel)
 539 {
 540     base_buffer_invoke_called++;
 541     ok(msg == (RPCOLEMESSAGE*)0xcafebabe, "msg ptr changed\n");
 542     ok(channel == (IRpcChannelBuffer*)0xdeadbeef, "channel ptr changed\n");
 543     return S_OK; /* returning any failure here results in an exception */
 544 }
 545
 546 static IRpcStubBufferVtbl base_buffer_vtbl = {
 547     (void*)0xcafebab0,
 548     (void*)0xcafebab1,
 549     (void*)0xcafebab2,
 550     (void*)0xcafebab3,
 551     (void*)0xcafebab4,
 552     base_buffer_Invoke,
 553     (void*)0xcafebab6,
 554     (void*)0xcafebab7,
 555     (void*)0xcafebab8,
 556     (void*)0xcafebab9
 557 };
 558
 559 static void test_NdrStubForwardingFunction(void)
 560 {
 561     void *This[5];
 562     void *real_this;
 563     IRpcChannelBuffer *channel = (IRpcChannelBuffer*)0xdeadbeef;
 564     RPC_MESSAGE *msg = (RPC_MESSAGE*)0xcafebabe;
 565     DWORD *phase = (DWORD*)0x12345678;
 566     IRpcStubBufferVtbl *base_buffer_vtbl_ptr = &base_buffer_vtbl;
 567     IRpcStubBuffer *base_stub_buffer = (IRpcStubBuffer*)&base_buffer_vtbl_ptr;
 568
 569     memset(This, 0xcc, sizeof(This));
 570     This[0] = base_stub_buffer;
 571     real_this = &This[1];
 572
 573     NdrStubForwardingFunction( real_this, channel, msg, phase );
 574     ok(base_buffer_invoke_called == 1, "base_buffer_invoke called %d times\n", base_buffer_invoke_called);
 575
 576 }
 577
 578 static IRpcStubBuffer *create_stub(IPSFactoryBuffer *ppsf, REFIID iid, IUnknown *obj, HRESULT expected_result)
 579 {
 580     IRpcStubBuffer *pstub = NULL;
 581     HRESULT r;
 582
 583     r = IPSFactoryBuffer_CreateStub(ppsf, iid, obj, &pstub);
 584     ok(r == expected_result, "CreateStub returned %08x expected %08x\n", r, expected_result);
 585     return pstub;
 586 }
 587
 588 static HRESULT WINAPI create_stub_test_QI(IUnknown *This, REFIID iid, void **ppv)
 589 {
 590     ok(IsEqualIID(iid, &IID_if1), "incorrect iid\n");
 591     *ppv = (void*)0xdeadbeef;
 592     return S_OK;
 593 }
 594
 595 static IUnknownVtbl create_stub_test_vtbl =
 596 {
 597     create_stub_test_QI,
 598     NULL,
 599     NULL
 600 };
 601
 602 static HRESULT WINAPI create_stub_test_fail_QI(IUnknown *This, REFIID iid, void **ppv)
 603 {
 604     ok(IsEqualIID(iid, &IID_if1), "incorrect iid\n");
 605     *ppv = NULL;
 606     return E_NOINTERFACE;
 607 }
 608
 609 static IUnknownVtbl create_stub_test_fail_vtbl =
 610 {
 611     create_stub_test_fail_QI,
 612     NULL,
 613     NULL
 614 };
 615
 616 struct dummy_unknown
 617 {
 618     const IUnknownVtbl *vtbl;
 619     LONG ref;
 620 };
 621
 622 static HRESULT WINAPI dummy_QueryInterface(IUnknown *This, REFIID iid, void **ppv)
 623 {
 624     *ppv = NULL;
 625     return E_NOINTERFACE;
 626 }
 627
 628 static ULONG WINAPI dummy_AddRef(LPUNKNOWN iface)
 629 {
 630     struct dummy_unknown *this = (struct dummy_unknown *)iface;
 631     return InterlockedIncrement( &this->ref );
 632 }
 633
 634 ULONG WINAPI dummy_Release(LPUNKNOWN iface)
 635 {
 636     struct dummy_unknown *this = (struct dummy_unknown *)iface;
 637     return InterlockedDecrement( &this->ref );
 638 }
 639
 640 static IUnknownVtbl dummy_unknown_vtbl =
 641 {
 642     dummy_QueryInterface,
 643     dummy_AddRef,
 644     dummy_Release
 645 };
 646 static struct dummy_unknown dummy_unknown = { &dummy_unknown_vtbl, 0 };
 647
 648 static void create_proxy_test( IPSFactoryBuffer *ppsf, REFIID iid, const void *expected_vtbl )
 649 {
 650     IRpcProxyBuffer *proxy = NULL;
 651     IUnknown *iface = NULL;
 652     HRESULT r;
 653     ULONG count;
 654
 655     r = IPSFactoryBuffer_CreateProxy(ppsf, NULL, iid, &proxy, (void **)&iface);
 656     ok( r == S_OK, "IPSFactoryBuffer_CreateProxy failed %x\n", r );
 657     ok( *(void **)iface == expected_vtbl, "wrong iface pointer %p/%p\n", *(void **)iface, expected_vtbl );
 658     count = IUnknown_Release( iface );
 659     ok( count == 1, "wrong refcount %u\n", count );
 660     count = IRpcProxyBuffer_Release( proxy );
 661     ok( count == 0, "wrong refcount %u\n", count );
 662
 663     dummy_unknown.ref = 4;
 664     r = IPSFactoryBuffer_CreateProxy(ppsf, (IUnknown *)&dummy_unknown, iid, &proxy, (void **)&iface);
 665     ok( r == S_OK, "IPSFactoryBuffer_CreateProxy failed %x\n", r );
 666     ok( dummy_unknown.ref == 5, "wrong refcount %u\n", dummy_unknown.ref );
 667     ok( *(void **)iface == expected_vtbl, "wrong iface pointer %p/%p\n", *(void **)iface, expected_vtbl );
 668     count = IUnknown_Release( iface );
 669     ok( count == 4, "wrong refcount %u\n", count );
 670     ok( dummy_unknown.ref == 4, "wrong refcount %u\n", dummy_unknown.ref );
 671     count = IRpcProxyBuffer_Release( proxy );
 672     ok( count == 0, "wrong refcount %u\n", count );
 673     ok( dummy_unknown.ref == 4, "wrong refcount %u\n", dummy_unknown.ref );
 674 }
 675
 676 static void test_CreateProxy( IPSFactoryBuffer *ppsf )
 677 {
 678     create_proxy_test( ppsf, &IID_if1, if1_proxy_vtbl.Vtbl );
 679     create_proxy_test( ppsf, &IID_if2, if2_proxy_vtbl.Vtbl );
 680     create_proxy_test( ppsf, &IID_if3, if3_proxy_vtbl.Vtbl );
 681     create_proxy_test( ppsf, &IID_if4, if4_proxy_vtbl.Vtbl );
 682 }
 683
 684 static void test_CreateStub(IPSFactoryBuffer *ppsf)
 685 {
 686     IUnknownVtbl *vtbl = &create_stub_test_vtbl;
 687     IUnknown *obj = (IUnknown*)&vtbl;
 688     IRpcStubBuffer *pstub = create_stub(ppsf, &IID_if1, obj, S_OK);
 689     CStdStubBuffer *cstd_stub = (CStdStubBuffer*)pstub;
 690     const CInterfaceStubHeader *header = &CONTAINING_RECORD(cstd_stub->lpVtbl, const CInterfaceStubVtbl, Vtbl)->header;
 691
 692     ok(IsEqualIID(header->piid, &IID_if1), "header iid differs\n");
 693     ok(cstd_stub->RefCount == 1, "ref count %d\n", cstd_stub->RefCount);
 694     /* 0xdeadbeef returned from create_stub_test_QI */
 695     ok(cstd_stub->pvServerObject == (void*)0xdeadbeef, "pvServerObject %p\n", cstd_stub->pvServerObject);
 696     ok(cstd_stub->pPSFactory != NULL, "pPSFactory was NULL\n");
 697
 698     vtbl = &create_stub_test_fail_vtbl;
 699     pstub = create_stub(ppsf, &IID_if1, obj, E_NOINTERFACE);
 700
 701 }
 702
 703 static HRESULT WINAPI connect_test_orig_QI(IUnknown *This, REFIID iid, void **ppv)
 704 {
 705     ok(IsEqualIID(iid, &IID_if1) ||
 706        IsEqualIID(iid, &IID_if2), "incorrect iid\n");
 707     *ppv = (void*)This;
 708     return S_OK;
 709 }
 710
 711 static int connect_test_orig_release_called;
 712 static ULONG WINAPI connect_test_orig_release(IUnknown *This)
 713 {
 714     connect_test_orig_release_called++;
 715     return 0;
 716 }
 717
 718 static IUnknownVtbl connect_test_orig_vtbl =
 719 {
 720     connect_test_orig_QI,
 721     NULL,
 722     connect_test_orig_release
 723 };
 724
 725 static HRESULT WINAPI connect_test_new_QI(IUnknown *This, REFIID iid, void **ppv)
 726 {
 727     ok(IsEqualIID(iid, &IID_if1) ||
 728        IsEqualIID(iid, &IID_if2), "incorrect iid\n");
 729     *ppv = (void*)0xcafebabe;
 730     return S_OK;
 731 }
 732
 733 static IUnknownVtbl connect_test_new_vtbl =
 734 {
 735     connect_test_new_QI,
 736     NULL,
 737     NULL
 738 };
 739
 740 static HRESULT WINAPI connect_test_new_fail_QI(IUnknown *This, REFIID iid, void **ppv)
 741 {
 742     ok(IsEqualIID(iid, &IID_if1), "incorrect iid\n");
 743     *ppv = (void*)0xdeadbeef;
 744     return E_NOINTERFACE;
 745 }
 746
 747 static IUnknownVtbl connect_test_new_fail_vtbl =
 748 {
 749     connect_test_new_fail_QI,
 750     NULL,
 751     NULL
 752 };
 753
 754 static int connect_test_base_Connect_called;
 755 static HRESULT WINAPI connect_test_base_Connect(IRpcStubBuffer *pstub, IUnknown *obj)
 756 {
 757     connect_test_base_Connect_called++;
 758     ok(*(void**)obj == (void*)0xbeefcafe, "unexpected obj %p\n", obj);
 759     return S_OK;
 760 }
 761
 762 static IRpcStubBufferVtbl connect_test_base_stub_buffer_vtbl =
 763 {
 764     (void*)0xcafebab0,
 765     (void*)0xcafebab1,
 766     (void*)0xcafebab2,
 767     connect_test_base_Connect,
 768     (void*)0xcafebab4,
 769     (void*)0xcafebab5,
 770     (void*)0xcafebab6,
 771     (void*)0xcafebab7,
 772     (void*)0xcafebab8,
 773     (void*)0xcafebab9
 774 };
 775
 776 static void test_Connect(IPSFactoryBuffer *ppsf)
 777 {
 778     IUnknownVtbl *orig_vtbl = &connect_test_orig_vtbl;
 779     IUnknownVtbl *new_vtbl = &connect_test_new_vtbl;
 780     IUnknownVtbl *new_fail_vtbl = &connect_test_new_fail_vtbl;
 781     IUnknown *obj = (IUnknown*)&orig_vtbl;
 782     IRpcStubBuffer *pstub = create_stub(ppsf, &IID_if1, obj, S_OK);
 783     CStdStubBuffer *cstd_stub = (CStdStubBuffer*)pstub;
 784     IRpcStubBufferVtbl *base_stub_buf_vtbl = &connect_test_base_stub_buffer_vtbl;
 785     HRESULT r;
 786
 787     obj = (IUnknown*)&new_vtbl;
 788     r = IRpcStubBuffer_Connect(pstub, obj);
 789     ok(r == S_OK, "r %08x\n", r);
 790     ok(connect_test_orig_release_called == 1, "release called %d\n", connect_test_orig_release_called);
 791     ok(cstd_stub->pvServerObject == (void*)0xcafebabe, "pvServerObject %p\n", cstd_stub->pvServerObject);
 792
 793     cstd_stub->pvServerObject = (IUnknown*)&orig_vtbl;
 794     obj = (IUnknown*)&new_fail_vtbl;
 795     r = IRpcStubBuffer_Connect(pstub, obj);
 796     ok(r == E_NOINTERFACE, "r %08x\n", r);
 797     ok(cstd_stub->pvServerObject == (void*)0xdeadbeef, "pvServerObject %p\n", cstd_stub->pvServerObject);
 798     ok(connect_test_orig_release_called == 2, "release called %d\n", connect_test_orig_release_called);
 799
 800     /* Now use a delegated stub.
 801
 802        We know from the NdrStubForwardFunction test that
 803        (void**)pstub-1 is the base interface stub buffer.  This shows
 804        that (void**)pstub-2 contains the address of a vtable that gets
 805        passed to the base interface's Connect method.  Note that
 806        (void**)pstub-2 itself gets passed to Connect and not
 807        *((void**)pstub-2), so it should contain the vtable ptr and not
 808        an interface ptr. */
 809
 810     obj = (IUnknown*)&orig_vtbl;
 811     pstub = create_stub(ppsf, &IID_if2, obj, S_OK);
 812     *((void**)pstub-1) = &base_stub_buf_vtbl;
 813     *((void**)pstub-2) = (void*)0xbeefcafe;
 814
 815     obj = (IUnknown*)&new_vtbl;
 816     r = IRpcStubBuffer_Connect(pstub, obj);
 817     ok(connect_test_base_Connect_called == 1, "connect_test_bsae_Connect called %d times\n",
 818        connect_test_base_Connect_called);
 819     ok(connect_test_orig_release_called == 3, "release called %d\n", connect_test_orig_release_called);
 820     cstd_stub = (CStdStubBuffer*)pstub;
 821     ok(cstd_stub->pvServerObject == (void*)0xcafebabe, "pvServerObject %p\n", cstd_stub->pvServerObject);
 822 }
 823
 824 static void test_Disconnect(IPSFactoryBuffer *ppsf)
 825 {
 826     IUnknownVtbl *orig_vtbl = &connect_test_orig_vtbl;
 827     IUnknown *obj = (IUnknown*)&orig_vtbl;
 828     IRpcStubBuffer *pstub = create_stub(ppsf, &IID_if1, obj, S_OK);
 829     CStdStubBuffer *cstd_stub = (CStdStubBuffer*)pstub;
 830
 831     connect_test_orig_release_called = 0;
 832     IRpcStubBuffer_Disconnect(pstub);
 833     ok(connect_test_orig_release_called == 1, "release called %d\n", connect_test_orig_release_called);
 834     ok(cstd_stub->pvServerObject == NULL, "pvServerObject %p\n", cstd_stub->pvServerObject);
 835 }
 836
 837
 838 static int release_test_psfacbuf_release_called;
 839 static ULONG WINAPI release_test_pretend_psfacbuf_release(IUnknown *pUnk)
 840 {
 841     release_test_psfacbuf_release_called++;
 842     return 1;
 843 }
 844
 845 static IUnknownVtbl release_test_pretend_psfacbuf_vtbl =
 846 {
 847     NULL,
 848     NULL,
 849     release_test_pretend_psfacbuf_release
 850 };
 851
 852 static void test_Release(IPSFactoryBuffer *ppsf)
 853 {
 854     LONG facbuf_refs;
 855     IUnknownVtbl *orig_vtbl = &connect_test_orig_vtbl;
 856     IUnknown *obj = (IUnknown*)&orig_vtbl;
 857     IUnknownVtbl *pretend_psfacbuf_vtbl = &release_test_pretend_psfacbuf_vtbl;
 858     IUnknown *pretend_psfacbuf = (IUnknown *)&pretend_psfacbuf_vtbl;
 859     IRpcStubBuffer *pstub = create_stub(ppsf, &IID_if1, obj, S_OK);
 860     CStdStubBuffer *cstd_stub = (CStdStubBuffer*)pstub;
 861
 862     facbuf_refs = PSFactoryBuffer.RefCount;
 863
 864     /* This shows that NdrCStdStubBuffer_Release doesn't call Disconnect */
 865     ok(cstd_stub->RefCount == 1, "ref count %d\n", cstd_stub->RefCount);
 866     connect_test_orig_release_called = 0;
 867     IRpcStubBuffer_Release(pstub);
 868 todo_wine {
 869     ok(connect_test_orig_release_called == 0, "release called %d\n", connect_test_orig_release_called);
 870 }
 871     ok(PSFactoryBuffer.RefCount == facbuf_refs - 1, "factory buffer refs %d orig %d\n", PSFactoryBuffer.RefCount, facbuf_refs);
 872
 873     /* This shows that NdrCStdStubBuffer_Release calls Release on its 2nd arg, rather than on This->pPSFactory
 874        (which are usually the same and indeed it's odd that _Release requires this 2nd arg). */
 875     pstub = create_stub(ppsf, &IID_if1, obj, S_OK);
 876     ok(PSFactoryBuffer.RefCount == facbuf_refs, "factory buffer refs %d orig %d\n", PSFactoryBuffer.RefCount, facbuf_refs);
 877     NdrCStdStubBuffer_Release(pstub, (IPSFactoryBuffer*)pretend_psfacbuf);
 878     ok(release_test_psfacbuf_release_called == 1, "pretend_psfacbuf_release called %d\n", release_test_psfacbuf_release_called);
 879     ok(PSFactoryBuffer.RefCount == facbuf_refs, "factory buffer refs %d orig %d\n", PSFactoryBuffer.RefCount, facbuf_refs);
 880 }
 881
 882 static HRESULT WINAPI delegating_invoke_test_QI(ITypeLib *pUnk, REFIID iid, void** ppv)
 883 {
 884
 885     *ppv = pUnk;
 886     return S_OK;
 887 }
 888
 889 static ULONG WINAPI delegating_invoke_test_addref(ITypeLib *pUnk)
 890 {
 891     return 1;
 892 }
 893
 894 static ULONG WINAPI delegating_invoke_test_release(ITypeLib *pUnk)
 895 {
 896     return 1;
 897 }
 898
 899 static UINT WINAPI delegating_invoke_test_get_type_info_count(ITypeLib *pUnk)
 900 {
 901     return 0xabcdef;
 902 }
 903
 904 static ITypeLibVtbl delegating_invoke_test_obj_vtbl =
 905 {
 906     delegating_invoke_test_QI,
 907     delegating_invoke_test_addref,
 908     delegating_invoke_test_release,
 909     delegating_invoke_test_get_type_info_count,
 910     NULL,
 911     NULL,
 912     NULL,
 913     NULL,
 914     NULL,
 915     NULL,
 916     NULL,
 917     NULL,
 918     NULL
 919 };
 920
 921 static HRESULT WINAPI delegating_invoke_chan_query_interface(IRpcChannelBuffer *pchan,
 922                                                              REFIID iid,
 923                                                              void **ppv)
 924 {
 925     ok(0, "call to QueryInterface not expected\n");
 926     return E_NOINTERFACE;
 927 }
 928
 929 static ULONG WINAPI delegating_invoke_chan_add_ref(IRpcChannelBuffer *pchan)
 930 {
 931     return 2;
 932 }
 933
 934 static ULONG WINAPI delegating_invoke_chan_release(IRpcChannelBuffer *pchan)
 935 {
 936     return 1;
 937 }
 938
 939 static HRESULT WINAPI delegating_invoke_chan_get_buffer(IRpcChannelBuffer *pchan,
 940                                                         RPCOLEMESSAGE *msg,
 941                                                         REFIID iid)
 942 {
 943     msg->Buffer = HeapAlloc(GetProcessHeap(), 0, msg->cbBuffer);
 944     return S_OK;
 945 }
 946
 947 static HRESULT WINAPI delegating_invoke_chan_send_receive(IRpcChannelBuffer *pchan,
 948                                                           RPCOLEMESSAGE *pMessage,
 949                                                           ULONG *pStatus)
 950 {
 951     ok(0, "call to SendReceive not expected\n");
 952     return E_NOTIMPL;
 953 }
 954
 955 static HRESULT WINAPI delegating_invoke_chan_free_buffer(IRpcChannelBuffer *pchan,
 956                                                          RPCOLEMESSAGE *pMessage)
 957 {
 958     ok(0, "call to FreeBuffer not expected\n");
 959     return E_NOTIMPL;
 960 }
 961
 962 static HRESULT WINAPI delegating_invoke_chan_get_dest_ctx(IRpcChannelBuffer *pchan,
 963                                                           DWORD *pdwDestContext,
 964                                                           void **ppvDestContext)
 965 {
 966     *pdwDestContext = MSHCTX_LOCAL;
 967     *ppvDestContext = NULL;
 968     return S_OK;
 969 }
 970
 971 static HRESULT WINAPI delegating_invoke_chan_is_connected(IRpcChannelBuffer *pchan)
 972 {
 973     ok(0, "call to IsConnected not expected\n");
 974     return E_NOTIMPL;
 975 }
 976
 977 static IRpcChannelBufferVtbl delegating_invoke_test_rpc_chan_vtbl =
 978 {
 979     delegating_invoke_chan_query_interface,
 980     delegating_invoke_chan_add_ref,
 981     delegating_invoke_chan_release,
 982     delegating_invoke_chan_get_buffer,
 983     delegating_invoke_chan_send_receive,
 984     delegating_invoke_chan_free_buffer,
 985     delegating_invoke_chan_get_dest_ctx,
 986     delegating_invoke_chan_is_connected
 987 };
 988
 989 static void test_delegating_Invoke(IPSFactoryBuffer *ppsf)
 990 {
 991     ITypeLibVtbl *obj_vtbl = &delegating_invoke_test_obj_vtbl;
 992     IUnknown *obj = (IUnknown*)&obj_vtbl;
 993     IRpcStubBuffer *pstub = create_stub(ppsf, &IID_if2, obj, S_OK);
 994     IRpcChannelBufferVtbl *pchan_vtbl = &delegating_invoke_test_rpc_chan_vtbl;
 995     IRpcChannelBuffer *pchan = (IRpcChannelBuffer *)&pchan_vtbl;
 996     HRESULT r = E_FAIL;
 997     RPCOLEMESSAGE msg;
 998
 999     memset(&msg, 0, sizeof(msg));
1000     msg.dataRepresentation = NDR_LOCAL_DATA_REPRESENTATION;
1001     msg.iMethod = 3;
1002     r = IRpcStubBuffer_Invoke(pstub, &msg, pchan);
1003     ok(r == S_OK, "ret %08x\n", r);
1004     if(r == S_OK)
1005     {
1006         ok(*(DWORD*)msg.Buffer == 0xabcdef, "buf[0] %08x\n", *(DWORD*)msg.Buffer);
1007         ok(*((DWORD*)msg.Buffer + 1) == S_OK, "buf[1] %08x\n", *((DWORD*)msg.Buffer + 1));
1008     }
1009     /* free the buffer allocated by delegating_invoke_chan_get_buffer */
1010     HeapFree(GetProcessHeap(), 0, msg.Buffer);
1011     IRpcStubBuffer_Release(pstub);
1012 }
1013
1014 START_TEST( cstub )
1015 {
1016     IPSFactoryBuffer *ppsf;
1017
1018     OleInitialize(NULL);
1019
1020     ppsf = test_NdrDllGetClassObject();
1021     test_NdrStubForwardingFunction();
1022     test_CreateProxy(ppsf);
1023     test_CreateStub(ppsf);
1024     test_Connect(ppsf);
1025     test_Disconnect(ppsf);
1026     test_Release(ppsf);
1027     test_delegating_Invoke(ppsf);
1028
1029     OleUninitialize();
1030 }