2002-10-11 00:46:08 +02:00
|
|
|
|
/*
|
|
|
|
|
* COM stub (CStdStubBuffer) implementation
|
|
|
|
|
*
|
|
|
|
|
* Copyright 2001 Ove K<EFBFBD>ven, TransGaming Technologies
|
|
|
|
|
*
|
|
|
|
|
* This library is free software; you can redistribute it and/or
|
|
|
|
|
* modify it under the terms of the GNU Lesser General Public
|
|
|
|
|
* License as published by the Free Software Foundation; either
|
|
|
|
|
* version 2.1 of the License, or (at your option) any later version.
|
|
|
|
|
*
|
|
|
|
|
* This library is distributed in the hope that it will be useful,
|
|
|
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
|
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
|
|
|
* Lesser General Public License for more details.
|
|
|
|
|
*
|
|
|
|
|
* You should have received a copy of the GNU Lesser General Public
|
|
|
|
|
* License along with this library; if not, write to the Free Software
|
2006-05-18 14:49:52 +02:00
|
|
|
|
* Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
|
2002-10-11 00:46:08 +02:00
|
|
|
|
*/
|
|
|
|
|
|
2003-09-06 01:08:26 +02:00
|
|
|
|
#include <stdarg.h>
|
|
|
|
|
|
2004-10-07 05:06:48 +02:00
|
|
|
|
#define COBJMACROS
|
|
|
|
|
|
2002-10-11 00:46:08 +02:00
|
|
|
|
#include "windef.h"
|
|
|
|
|
#include "winbase.h"
|
|
|
|
|
#include "winerror.h"
|
2006-06-10 13:33:40 +02:00
|
|
|
|
#include "excpt.h"
|
2002-10-11 00:46:08 +02:00
|
|
|
|
|
2002-12-05 21:33:07 +01:00
|
|
|
|
#include "objbase.h"
|
2002-10-11 00:46:08 +02:00
|
|
|
|
#include "rpcproxy.h"
|
|
|
|
|
|
|
|
|
|
#include "wine/debug.h"
|
2006-06-10 13:33:40 +02:00
|
|
|
|
#include "wine/exception.h"
|
2002-10-11 00:46:08 +02:00
|
|
|
|
|
|
|
|
|
#include "cpsf.h"
|
|
|
|
|
|
|
|
|
|
WINE_DEFAULT_DEBUG_CHANNEL(ole);
|
|
|
|
|
|
2006-11-11 15:56:45 +01:00
|
|
|
|
#define STUB_HEADER(This) (((const CInterfaceStubHeader*)((This)->lpVtbl))[-1])
|
2002-10-11 00:46:08 +02:00
|
|
|
|
|
2008-02-15 16:19:29 +01:00
|
|
|
|
static LONG WINAPI stub_filter(EXCEPTION_POINTERS *eptr)
|
2006-06-10 13:33:40 +02:00
|
|
|
|
{
|
2008-02-15 16:19:29 +01:00
|
|
|
|
if (eptr->ExceptionRecord->ExceptionFlags & EXCEPTION_NONCONTINUABLE)
|
2006-06-10 13:33:40 +02:00
|
|
|
|
return EXCEPTION_CONTINUE_SEARCH;
|
|
|
|
|
return EXCEPTION_EXECUTE_HANDLER;
|
|
|
|
|
}
|
|
|
|
|
|
2006-08-31 15:58:26 +02:00
|
|
|
|
typedef struct
|
|
|
|
|
{
|
|
|
|
|
IUnknownVtbl *base_obj;
|
|
|
|
|
IRpcStubBuffer *base_stub;
|
|
|
|
|
CStdStubBuffer stub_buffer;
|
|
|
|
|
} cstdstubbuffer_delegating_t;
|
|
|
|
|
|
|
|
|
|
static inline cstdstubbuffer_delegating_t *impl_from_delegating( IRpcStubBuffer *iface )
|
|
|
|
|
{
|
|
|
|
|
return (cstdstubbuffer_delegating_t*)((char *)iface - FIELD_OFFSET(cstdstubbuffer_delegating_t, stub_buffer));
|
|
|
|
|
}
|
|
|
|
|
|
2002-10-11 00:46:08 +02:00
|
|
|
|
HRESULT WINAPI CStdStubBuffer_Construct(REFIID riid,
|
|
|
|
|
LPUNKNOWN pUnkServer,
|
2003-05-20 06:13:44 +02:00
|
|
|
|
PCInterfaceName name,
|
2002-10-11 00:46:08 +02:00
|
|
|
|
CInterfaceStubVtbl *vtbl,
|
|
|
|
|
LPPSFACTORYBUFFER pPSFactory,
|
|
|
|
|
LPRPCSTUBBUFFER *ppStub)
|
|
|
|
|
{
|
|
|
|
|
CStdStubBuffer *This;
|
2006-08-23 15:53:24 +02:00
|
|
|
|
IUnknown *pvServer;
|
|
|
|
|
HRESULT r;
|
2003-05-20 06:13:44 +02:00
|
|
|
|
TRACE("(%p,%p,%p,%p) %s\n", pUnkServer, vtbl, pPSFactory, ppStub, name);
|
2002-10-11 00:46:08 +02:00
|
|
|
|
TRACE("iid=%s\n", debugstr_guid(vtbl->header.piid));
|
|
|
|
|
TRACE("vtbl=%p\n", &vtbl->Vtbl);
|
|
|
|
|
|
|
|
|
|
if (!IsEqualGUID(vtbl->header.piid, riid)) {
|
|
|
|
|
ERR("IID mismatch during stub creation\n");
|
|
|
|
|
return RPC_E_UNEXPECTED;
|
|
|
|
|
}
|
|
|
|
|
|
2006-08-23 15:53:24 +02:00
|
|
|
|
r = IUnknown_QueryInterface(pUnkServer, riid, (void**)&pvServer);
|
|
|
|
|
if(FAILED(r))
|
|
|
|
|
return r;
|
|
|
|
|
|
2002-10-11 00:46:08 +02:00
|
|
|
|
This = HeapAlloc(GetProcessHeap(),HEAP_ZERO_MEMORY,sizeof(CStdStubBuffer));
|
2006-08-23 15:53:24 +02:00
|
|
|
|
if (!This) {
|
|
|
|
|
IUnknown_Release(pvServer);
|
|
|
|
|
return E_OUTOFMEMORY;
|
|
|
|
|
}
|
2002-10-11 00:46:08 +02:00
|
|
|
|
|
|
|
|
|
This->lpVtbl = &vtbl->Vtbl;
|
|
|
|
|
This->RefCount = 1;
|
2006-08-23 15:53:24 +02:00
|
|
|
|
This->pvServerObject = pvServer;
|
2002-10-11 00:46:08 +02:00
|
|
|
|
This->pPSFactory = pPSFactory;
|
|
|
|
|
*ppStub = (LPRPCSTUBBUFFER)This;
|
|
|
|
|
|
|
|
|
|
IPSFactoryBuffer_AddRef(pPSFactory);
|
|
|
|
|
return S_OK;
|
|
|
|
|
}
|
|
|
|
|
|
2006-08-31 15:58:26 +02:00
|
|
|
|
static CRITICAL_SECTION delegating_vtbl_section;
|
|
|
|
|
static CRITICAL_SECTION_DEBUG critsect_debug =
|
|
|
|
|
{
|
|
|
|
|
0, 0, &delegating_vtbl_section,
|
|
|
|
|
{ &critsect_debug.ProcessLocksList, &critsect_debug.ProcessLocksList },
|
|
|
|
|
0, 0, { (DWORD_PTR)(__FILE__ ": delegating_vtbl_section") }
|
|
|
|
|
};
|
|
|
|
|
static CRITICAL_SECTION delegating_vtbl_section = { &critsect_debug, -1, 0, 0, 0, 0 };
|
|
|
|
|
|
|
|
|
|
typedef struct
|
|
|
|
|
{
|
|
|
|
|
DWORD ref;
|
2007-12-03 17:09:05 +01:00
|
|
|
|
DWORD size;
|
|
|
|
|
void **methods;
|
2006-08-31 15:58:26 +02:00
|
|
|
|
IUnknownVtbl vtbl;
|
2007-12-03 17:09:05 +01:00
|
|
|
|
/* remaining entries in vtbl */
|
2006-08-31 15:58:26 +02:00
|
|
|
|
} ref_counted_vtbl;
|
|
|
|
|
|
|
|
|
|
static struct
|
|
|
|
|
{
|
|
|
|
|
ref_counted_vtbl *table;
|
|
|
|
|
} current_vtbl;
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
static HRESULT WINAPI delegating_QueryInterface(IUnknown *pUnk, REFIID iid, void **ppv)
|
|
|
|
|
{
|
|
|
|
|
*ppv = (void *)pUnk;
|
|
|
|
|
return S_OK;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
static ULONG WINAPI delegating_AddRef(IUnknown *pUnk)
|
|
|
|
|
{
|
|
|
|
|
return 1;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
static ULONG WINAPI delegating_Release(IUnknown *pUnk)
|
|
|
|
|
{
|
|
|
|
|
return 1;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
#if defined(__i386__)
|
|
|
|
|
|
|
|
|
|
/* The idea here is to replace the first param on the stack
|
|
|
|
|
ie. This (which will point to cstdstubbuffer_delegating_t)
|
|
|
|
|
with This->stub_buffer.pvServerObject and then jump to the
|
|
|
|
|
relevant offset in This->stub_buffer.pvServerObject's vtbl.
|
|
|
|
|
*/
|
|
|
|
|
#include "pshpack1.h"
|
|
|
|
|
typedef struct {
|
|
|
|
|
DWORD mov1; /* mov 0x4(%esp), %eax 8b 44 24 04 */
|
|
|
|
|
WORD mov2; /* mov 0x10(%eax), %eax 8b 40 */
|
|
|
|
|
BYTE sixteen; /* 10 */
|
|
|
|
|
DWORD mov3; /* mov %eax, 0x4(%esp) 89 44 24 04 */
|
|
|
|
|
WORD mov4; /* mov (%eax), %eax 8b 00 */
|
|
|
|
|
WORD mov5; /* mov offset(%eax), %eax 8b 80 */
|
|
|
|
|
DWORD offset; /* xx xx xx xx */
|
|
|
|
|
WORD jmp; /* jmp *%eax ff e0 */
|
|
|
|
|
BYTE pad[3]; /* lea 0x0(%esi), %esi 8d 76 00 */
|
|
|
|
|
} vtbl_method_t;
|
|
|
|
|
#include "poppack.h"
|
|
|
|
|
|
2007-12-03 17:09:05 +01:00
|
|
|
|
static void fill_table(IUnknownVtbl *vtbl, void **methods, DWORD num)
|
2006-08-31 15:58:26 +02:00
|
|
|
|
{
|
|
|
|
|
vtbl_method_t *method;
|
|
|
|
|
void **entry;
|
|
|
|
|
DWORD i;
|
|
|
|
|
|
|
|
|
|
vtbl->QueryInterface = delegating_QueryInterface;
|
|
|
|
|
vtbl->AddRef = delegating_AddRef;
|
|
|
|
|
vtbl->Release = delegating_Release;
|
|
|
|
|
|
2007-12-03 17:09:05 +01:00
|
|
|
|
method = (vtbl_method_t*)methods;
|
2006-08-31 15:58:26 +02:00
|
|
|
|
entry = (void**)(vtbl + 1);
|
|
|
|
|
|
|
|
|
|
for(i = 3; i < num; i++)
|
|
|
|
|
{
|
|
|
|
|
*entry = method;
|
|
|
|
|
method->mov1 = 0x0424448b;
|
|
|
|
|
method->mov2 = 0x408b;
|
|
|
|
|
method->sixteen = 0x10;
|
|
|
|
|
method->mov3 = 0x04244489;
|
|
|
|
|
method->mov4 = 0x008b;
|
|
|
|
|
method->mov5 = 0x808b;
|
|
|
|
|
method->offset = i << 2;
|
|
|
|
|
method->jmp = 0xe0ff;
|
|
|
|
|
method->pad[0] = 0x8d;
|
|
|
|
|
method->pad[1] = 0x76;
|
|
|
|
|
method->pad[2] = 0x00;
|
|
|
|
|
|
|
|
|
|
method++;
|
|
|
|
|
entry++;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
#else /* __i386__ */
|
|
|
|
|
|
|
|
|
|
typedef struct {int dummy;} vtbl_method_t;
|
|
|
|
|
static void fill_table(IUnknownVtbl *vtbl, DWORD num)
|
|
|
|
|
{
|
|
|
|
|
ERR("delegated stubs are not supported on this architecture\n");
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
#endif /* __i386__ */
|
|
|
|
|
|
|
|
|
|
void create_delegating_vtbl(DWORD num_methods)
|
|
|
|
|
{
|
2006-11-09 23:04:48 +01:00
|
|
|
|
TRACE("%d\n", num_methods);
|
2006-08-31 15:58:26 +02:00
|
|
|
|
if(num_methods <= 3)
|
|
|
|
|
{
|
2006-11-09 23:04:48 +01:00
|
|
|
|
ERR("should have more than %d methods\n", num_methods);
|
2006-08-31 15:58:26 +02:00
|
|
|
|
return;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
EnterCriticalSection(&delegating_vtbl_section);
|
2007-12-03 17:09:05 +01:00
|
|
|
|
if(!current_vtbl.table || num_methods > current_vtbl.table->size)
|
2006-08-31 15:58:26 +02:00
|
|
|
|
{
|
|
|
|
|
DWORD size;
|
2007-12-03 17:09:05 +01:00
|
|
|
|
DWORD old_protect;
|
2006-08-31 15:58:26 +02:00
|
|
|
|
if(current_vtbl.table && current_vtbl.table->ref == 0)
|
|
|
|
|
{
|
|
|
|
|
TRACE("freeing old table\n");
|
2007-12-03 17:09:05 +01:00
|
|
|
|
VirtualFree(current_vtbl.table->methods,
|
|
|
|
|
(current_vtbl.table->size - 3) * sizeof(vtbl_method_t),
|
|
|
|
|
MEM_RELEASE);
|
2006-08-31 15:58:26 +02:00
|
|
|
|
HeapFree(GetProcessHeap(), 0, current_vtbl.table);
|
|
|
|
|
}
|
2007-12-03 17:09:05 +01:00
|
|
|
|
size = (num_methods - 3) * sizeof(vtbl_method_t);
|
|
|
|
|
current_vtbl.table = HeapAlloc(GetProcessHeap(), 0, FIELD_OFFSET(ref_counted_vtbl, vtbl) + num_methods * sizeof(void*));
|
2006-08-31 15:58:26 +02:00
|
|
|
|
current_vtbl.table->ref = 0;
|
2007-12-03 17:09:05 +01:00
|
|
|
|
current_vtbl.table->size = num_methods;
|
|
|
|
|
current_vtbl.table->methods = VirtualAlloc(NULL, size, MEM_COMMIT | MEM_RESERVE, PAGE_EXECUTE_READWRITE);
|
|
|
|
|
fill_table(¤t_vtbl.table->vtbl, current_vtbl.table->methods, num_methods);
|
|
|
|
|
VirtualProtect(current_vtbl.table->methods, size, PAGE_EXECUTE_READ, &old_protect);
|
2006-08-31 15:58:26 +02:00
|
|
|
|
}
|
|
|
|
|
LeaveCriticalSection(&delegating_vtbl_section);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
static IUnknownVtbl *get_delegating_vtbl(void)
|
|
|
|
|
{
|
|
|
|
|
IUnknownVtbl *ret;
|
|
|
|
|
|
|
|
|
|
EnterCriticalSection(&delegating_vtbl_section);
|
|
|
|
|
current_vtbl.table->ref++;
|
|
|
|
|
ret = ¤t_vtbl.table->vtbl;
|
|
|
|
|
LeaveCriticalSection(&delegating_vtbl_section);
|
|
|
|
|
return ret;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
static void release_delegating_vtbl(IUnknownVtbl *vtbl)
|
|
|
|
|
{
|
|
|
|
|
ref_counted_vtbl *table = (ref_counted_vtbl*)((DWORD *)vtbl - 1);
|
|
|
|
|
|
|
|
|
|
EnterCriticalSection(&delegating_vtbl_section);
|
|
|
|
|
table->ref--;
|
2006-11-09 23:04:48 +01:00
|
|
|
|
TRACE("ref now %d\n", table->ref);
|
2006-08-31 15:58:26 +02:00
|
|
|
|
if(table->ref == 0 && table != current_vtbl.table)
|
|
|
|
|
{
|
|
|
|
|
TRACE("... and we're not current so free'ing\n");
|
2007-12-03 17:09:05 +01:00
|
|
|
|
VirtualFree(current_vtbl.table->methods,
|
|
|
|
|
(current_vtbl.table->size - 3) * sizeof(vtbl_method_t),
|
|
|
|
|
MEM_RELEASE);
|
2006-08-31 15:58:26 +02:00
|
|
|
|
HeapFree(GetProcessHeap(), 0, table);
|
|
|
|
|
}
|
|
|
|
|
LeaveCriticalSection(&delegating_vtbl_section);
|
|
|
|
|
}
|
|
|
|
|
|
2006-08-31 16:40:05 +02:00
|
|
|
|
HRESULT WINAPI CStdStubBuffer_Delegating_Construct(REFIID riid,
|
|
|
|
|
LPUNKNOWN pUnkServer,
|
|
|
|
|
PCInterfaceName name,
|
|
|
|
|
CInterfaceStubVtbl *vtbl,
|
|
|
|
|
REFIID delegating_iid,
|
|
|
|
|
LPPSFACTORYBUFFER pPSFactory,
|
|
|
|
|
LPRPCSTUBBUFFER *ppStub)
|
|
|
|
|
{
|
|
|
|
|
cstdstubbuffer_delegating_t *This;
|
|
|
|
|
IUnknown *pvServer;
|
|
|
|
|
HRESULT r;
|
|
|
|
|
|
|
|
|
|
TRACE("(%p,%p,%p,%p) %s\n", pUnkServer, vtbl, pPSFactory, ppStub, name);
|
|
|
|
|
TRACE("iid=%s delegating to %s\n", debugstr_guid(vtbl->header.piid), debugstr_guid(delegating_iid));
|
|
|
|
|
TRACE("vtbl=%p\n", &vtbl->Vtbl);
|
|
|
|
|
|
|
|
|
|
if (!IsEqualGUID(vtbl->header.piid, riid))
|
|
|
|
|
{
|
|
|
|
|
ERR("IID mismatch during stub creation\n");
|
|
|
|
|
return RPC_E_UNEXPECTED;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
r = IUnknown_QueryInterface(pUnkServer, riid, (void**)&pvServer);
|
|
|
|
|
if(FAILED(r)) return r;
|
|
|
|
|
|
|
|
|
|
This = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(*This));
|
|
|
|
|
if (!This)
|
|
|
|
|
{
|
|
|
|
|
IUnknown_Release(pvServer);
|
|
|
|
|
return E_OUTOFMEMORY;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
This->base_obj = get_delegating_vtbl();
|
|
|
|
|
r = create_stub(delegating_iid, (IUnknown*)&This->base_obj, &This->base_stub);
|
|
|
|
|
if(FAILED(r))
|
|
|
|
|
{
|
|
|
|
|
release_delegating_vtbl(This->base_obj);
|
|
|
|
|
HeapFree(GetProcessHeap(), 0, This);
|
|
|
|
|
IUnknown_Release(pvServer);
|
|
|
|
|
return r;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
This->stub_buffer.lpVtbl = &vtbl->Vtbl;
|
|
|
|
|
This->stub_buffer.RefCount = 1;
|
|
|
|
|
This->stub_buffer.pvServerObject = pvServer;
|
|
|
|
|
This->stub_buffer.pPSFactory = pPSFactory;
|
|
|
|
|
*ppStub = (LPRPCSTUBBUFFER)&This->stub_buffer;
|
|
|
|
|
|
|
|
|
|
IPSFactoryBuffer_AddRef(pPSFactory);
|
|
|
|
|
return S_OK;
|
|
|
|
|
}
|
|
|
|
|
|
2002-10-11 00:46:08 +02:00
|
|
|
|
HRESULT WINAPI CStdStubBuffer_QueryInterface(LPRPCSTUBBUFFER iface,
|
|
|
|
|
REFIID riid,
|
|
|
|
|
LPVOID *obj)
|
|
|
|
|
{
|
2004-09-06 22:34:29 +02:00
|
|
|
|
CStdStubBuffer *This = (CStdStubBuffer *)iface;
|
2002-10-11 00:46:08 +02:00
|
|
|
|
TRACE("(%p)->QueryInterface(%s,%p)\n",This,debugstr_guid(riid),obj);
|
|
|
|
|
|
2006-08-30 15:57:20 +02:00
|
|
|
|
if (IsEqualIID(&IID_IUnknown, riid) ||
|
|
|
|
|
IsEqualIID(&IID_IRpcStubBuffer, riid))
|
|
|
|
|
{
|
|
|
|
|
IUnknown_AddRef(iface);
|
|
|
|
|
*obj = iface;
|
2002-10-11 00:46:08 +02:00
|
|
|
|
return S_OK;
|
|
|
|
|
}
|
2006-08-30 15:57:20 +02:00
|
|
|
|
*obj = NULL;
|
2002-10-11 00:46:08 +02:00
|
|
|
|
return E_NOINTERFACE;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
ULONG WINAPI CStdStubBuffer_AddRef(LPRPCSTUBBUFFER iface)
|
|
|
|
|
{
|
2004-09-06 22:34:29 +02:00
|
|
|
|
CStdStubBuffer *This = (CStdStubBuffer *)iface;
|
2002-10-11 00:46:08 +02:00
|
|
|
|
TRACE("(%p)->AddRef()\n",This);
|
2006-08-30 16:16:29 +02:00
|
|
|
|
return InterlockedIncrement(&This->RefCount);
|
2002-10-11 00:46:08 +02:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
ULONG WINAPI NdrCStdStubBuffer_Release(LPRPCSTUBBUFFER iface,
|
|
|
|
|
LPPSFACTORYBUFFER pPSF)
|
|
|
|
|
{
|
2004-09-06 22:34:29 +02:00
|
|
|
|
CStdStubBuffer *This = (CStdStubBuffer *)iface;
|
2006-08-30 16:16:29 +02:00
|
|
|
|
ULONG refs;
|
|
|
|
|
|
2002-10-11 00:46:08 +02:00
|
|
|
|
TRACE("(%p)->Release()\n",This);
|
|
|
|
|
|
2006-08-30 16:16:29 +02:00
|
|
|
|
refs = InterlockedDecrement(&This->RefCount);
|
|
|
|
|
if (!refs)
|
|
|
|
|
{
|
2006-08-30 18:20:17 +02:00
|
|
|
|
/* test_Release shows that native doesn't call Disconnect here.
|
|
|
|
|
We'll leave it in for the time being. */
|
2005-07-19 21:12:54 +02:00
|
|
|
|
IRpcStubBuffer_Disconnect(iface);
|
2006-08-30 18:20:17 +02:00
|
|
|
|
|
|
|
|
|
IPSFactoryBuffer_Release(pPSF);
|
2002-10-11 00:46:08 +02:00
|
|
|
|
HeapFree(GetProcessHeap(),0,This);
|
|
|
|
|
}
|
2006-08-30 16:16:29 +02:00
|
|
|
|
return refs;
|
2002-10-11 00:46:08 +02:00
|
|
|
|
}
|
|
|
|
|
|
2005-10-26 15:56:00 +02:00
|
|
|
|
ULONG WINAPI NdrCStdStubBuffer2_Release(LPRPCSTUBBUFFER iface,
|
|
|
|
|
LPPSFACTORYBUFFER pPSF)
|
|
|
|
|
{
|
2006-08-31 16:40:05 +02:00
|
|
|
|
cstdstubbuffer_delegating_t *This = impl_from_delegating( iface );
|
|
|
|
|
ULONG refs;
|
|
|
|
|
|
|
|
|
|
TRACE("(%p)->Release()\n", This);
|
|
|
|
|
|
|
|
|
|
refs = InterlockedDecrement(&This->stub_buffer.RefCount);
|
|
|
|
|
if (!refs)
|
|
|
|
|
{
|
|
|
|
|
/* Just like NdrCStdStubBuffer_Release, we shouldn't call
|
|
|
|
|
Disconnect here */
|
|
|
|
|
IRpcStubBuffer_Disconnect((IRpcStubBuffer *)&This->stub_buffer);
|
|
|
|
|
|
|
|
|
|
IRpcStubBuffer_Release(This->base_stub);
|
|
|
|
|
release_delegating_vtbl(This->base_obj);
|
|
|
|
|
|
|
|
|
|
IPSFactoryBuffer_Release(pPSF);
|
|
|
|
|
HeapFree(GetProcessHeap(), 0, This);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return refs;
|
2005-10-26 15:56:00 +02:00
|
|
|
|
}
|
|
|
|
|
|
2002-10-11 00:46:08 +02:00
|
|
|
|
HRESULT WINAPI CStdStubBuffer_Connect(LPRPCSTUBBUFFER iface,
|
|
|
|
|
LPUNKNOWN lpUnkServer)
|
|
|
|
|
{
|
2006-08-23 17:37:48 +02:00
|
|
|
|
CStdStubBuffer *This = (CStdStubBuffer *)iface;
|
|
|
|
|
HRESULT r;
|
|
|
|
|
IUnknown *new = NULL;
|
|
|
|
|
|
|
|
|
|
TRACE("(%p)->Connect(%p)\n",This,lpUnkServer);
|
|
|
|
|
|
|
|
|
|
r = IUnknown_QueryInterface(lpUnkServer, STUB_HEADER(This).piid, (void**)&new);
|
|
|
|
|
new = InterlockedExchangePointer((void**)&This->pvServerObject, new);
|
|
|
|
|
if(new)
|
|
|
|
|
IUnknown_Release(new);
|
|
|
|
|
return r;
|
2002-10-11 00:46:08 +02:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
void WINAPI CStdStubBuffer_Disconnect(LPRPCSTUBBUFFER iface)
|
|
|
|
|
{
|
2006-08-23 17:36:43 +02:00
|
|
|
|
CStdStubBuffer *This = (CStdStubBuffer *)iface;
|
|
|
|
|
IUnknown *old;
|
|
|
|
|
TRACE("(%p)->Disconnect()\n",This);
|
|
|
|
|
|
|
|
|
|
old = InterlockedExchangePointer((void**)&This->pvServerObject, NULL);
|
|
|
|
|
|
|
|
|
|
if(old)
|
|
|
|
|
IUnknown_Release(old);
|
2002-10-11 00:46:08 +02:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
HRESULT WINAPI CStdStubBuffer_Invoke(LPRPCSTUBBUFFER iface,
|
|
|
|
|
PRPCOLEMESSAGE pMsg,
|
|
|
|
|
LPRPCCHANNELBUFFER pChannel)
|
|
|
|
|
{
|
2004-09-06 22:34:29 +02:00
|
|
|
|
CStdStubBuffer *This = (CStdStubBuffer *)iface;
|
2002-10-11 00:46:08 +02:00
|
|
|
|
DWORD dwPhase = STUB_UNMARSHAL;
|
2006-06-10 13:33:40 +02:00
|
|
|
|
HRESULT hr = S_OK;
|
|
|
|
|
|
2002-10-11 00:46:08 +02:00
|
|
|
|
TRACE("(%p)->Invoke(%p,%p)\n",This,pMsg,pChannel);
|
|
|
|
|
|
2006-06-10 13:33:40 +02:00
|
|
|
|
__TRY
|
|
|
|
|
{
|
|
|
|
|
if (STUB_HEADER(This).pDispatchTable)
|
|
|
|
|
STUB_HEADER(This).pDispatchTable[pMsg->iMethod](iface, pChannel, (PRPC_MESSAGE)pMsg, &dwPhase);
|
|
|
|
|
else /* pure interpreted */
|
|
|
|
|
NdrStubCall2(iface, pChannel, (PRPC_MESSAGE)pMsg, &dwPhase);
|
|
|
|
|
}
|
|
|
|
|
__EXCEPT(stub_filter)
|
|
|
|
|
{
|
|
|
|
|
DWORD dwExceptionCode = GetExceptionCode();
|
2006-11-09 23:04:48 +01:00
|
|
|
|
WARN("a stub call failed with exception 0x%08x (%d)\n", dwExceptionCode, dwExceptionCode);
|
2006-06-10 13:33:40 +02:00
|
|
|
|
if (FAILED(dwExceptionCode))
|
|
|
|
|
hr = dwExceptionCode;
|
|
|
|
|
else
|
|
|
|
|
hr = HRESULT_FROM_WIN32(dwExceptionCode);
|
|
|
|
|
}
|
|
|
|
|
__ENDTRY
|
|
|
|
|
|
|
|
|
|
return hr;
|
2002-10-11 00:46:08 +02:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
LPRPCSTUBBUFFER WINAPI CStdStubBuffer_IsIIDSupported(LPRPCSTUBBUFFER iface,
|
|
|
|
|
REFIID riid)
|
|
|
|
|
{
|
2004-09-06 22:34:29 +02:00
|
|
|
|
CStdStubBuffer *This = (CStdStubBuffer *)iface;
|
2002-10-11 00:46:08 +02:00
|
|
|
|
TRACE("(%p)->IsIIDSupported(%s)\n",This,debugstr_guid(riid));
|
|
|
|
|
return IsEqualGUID(STUB_HEADER(This).piid, riid) ? iface : NULL;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
ULONG WINAPI CStdStubBuffer_CountRefs(LPRPCSTUBBUFFER iface)
|
|
|
|
|
{
|
2004-09-06 22:34:29 +02:00
|
|
|
|
CStdStubBuffer *This = (CStdStubBuffer *)iface;
|
2002-10-11 00:46:08 +02:00
|
|
|
|
TRACE("(%p)->CountRefs()\n",This);
|
|
|
|
|
return This->RefCount;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
HRESULT WINAPI CStdStubBuffer_DebugServerQueryInterface(LPRPCSTUBBUFFER iface,
|
|
|
|
|
LPVOID *ppv)
|
|
|
|
|
{
|
2004-09-06 22:34:29 +02:00
|
|
|
|
CStdStubBuffer *This = (CStdStubBuffer *)iface;
|
2002-10-11 00:46:08 +02:00
|
|
|
|
TRACE("(%p)->DebugServerQueryInterface(%p)\n",This,ppv);
|
|
|
|
|
return S_OK;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
void WINAPI CStdStubBuffer_DebugServerRelease(LPRPCSTUBBUFFER iface,
|
|
|
|
|
LPVOID pv)
|
|
|
|
|
{
|
2004-09-06 22:34:29 +02:00
|
|
|
|
CStdStubBuffer *This = (CStdStubBuffer *)iface;
|
2002-10-11 00:46:08 +02:00
|
|
|
|
TRACE("(%p)->DebugServerRelease(%p)\n",This,pv);
|
|
|
|
|
}
|
2005-11-29 10:42:04 +01:00
|
|
|
|
|
2006-01-30 14:56:07 +01:00
|
|
|
|
const IRpcStubBufferVtbl CStdStubBuffer_Vtbl =
|
|
|
|
|
{
|
|
|
|
|
CStdStubBuffer_QueryInterface,
|
|
|
|
|
CStdStubBuffer_AddRef,
|
|
|
|
|
NULL,
|
|
|
|
|
CStdStubBuffer_Connect,
|
|
|
|
|
CStdStubBuffer_Disconnect,
|
|
|
|
|
CStdStubBuffer_Invoke,
|
|
|
|
|
CStdStubBuffer_IsIIDSupported,
|
|
|
|
|
CStdStubBuffer_CountRefs,
|
|
|
|
|
CStdStubBuffer_DebugServerQueryInterface,
|
|
|
|
|
CStdStubBuffer_DebugServerRelease
|
|
|
|
|
};
|
|
|
|
|
|
2006-08-31 16:40:05 +02:00
|
|
|
|
static HRESULT WINAPI CStdStubBuffer_Delegating_Connect(LPRPCSTUBBUFFER iface,
|
|
|
|
|
LPUNKNOWN lpUnkServer)
|
|
|
|
|
{
|
|
|
|
|
cstdstubbuffer_delegating_t *This = impl_from_delegating(iface);
|
|
|
|
|
HRESULT r;
|
|
|
|
|
TRACE("(%p)->Connect(%p)\n", This, lpUnkServer);
|
|
|
|
|
|
|
|
|
|
r = CStdStubBuffer_Connect(iface, lpUnkServer);
|
|
|
|
|
if(SUCCEEDED(r))
|
|
|
|
|
r = IRpcStubBuffer_Connect(This->base_stub, (IUnknown*)&This->base_obj);
|
|
|
|
|
|
|
|
|
|
return r;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
static void WINAPI CStdStubBuffer_Delegating_Disconnect(LPRPCSTUBBUFFER iface)
|
|
|
|
|
{
|
|
|
|
|
cstdstubbuffer_delegating_t *This = impl_from_delegating(iface);
|
|
|
|
|
TRACE("(%p)->Disconnect()\n", This);
|
|
|
|
|
|
|
|
|
|
IRpcStubBuffer_Disconnect(This->base_stub);
|
|
|
|
|
CStdStubBuffer_Disconnect(iface);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
static ULONG WINAPI CStdStubBuffer_Delegating_CountRefs(LPRPCSTUBBUFFER iface)
|
|
|
|
|
{
|
|
|
|
|
cstdstubbuffer_delegating_t *This = impl_from_delegating(iface);
|
|
|
|
|
ULONG ret;
|
|
|
|
|
TRACE("(%p)->CountRefs()\n", This);
|
|
|
|
|
|
|
|
|
|
ret = CStdStubBuffer_CountRefs(iface);
|
|
|
|
|
ret += IRpcStubBuffer_CountRefs(This->base_stub);
|
|
|
|
|
|
|
|
|
|
return ret;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
const IRpcStubBufferVtbl CStdStubBuffer_Delegating_Vtbl =
|
|
|
|
|
{
|
|
|
|
|
CStdStubBuffer_QueryInterface,
|
|
|
|
|
CStdStubBuffer_AddRef,
|
|
|
|
|
NULL,
|
|
|
|
|
CStdStubBuffer_Delegating_Connect,
|
|
|
|
|
CStdStubBuffer_Delegating_Disconnect,
|
|
|
|
|
CStdStubBuffer_Invoke,
|
|
|
|
|
CStdStubBuffer_IsIIDSupported,
|
|
|
|
|
CStdStubBuffer_Delegating_CountRefs,
|
|
|
|
|
CStdStubBuffer_DebugServerQueryInterface,
|
|
|
|
|
CStdStubBuffer_DebugServerRelease
|
|
|
|
|
};
|
|
|
|
|
|
2005-11-29 10:42:04 +01:00
|
|
|
|
const MIDL_SERVER_INFO *CStdStubBuffer_GetServerInfo(IRpcStubBuffer *iface)
|
|
|
|
|
{
|
|
|
|
|
CStdStubBuffer *This = (CStdStubBuffer *)iface;
|
|
|
|
|
return STUB_HEADER(This).pServerInfo;
|
|
|
|
|
}
|
2006-08-23 14:56:04 +02:00
|
|
|
|
|
|
|
|
|
/************************************************************************
|
|
|
|
|
* NdrStubForwardingFunction [RPCRT4.@]
|
|
|
|
|
*/
|
2006-08-31 15:58:26 +02:00
|
|
|
|
void __RPC_STUB NdrStubForwardingFunction( IRpcStubBuffer *iface, IRpcChannelBuffer *pChannel,
|
2006-08-23 14:56:04 +02:00
|
|
|
|
PRPC_MESSAGE pMsg, DWORD *pdwStubPhase )
|
|
|
|
|
{
|
2006-08-31 16:40:05 +02:00
|
|
|
|
/* Note pMsg is passed intact since RPCOLEMESSAGE is basically a RPC_MESSAGE. */
|
2006-08-23 14:56:04 +02:00
|
|
|
|
|
2006-08-31 15:58:26 +02:00
|
|
|
|
cstdstubbuffer_delegating_t *This = impl_from_delegating(iface);
|
|
|
|
|
HRESULT r = IRpcStubBuffer_Invoke(This->base_stub, (RPCOLEMESSAGE*)pMsg, pChannel);
|
2006-08-23 14:56:04 +02:00
|
|
|
|
if(FAILED(r)) RpcRaiseException(r);
|
|
|
|
|
return;
|
|
|
|
|
}
|
2006-08-29 22:12:42 +02:00
|
|
|
|
|
|
|
|
|
/***********************************************************************
|
|
|
|
|
* NdrStubInitialize [RPCRT4.@]
|
|
|
|
|
*/
|
|
|
|
|
void WINAPI NdrStubInitialize(PRPC_MESSAGE pRpcMsg,
|
|
|
|
|
PMIDL_STUB_MESSAGE pStubMsg,
|
|
|
|
|
PMIDL_STUB_DESC pStubDescriptor,
|
|
|
|
|
LPRPCCHANNELBUFFER pRpcChannelBuffer)
|
|
|
|
|
{
|
|
|
|
|
TRACE("(%p,%p,%p,%p)\n", pRpcMsg, pStubMsg, pStubDescriptor, pRpcChannelBuffer);
|
|
|
|
|
NdrServerInitializeNew(pRpcMsg, pStubMsg, pStubDescriptor);
|
|
|
|
|
pStubMsg->pRpcChannelBuffer = pRpcChannelBuffer;
|
2007-12-03 17:08:31 +01:00
|
|
|
|
IRpcChannelBuffer_GetDestCtx(pStubMsg->pRpcChannelBuffer,
|
|
|
|
|
&pStubMsg->dwDestContext,
|
|
|
|
|
&pStubMsg->pvDestContext);
|
2006-08-29 22:12:42 +02:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/***********************************************************************
|
|
|
|
|
* NdrStubGetBuffer [RPCRT4.@]
|
|
|
|
|
*/
|
2006-08-29 22:14:42 +02:00
|
|
|
|
void WINAPI NdrStubGetBuffer(LPRPCSTUBBUFFER iface,
|
2006-08-29 22:12:42 +02:00
|
|
|
|
LPRPCCHANNELBUFFER pRpcChannelBuffer,
|
|
|
|
|
PMIDL_STUB_MESSAGE pStubMsg)
|
|
|
|
|
{
|
2006-08-29 22:14:42 +02:00
|
|
|
|
CStdStubBuffer *This = (CStdStubBuffer *)iface;
|
|
|
|
|
HRESULT hr;
|
|
|
|
|
|
|
|
|
|
TRACE("(%p, %p, %p)\n", This, pRpcChannelBuffer, pStubMsg);
|
|
|
|
|
|
2006-08-29 22:12:42 +02:00
|
|
|
|
pStubMsg->RpcMsg->BufferLength = pStubMsg->BufferLength;
|
2006-08-29 22:14:42 +02:00
|
|
|
|
hr = IRpcChannelBuffer_GetBuffer(pRpcChannelBuffer,
|
|
|
|
|
(RPCOLEMESSAGE *)pStubMsg->RpcMsg, STUB_HEADER(This).piid);
|
|
|
|
|
if (FAILED(hr))
|
|
|
|
|
{
|
|
|
|
|
RpcRaiseException(hr);
|
|
|
|
|
return;
|
|
|
|
|
}
|
|
|
|
|
|
2007-12-13 17:11:37 +01:00
|
|
|
|
pStubMsg->Buffer = pStubMsg->RpcMsg->Buffer;
|
2006-08-29 22:12:42 +02:00
|
|
|
|
}
|