1023 lines
28 KiB
C
1023 lines
28 KiB
C
/*
|
|
* OLE32 proxy/stub handler
|
|
*
|
|
* Copyright 2002 Marcus Meissner
|
|
* Copyright 2001 Ove Kå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
|
|
* Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
|
|
*/
|
|
|
|
/* Documentation on MSDN:
|
|
*
|
|
* (Top level COM documentation)
|
|
* http://msdn.microsoft.com/library/default.asp?url=/library/en-us/dnanchor/html/componentdevelopmentank.asp
|
|
*
|
|
* (COM Proxy)
|
|
* http://msdn.microsoft.com/library/default.asp?url=/library/en-us/com/htm/comext_1q0p.asp
|
|
*
|
|
* (COM Stub)
|
|
* http://msdn.microsoft.com/library/default.asp?url=/library/en-us/com/htm/comext_1lia.asp
|
|
*
|
|
* (Marshal)
|
|
* http://msdn.microsoft.com/library/default.asp?url=/library/en-us/com/htm/comext_1gfn.asp
|
|
*
|
|
*/
|
|
|
|
#include "config.h"
|
|
|
|
#include <stdlib.h>
|
|
#include <stdarg.h>
|
|
#include <stdio.h>
|
|
#include <string.h>
|
|
|
|
#define COBJMACROS
|
|
#define NONAMELESSUNION
|
|
#define NONAMELESSSTRUCT
|
|
|
|
#include "windef.h"
|
|
#include "winbase.h"
|
|
#include "winuser.h"
|
|
#include "objbase.h"
|
|
#include "ole2.h"
|
|
#include "rpc.h"
|
|
#include "winerror.h"
|
|
#include "winreg.h"
|
|
#include "wtypes.h"
|
|
|
|
#include "compobj_private.h"
|
|
#include "moniker.h"
|
|
|
|
#include "wine/debug.h"
|
|
|
|
WINE_DEFAULT_DEBUG_CHANNEL(ole);
|
|
|
|
const CLSID CLSID_DfMarshal = { 0x0000030b, 0, 0, {0xc0, 0, 0, 0, 0, 0, 0, 0x46} };
|
|
const CLSID CLSID_PSFactoryBuffer = { 0x00000320, 0, 0, {0xc0, 0, 0, 0, 0, 0, 0, 0x46} };
|
|
|
|
/* From: http://msdn.microsoft.com/library/en-us/com/cmi_m_4lda.asp
|
|
*
|
|
* The first time a client requests a pointer to an interface on a
|
|
* particular object, COM loads an IClassFactory stub in the server
|
|
* process and uses it to marshal the first pointer back to the
|
|
* client. In the client process, COM loads the generic proxy for the
|
|
* class factory object and calls its implementation of IMarshal to
|
|
* unmarshal that first pointer. COM then creates the first interface
|
|
* proxy and hands it a pointer to the RPC channel. Finally, COM returns
|
|
* the IClassFactory pointer to the client, which uses it to call
|
|
* IClassFactory::CreateInstance, passing it a reference to the interface.
|
|
*
|
|
* Back in the server process, COM now creates a new instance of the
|
|
* object, along with a stub for the requested interface. This stub marshals
|
|
* the interface pointer back to the client process, where another object
|
|
* proxy is created, this time for the object itself. Also created is a
|
|
* proxy for the requested interface, a pointer to which is returned to
|
|
* the client. With subsequent calls to other interfaces on the object,
|
|
* COM will load the appropriate interface stubs and proxies as needed.
|
|
*/
|
|
typedef struct _CFStub {
|
|
const IRpcStubBufferVtbl *lpvtbl;
|
|
LONG ref;
|
|
|
|
LPUNKNOWN pUnkServer;
|
|
} CFStub;
|
|
|
|
static HRESULT WINAPI
|
|
CFStub_QueryInterface(LPRPCSTUBBUFFER iface, REFIID riid, LPVOID *ppv) {
|
|
if (IsEqualIID(&IID_IUnknown,riid)||IsEqualIID(&IID_IRpcStubBuffer,riid)) {
|
|
*ppv = (LPVOID)iface;
|
|
IUnknown_AddRef(iface);
|
|
return S_OK;
|
|
}
|
|
FIXME("(%s), interface not supported.\n",debugstr_guid(riid));
|
|
return E_NOINTERFACE;
|
|
}
|
|
|
|
static ULONG WINAPI
|
|
CFStub_AddRef(LPRPCSTUBBUFFER iface) {
|
|
CFStub *This = (CFStub *)iface;
|
|
return InterlockedIncrement(&This->ref);
|
|
}
|
|
|
|
static ULONG WINAPI
|
|
CFStub_Release(LPRPCSTUBBUFFER iface) {
|
|
CFStub *This = (CFStub *)iface;
|
|
ULONG ref;
|
|
|
|
ref = InterlockedDecrement(&This->ref);
|
|
if (!ref) {
|
|
IRpcStubBuffer_Disconnect(iface);
|
|
HeapFree(GetProcessHeap(),0,This);
|
|
}
|
|
return ref;
|
|
}
|
|
|
|
static HRESULT WINAPI
|
|
CFStub_Connect(LPRPCSTUBBUFFER iface, IUnknown *pUnkServer) {
|
|
CFStub *This = (CFStub *)iface;
|
|
|
|
This->pUnkServer = pUnkServer;
|
|
IUnknown_AddRef(pUnkServer);
|
|
return S_OK;
|
|
}
|
|
|
|
static void WINAPI
|
|
CFStub_Disconnect(LPRPCSTUBBUFFER iface) {
|
|
CFStub *This = (CFStub *)iface;
|
|
|
|
if (This->pUnkServer) {
|
|
IUnknown_Release(This->pUnkServer);
|
|
This->pUnkServer = NULL;
|
|
}
|
|
}
|
|
|
|
static HRESULT WINAPI
|
|
CFStub_Invoke(
|
|
LPRPCSTUBBUFFER iface,RPCOLEMESSAGE* msg,IRpcChannelBuffer* chanbuf
|
|
) {
|
|
CFStub *This = (CFStub *)iface;
|
|
HRESULT hres;
|
|
|
|
if (msg->iMethod == 3) { /* CreateInstance */
|
|
IID iid;
|
|
IClassFactory *classfac;
|
|
IUnknown *ppv;
|
|
IStream *pStm;
|
|
STATSTG ststg;
|
|
ULARGE_INTEGER newpos;
|
|
LARGE_INTEGER seekto;
|
|
ULONG res;
|
|
|
|
if (msg->cbBuffer < sizeof(IID)) {
|
|
FIXME("Not enough bytes in buffer (%ld instead of %d)?\n",msg->cbBuffer,sizeof(IID));
|
|
return E_FAIL;
|
|
}
|
|
memcpy(&iid,msg->Buffer,sizeof(iid));
|
|
TRACE("->CreateInstance(%s)\n",debugstr_guid(&iid));
|
|
hres = IUnknown_QueryInterface(This->pUnkServer,&IID_IClassFactory,(LPVOID*)&classfac);
|
|
if (hres) {
|
|
FIXME("Ole server does not provide an IClassFactory?\n");
|
|
return hres;
|
|
}
|
|
hres = IClassFactory_CreateInstance(classfac,NULL,&iid,(LPVOID*)&ppv);
|
|
IClassFactory_Release(classfac);
|
|
if (hres) {
|
|
msg->cbBuffer = 0;
|
|
FIXME("Failed to create an instance of %s\n",debugstr_guid(&iid));
|
|
return hres;
|
|
}
|
|
hres = CreateStreamOnHGlobal(0,TRUE,&pStm);
|
|
if (hres) {
|
|
FIXME("Failed to create stream on hglobal\n");
|
|
return hres;
|
|
}
|
|
hres = CoMarshalInterface(pStm,&iid,ppv,0,NULL,0);
|
|
IUnknown_Release((IUnknown*)ppv);
|
|
if (hres) {
|
|
FIXME("CoMarshalInterface failed, %lx!\n",hres);
|
|
msg->cbBuffer = 0;
|
|
return hres;
|
|
}
|
|
hres = IStream_Stat(pStm,&ststg,0);
|
|
if (hres) {
|
|
FIXME("Stat failed.\n");
|
|
return hres;
|
|
}
|
|
|
|
msg->cbBuffer = ststg.cbSize.u.LowPart;
|
|
|
|
I_RpcGetBuffer((RPC_MESSAGE *)msg);
|
|
if (hres) return hres;
|
|
|
|
seekto.u.LowPart = 0;seekto.u.HighPart = 0;
|
|
hres = IStream_Seek(pStm,seekto,SEEK_SET,&newpos);
|
|
if (hres) {
|
|
FIXME("IStream_Seek failed, %lx\n",hres);
|
|
return hres;
|
|
}
|
|
hres = IStream_Read(pStm,msg->Buffer,msg->cbBuffer,&res);
|
|
if (hres) {
|
|
FIXME("Stream Read failed, %lx\n",hres);
|
|
return hres;
|
|
}
|
|
IStream_Release(pStm);
|
|
return S_OK;
|
|
}
|
|
FIXME("(%p,%p), stub!\n",msg,chanbuf);
|
|
FIXME("iMethod is %ld\n",msg->iMethod);
|
|
FIXME("cbBuffer is %ld\n",msg->cbBuffer);
|
|
return E_FAIL;
|
|
}
|
|
|
|
static LPRPCSTUBBUFFER WINAPI
|
|
CFStub_IsIIDSupported(LPRPCSTUBBUFFER iface,REFIID riid) {
|
|
FIXME("(%s), stub!\n",debugstr_guid(riid));
|
|
return NULL;
|
|
}
|
|
|
|
static ULONG WINAPI
|
|
CFStub_CountRefs(LPRPCSTUBBUFFER iface) {
|
|
FIXME("(), stub!\n");
|
|
return 1;
|
|
}
|
|
|
|
static HRESULT WINAPI
|
|
CFStub_DebugServerQueryInterface(LPRPCSTUBBUFFER iface,void** ppv) {
|
|
FIXME("(%p), stub!\n",ppv);
|
|
return E_FAIL;
|
|
}
|
|
static void WINAPI
|
|
CFStub_DebugServerRelease(LPRPCSTUBBUFFER iface,void *pv) {
|
|
FIXME("(%p), stub!\n",pv);
|
|
}
|
|
|
|
static const IRpcStubBufferVtbl cfstubvt = {
|
|
CFStub_QueryInterface,
|
|
CFStub_AddRef,
|
|
CFStub_Release,
|
|
CFStub_Connect,
|
|
CFStub_Disconnect,
|
|
CFStub_Invoke,
|
|
CFStub_IsIIDSupported,
|
|
CFStub_CountRefs,
|
|
CFStub_DebugServerQueryInterface,
|
|
CFStub_DebugServerRelease
|
|
};
|
|
|
|
static HRESULT
|
|
CFStub_Construct(LPRPCSTUBBUFFER *ppv) {
|
|
CFStub *cfstub;
|
|
cfstub = HeapAlloc(GetProcessHeap(),HEAP_ZERO_MEMORY,sizeof(CFStub));
|
|
if (!cfstub)
|
|
return E_OUTOFMEMORY;
|
|
*ppv = (LPRPCSTUBBUFFER)cfstub;
|
|
cfstub->lpvtbl = &cfstubvt;
|
|
cfstub->ref = 1;
|
|
return S_OK;
|
|
}
|
|
|
|
/* Since we create proxy buffers and classfactory in a pair, there is
|
|
* no need for 2 separate structs. Just put them in one, but remember
|
|
* the refcount.
|
|
*/
|
|
typedef struct _CFProxy {
|
|
const IClassFactoryVtbl *lpvtbl_cf;
|
|
const IRpcProxyBufferVtbl *lpvtbl_proxy;
|
|
LONG ref;
|
|
|
|
IRpcChannelBuffer *chanbuf;
|
|
IUnknown *outer_unknown;
|
|
} CFProxy;
|
|
|
|
static HRESULT WINAPI IRpcProxyBufferImpl_QueryInterface(LPRPCPROXYBUFFER iface,REFIID riid,LPVOID *ppv) {
|
|
*ppv = NULL;
|
|
if (IsEqualIID(riid,&IID_IRpcProxyBuffer)||IsEqualIID(riid,&IID_IUnknown)) {
|
|
IRpcProxyBuffer_AddRef(iface);
|
|
*ppv = (LPVOID)iface;
|
|
return S_OK;
|
|
}
|
|
FIXME("(%s), no interface.\n",debugstr_guid(riid));
|
|
return E_NOINTERFACE;
|
|
}
|
|
|
|
static ULONG WINAPI IRpcProxyBufferImpl_AddRef(LPRPCPROXYBUFFER iface) {
|
|
ICOM_THIS_MULTI(CFProxy,lpvtbl_proxy,iface);
|
|
return InterlockedIncrement(&This->ref);
|
|
}
|
|
|
|
static ULONG WINAPI IRpcProxyBufferImpl_Release(LPRPCPROXYBUFFER iface) {
|
|
ICOM_THIS_MULTI(CFProxy,lpvtbl_proxy,iface);
|
|
ULONG ref = InterlockedDecrement(&This->ref);
|
|
|
|
if (!ref) {
|
|
IRpcProxyBuffer_Disconnect(iface);
|
|
HeapFree(GetProcessHeap(),0,This);
|
|
}
|
|
return ref;
|
|
}
|
|
|
|
static HRESULT WINAPI IRpcProxyBufferImpl_Connect(LPRPCPROXYBUFFER iface,IRpcChannelBuffer* pRpcChannelBuffer) {
|
|
ICOM_THIS_MULTI(CFProxy,lpvtbl_proxy,iface);
|
|
|
|
This->chanbuf = pRpcChannelBuffer;
|
|
IRpcChannelBuffer_AddRef(This->chanbuf);
|
|
return S_OK;
|
|
}
|
|
static void WINAPI IRpcProxyBufferImpl_Disconnect(LPRPCPROXYBUFFER iface) {
|
|
ICOM_THIS_MULTI(CFProxy,lpvtbl_proxy,iface);
|
|
if (This->chanbuf) {
|
|
IRpcChannelBuffer_Release(This->chanbuf);
|
|
This->chanbuf = NULL;
|
|
}
|
|
}
|
|
|
|
static HRESULT WINAPI
|
|
CFProxy_QueryInterface(LPCLASSFACTORY iface,REFIID riid, LPVOID *ppv) {
|
|
ICOM_THIS_MULTI(CFProxy,lpvtbl_cf,iface);
|
|
if (This->outer_unknown) return IUnknown_QueryInterface(This->outer_unknown, riid, ppv);
|
|
*ppv = NULL;
|
|
if (IsEqualIID(&IID_IClassFactory,riid) || IsEqualIID(&IID_IUnknown,riid)) {
|
|
*ppv = (LPVOID)iface;
|
|
IClassFactory_AddRef(iface);
|
|
return S_OK;
|
|
}
|
|
if (IsEqualIID(riid,&IID_IMarshal)) /* just to avoid debug output */
|
|
return E_NOINTERFACE;
|
|
FIXME("Unhandled interface: %s\n",debugstr_guid(riid));
|
|
return E_NOINTERFACE;
|
|
}
|
|
|
|
static ULONG WINAPI CFProxy_AddRef(LPCLASSFACTORY iface) {
|
|
ICOM_THIS_MULTI(CFProxy,lpvtbl_cf,iface);
|
|
if (This->outer_unknown) return IUnknown_AddRef(This->outer_unknown);
|
|
return InterlockedIncrement(&This->ref);
|
|
}
|
|
|
|
static ULONG WINAPI CFProxy_Release(LPCLASSFACTORY iface) {
|
|
ULONG ref;
|
|
ICOM_THIS_MULTI(CFProxy,lpvtbl_cf,iface);
|
|
if (This->outer_unknown)
|
|
ref = IUnknown_Release(This->outer_unknown);
|
|
else
|
|
ref = InterlockedDecrement(&This->ref);
|
|
|
|
if (!ref) {
|
|
if (This->chanbuf) IRpcChannelBuffer_Release(This->chanbuf);
|
|
HeapFree(GetProcessHeap(),0,This);
|
|
}
|
|
return ref;
|
|
}
|
|
|
|
static HRESULT WINAPI CFProxy_CreateInstance(
|
|
LPCLASSFACTORY iface,
|
|
LPUNKNOWN pUnkOuter,/* [in] */
|
|
REFIID riid, /* [in] */
|
|
LPVOID *ppv /* [out] */
|
|
) {
|
|
ICOM_THIS_MULTI(CFProxy,lpvtbl_cf,iface);
|
|
HRESULT hres;
|
|
LPSTREAM pStream;
|
|
HGLOBAL hGlobal;
|
|
ULONG srstatus;
|
|
RPCOLEMESSAGE msg;
|
|
|
|
TRACE("(%p,%s,%p)\n",pUnkOuter,debugstr_guid(riid),ppv);
|
|
|
|
/* Send CreateInstance to the remote classfactory.
|
|
*
|
|
* Data: Only the 'IID'.
|
|
*/
|
|
msg.iMethod = 3;
|
|
msg.cbBuffer = sizeof(*riid);
|
|
msg.Buffer = NULL;
|
|
hres = IRpcChannelBuffer_GetBuffer(This->chanbuf,&msg,&IID_IClassFactory);
|
|
if (hres) {
|
|
FIXME("IRpcChannelBuffer_GetBuffer failed with %lx?\n",hres);
|
|
return hres;
|
|
}
|
|
memcpy(msg.Buffer,riid,sizeof(*riid));
|
|
hres = IRpcChannelBuffer_SendReceive(This->chanbuf,&msg,&srstatus);
|
|
if (hres) {
|
|
FIXME("IRpcChannelBuffer_SendReceive failed with %lx?\n",hres);
|
|
return hres;
|
|
}
|
|
|
|
if (!msg.cbBuffer) /* interface not found on remote */
|
|
return srstatus;
|
|
|
|
/* We got back: [Marshalled Interface data] */
|
|
TRACE("got %ld bytes data.\n",msg.cbBuffer);
|
|
hGlobal = GlobalAlloc(GMEM_MOVEABLE|GMEM_NODISCARD|GMEM_SHARE,msg.cbBuffer);
|
|
memcpy(GlobalLock(hGlobal),msg.Buffer,msg.cbBuffer);
|
|
hres = CreateStreamOnHGlobal(hGlobal,TRUE,&pStream);
|
|
if (hres) {
|
|
FIXME("CreateStreamOnHGlobal failed with %lx\n",hres);
|
|
return hres;
|
|
}
|
|
hres = CoUnmarshalInterface(
|
|
pStream,
|
|
riid,
|
|
ppv
|
|
);
|
|
IStream_Release(pStream); /* Does GlobalFree hGlobal too. */
|
|
if (hres) {
|
|
FIXME("CoMarshalInterface failed, %lx\n",hres);
|
|
return hres;
|
|
}
|
|
return S_OK;
|
|
}
|
|
|
|
static HRESULT WINAPI CFProxy_LockServer(LPCLASSFACTORY iface,BOOL fLock) {
|
|
/*ICOM_THIS_MULTI(CFProxy,lpvtbl_cf,iface);*/
|
|
FIXME("(%d), stub!\n",fLock);
|
|
/* basically: write BOOL, read empty */
|
|
return S_OK;
|
|
}
|
|
|
|
static const IRpcProxyBufferVtbl pspbvtbl = {
|
|
IRpcProxyBufferImpl_QueryInterface,
|
|
IRpcProxyBufferImpl_AddRef,
|
|
IRpcProxyBufferImpl_Release,
|
|
IRpcProxyBufferImpl_Connect,
|
|
IRpcProxyBufferImpl_Disconnect
|
|
};
|
|
static const IClassFactoryVtbl cfproxyvt = {
|
|
CFProxy_QueryInterface,
|
|
CFProxy_AddRef,
|
|
CFProxy_Release,
|
|
CFProxy_CreateInstance,
|
|
CFProxy_LockServer
|
|
};
|
|
|
|
static HRESULT
|
|
CFProxy_Construct(IUnknown *pUnkOuter, LPVOID *ppv,LPVOID *ppProxy) {
|
|
CFProxy *cf;
|
|
|
|
cf = HeapAlloc(GetProcessHeap(),HEAP_ZERO_MEMORY,sizeof(CFProxy));
|
|
if (!cf)
|
|
return E_OUTOFMEMORY;
|
|
|
|
cf->lpvtbl_cf = &cfproxyvt;
|
|
cf->lpvtbl_proxy = &pspbvtbl;
|
|
/* one reference for the proxy buffer */
|
|
cf->ref = 1;
|
|
cf->outer_unknown = pUnkOuter;
|
|
*ppv = &(cf->lpvtbl_cf);
|
|
*ppProxy = &(cf->lpvtbl_proxy);
|
|
/* and one reference for the object */
|
|
IUnknown_AddRef((IUnknown *)*ppv);
|
|
return S_OK;
|
|
}
|
|
|
|
|
|
/********************* IRemUnknown Proxy/Stub ********************************/
|
|
|
|
typedef struct
|
|
{
|
|
const IRpcStubBufferVtbl *lpVtbl;
|
|
LONG refs;
|
|
IRemUnknown *iface;
|
|
} RemUnkStub;
|
|
|
|
static HRESULT WINAPI RemUnkStub_QueryInterface(LPRPCSTUBBUFFER iface,
|
|
REFIID riid,
|
|
LPVOID *obj)
|
|
{
|
|
RemUnkStub *This = (RemUnkStub *)iface;
|
|
TRACE("(%p)->QueryInterface(%s,%p)\n",This,debugstr_guid(riid),obj);
|
|
if (IsEqualGUID(&IID_IUnknown,riid) ||
|
|
IsEqualGUID(&IID_IRpcStubBuffer,riid)) {
|
|
*obj = This;
|
|
return S_OK;
|
|
}
|
|
return E_NOINTERFACE;
|
|
}
|
|
|
|
static ULONG WINAPI RemUnkStub_AddRef(LPRPCSTUBBUFFER iface)
|
|
{
|
|
RemUnkStub *This = (RemUnkStub *)iface;
|
|
TRACE("(%p)->AddRef()\n",This);
|
|
return InterlockedIncrement(&This->refs);
|
|
}
|
|
|
|
static ULONG WINAPI RemUnkStub_Release(LPRPCSTUBBUFFER iface)
|
|
{
|
|
RemUnkStub *This = (RemUnkStub *)iface;
|
|
ULONG refs;
|
|
TRACE("(%p)->Release()\n",This);
|
|
refs = InterlockedDecrement(&This->refs);
|
|
if (!refs)
|
|
HeapFree(GetProcessHeap(), 0, This);
|
|
return refs;
|
|
}
|
|
|
|
static HRESULT WINAPI RemUnkStub_Connect(LPRPCSTUBBUFFER iface,
|
|
LPUNKNOWN lpUnkServer)
|
|
{
|
|
RemUnkStub *This = (RemUnkStub *)iface;
|
|
TRACE("(%p)->Connect(%p)\n",This,lpUnkServer);
|
|
This->iface = (IRemUnknown*)lpUnkServer;
|
|
IRemUnknown_AddRef(This->iface);
|
|
return S_OK;
|
|
}
|
|
|
|
static void WINAPI RemUnkStub_Disconnect(LPRPCSTUBBUFFER iface)
|
|
{
|
|
RemUnkStub *This = (RemUnkStub *)iface;
|
|
TRACE("(%p)->Disconnect()\n",This);
|
|
IUnknown_Release(This->iface);
|
|
This->iface = NULL;
|
|
}
|
|
|
|
static HRESULT WINAPI RemUnkStub_Invoke(LPRPCSTUBBUFFER iface,
|
|
PRPCOLEMESSAGE pMsg,
|
|
LPRPCCHANNELBUFFER pChannel)
|
|
{
|
|
RemUnkStub *This = (RemUnkStub *)iface;
|
|
ULONG iMethod = pMsg->iMethod;
|
|
LPBYTE buf = pMsg->Buffer;
|
|
HRESULT hr = RPC_E_INVALIDMETHOD;
|
|
|
|
TRACE("(%p)->Invoke(%p,%p) method %ld\n", This, pMsg, pChannel, iMethod);
|
|
switch (iMethod)
|
|
{
|
|
case 3: /* RemQueryInterface */
|
|
{
|
|
IPID ipid;
|
|
ULONG cRefs;
|
|
USHORT cIids;
|
|
IID *iids;
|
|
REMQIRESULT *pQIResults = NULL;
|
|
|
|
/* in */
|
|
memcpy(&ipid, buf, sizeof(ipid));
|
|
buf += sizeof(ipid);
|
|
memcpy(&cRefs, buf, sizeof(cRefs));
|
|
buf += sizeof(cRefs);
|
|
memcpy(&cIids, buf, sizeof(cIids));
|
|
buf += sizeof(cIids);
|
|
iids = (IID *)buf;
|
|
|
|
hr = IRemUnknown_RemQueryInterface(This->iface, &ipid, cRefs, cIids, iids, &pQIResults);
|
|
|
|
/* out */
|
|
pMsg->cbBuffer = cIids * sizeof(REMQIRESULT) + sizeof(HRESULT);
|
|
|
|
I_RpcGetBuffer((RPC_MESSAGE *)pMsg);
|
|
|
|
buf = pMsg->Buffer;
|
|
*(HRESULT *)buf = hr;
|
|
buf += sizeof(HRESULT);
|
|
|
|
if (hr) return hr;
|
|
/* FIXME: pQIResults is a unique pointer so pQIResults can be NULL! */
|
|
memcpy(buf, pQIResults, cIids * sizeof(REMQIRESULT));
|
|
|
|
break;
|
|
}
|
|
case 4: /* RemAddRef */
|
|
{
|
|
USHORT cIids;
|
|
REMINTERFACEREF *ir;
|
|
HRESULT *pResults;
|
|
|
|
/* in */
|
|
memcpy(&cIids, buf, sizeof(USHORT));
|
|
buf += sizeof(USHORT);
|
|
ir = (REMINTERFACEREF*)buf;
|
|
pResults = CoTaskMemAlloc(cIids * sizeof(HRESULT));
|
|
if (!pResults) return E_OUTOFMEMORY;
|
|
|
|
hr = IRemUnknown_RemAddRef(This->iface, cIids, ir, pResults);
|
|
|
|
/* out */
|
|
pMsg->cbBuffer = cIids * sizeof(HRESULT);
|
|
|
|
I_RpcGetBuffer((RPC_MESSAGE *)pMsg);
|
|
if (!hr)
|
|
{
|
|
buf = pMsg->Buffer;
|
|
memcpy(buf, pResults, cIids * sizeof(HRESULT));
|
|
}
|
|
|
|
CoTaskMemFree(pResults);
|
|
|
|
break;
|
|
}
|
|
case 5: /* RemRelease */
|
|
{
|
|
USHORT cIids;
|
|
REMINTERFACEREF *ir;
|
|
|
|
/* in */
|
|
memcpy(&cIids, buf, sizeof(USHORT));
|
|
buf += sizeof(USHORT);
|
|
ir = (REMINTERFACEREF*)buf;
|
|
|
|
hr = IRemUnknown_RemRelease(This->iface, cIids, ir);
|
|
|
|
/* out */
|
|
pMsg->cbBuffer = 0;
|
|
break;
|
|
}
|
|
}
|
|
return hr;
|
|
}
|
|
|
|
static LPRPCSTUBBUFFER WINAPI RemUnkStub_IsIIDSupported(LPRPCSTUBBUFFER iface,
|
|
REFIID riid)
|
|
{
|
|
RemUnkStub *This = (RemUnkStub *)iface;
|
|
TRACE("(%p)->IsIIDSupported(%s)\n", This, debugstr_guid(riid));
|
|
return IsEqualGUID(&IID_IRemUnknown, riid) ? iface : NULL;
|
|
}
|
|
|
|
static ULONG WINAPI RemUnkStub_CountRefs(LPRPCSTUBBUFFER iface)
|
|
{
|
|
RemUnkStub *This = (RemUnkStub *)iface;
|
|
FIXME("(%p)->CountRefs()\n", This);
|
|
return 1;
|
|
}
|
|
|
|
static HRESULT WINAPI RemUnkStub_DebugServerQueryInterface(LPRPCSTUBBUFFER iface,
|
|
LPVOID *ppv)
|
|
{
|
|
RemUnkStub *This = (RemUnkStub *)iface;
|
|
FIXME("(%p)->DebugServerQueryInterface(%p)\n",This,ppv);
|
|
return E_NOINTERFACE;
|
|
}
|
|
|
|
static void WINAPI RemUnkStub_DebugServerRelease(LPRPCSTUBBUFFER iface,
|
|
LPVOID pv)
|
|
{
|
|
RemUnkStub *This = (RemUnkStub *)iface;
|
|
FIXME("(%p)->DebugServerRelease(%p)\n", This, pv);
|
|
}
|
|
|
|
static const IRpcStubBufferVtbl RemUnkStub_VTable =
|
|
{
|
|
RemUnkStub_QueryInterface,
|
|
RemUnkStub_AddRef,
|
|
RemUnkStub_Release,
|
|
RemUnkStub_Connect,
|
|
RemUnkStub_Disconnect,
|
|
RemUnkStub_Invoke,
|
|
RemUnkStub_IsIIDSupported,
|
|
RemUnkStub_CountRefs,
|
|
RemUnkStub_DebugServerQueryInterface,
|
|
RemUnkStub_DebugServerRelease
|
|
};
|
|
|
|
static HRESULT RemUnkStub_Construct(IRpcStubBuffer **ppStub)
|
|
{
|
|
RemUnkStub *This = HeapAlloc(GetProcessHeap(), 0, sizeof(*This));
|
|
if (!This) return E_OUTOFMEMORY;
|
|
This->lpVtbl = &RemUnkStub_VTable;
|
|
This->refs = 1;
|
|
This->iface = NULL;
|
|
*ppStub = (IRpcStubBuffer*)This;
|
|
return S_OK;
|
|
}
|
|
|
|
|
|
typedef struct _RemUnkProxy {
|
|
const IRemUnknownVtbl *lpvtbl_remunk;
|
|
const IRpcProxyBufferVtbl *lpvtbl_proxy;
|
|
LONG refs;
|
|
|
|
IRpcChannelBuffer *chan;
|
|
IUnknown *outer_unknown;
|
|
} RemUnkProxy;
|
|
|
|
static HRESULT WINAPI RemUnkProxy_QueryInterface(LPREMUNKNOWN iface, REFIID riid, void **ppv)
|
|
{
|
|
RemUnkProxy *This = (RemUnkProxy *)iface;
|
|
if (This->outer_unknown)
|
|
return IUnknown_QueryInterface(This->outer_unknown, riid, ppv);
|
|
if (IsEqualIID(riid, &IID_IUnknown) ||
|
|
IsEqualIID(riid, &IID_IRemUnknown))
|
|
{
|
|
IRemUnknown_AddRef(iface);
|
|
*ppv = (LPVOID)iface;
|
|
return S_OK;
|
|
}
|
|
return E_NOINTERFACE;
|
|
}
|
|
|
|
static ULONG WINAPI RemUnkProxy_AddRef(LPREMUNKNOWN iface)
|
|
{
|
|
RemUnkProxy *This = (RemUnkProxy *)iface;
|
|
ULONG refs;
|
|
|
|
TRACE("(%p)->AddRef()\n",This);
|
|
|
|
if (This->outer_unknown)
|
|
refs = IUnknown_AddRef(This->outer_unknown);
|
|
else
|
|
refs = InterlockedIncrement(&This->refs);
|
|
return refs;
|
|
}
|
|
|
|
static ULONG WINAPI RemUnkProxy_Release(LPREMUNKNOWN iface)
|
|
{
|
|
RemUnkProxy *This = (RemUnkProxy *)iface;
|
|
ULONG refs;
|
|
|
|
TRACE("(%p)->Release()\n",This);
|
|
if (This->outer_unknown)
|
|
refs = IUnknown_Release(This->outer_unknown);
|
|
else
|
|
refs = InterlockedDecrement(&This->refs);
|
|
|
|
if (!refs) {
|
|
if (This->chan) IRpcChannelBuffer_Release(This->chan);
|
|
HeapFree(GetProcessHeap(),0,This);
|
|
}
|
|
return refs;
|
|
}
|
|
|
|
static HRESULT WINAPI RemUnkProxy_RemQueryInterface(LPREMUNKNOWN iface,
|
|
REFIPID ripid,
|
|
ULONG cRefs,
|
|
USHORT cIids,
|
|
IID* iids,
|
|
REMQIRESULT** ppQIResults)
|
|
{
|
|
RemUnkProxy *This = (RemUnkProxy *)iface;
|
|
RPCOLEMESSAGE msg;
|
|
HRESULT hr = S_OK;
|
|
ULONG status;
|
|
|
|
TRACE("(%p)->(%s,%ld,%d,%p,%p)\n",This,
|
|
debugstr_guid(ripid),cRefs,cIids,iids,ppQIResults);
|
|
|
|
*ppQIResults = NULL;
|
|
memset(&msg, 0, sizeof(msg));
|
|
msg.iMethod = 3;
|
|
msg.cbBuffer = sizeof(IPID) + sizeof(ULONG) +
|
|
sizeof(USHORT) + cIids*sizeof(IID);
|
|
hr = IRpcChannelBuffer_GetBuffer(This->chan, &msg, &IID_IRemUnknown);
|
|
if (SUCCEEDED(hr)) {
|
|
LPBYTE buf = msg.Buffer;
|
|
memcpy(buf, ripid, sizeof(IPID));
|
|
buf += sizeof(IPID);
|
|
memcpy(buf, &cRefs, sizeof(ULONG));
|
|
buf += sizeof(ULONG);
|
|
memcpy(buf, &cIids, sizeof(USHORT));
|
|
buf += sizeof(USHORT);
|
|
memcpy(buf, iids, cIids*sizeof(IID));
|
|
|
|
hr = IRpcChannelBuffer_SendReceive(This->chan, &msg, &status);
|
|
|
|
buf = msg.Buffer;
|
|
|
|
if (SUCCEEDED(hr)) {
|
|
hr = *(HRESULT *)buf;
|
|
buf += sizeof(HRESULT);
|
|
}
|
|
|
|
if (SUCCEEDED(hr)) {
|
|
*ppQIResults = CoTaskMemAlloc(cIids*sizeof(REMQIRESULT));
|
|
memcpy(*ppQIResults, buf, cIids*sizeof(REMQIRESULT));
|
|
}
|
|
|
|
IRpcChannelBuffer_FreeBuffer(This->chan, &msg);
|
|
}
|
|
|
|
return hr;
|
|
}
|
|
|
|
static HRESULT WINAPI RemUnkProxy_RemAddRef(LPREMUNKNOWN iface,
|
|
USHORT cInterfaceRefs,
|
|
REMINTERFACEREF* InterfaceRefs,
|
|
HRESULT* pResults)
|
|
{
|
|
RemUnkProxy *This = (RemUnkProxy *)iface;
|
|
RPCOLEMESSAGE msg;
|
|
HRESULT hr = S_OK;
|
|
ULONG status;
|
|
|
|
TRACE("(%p)->(%d,%p,%p)\n",This,
|
|
cInterfaceRefs,InterfaceRefs,pResults);
|
|
|
|
memset(&msg, 0, sizeof(msg));
|
|
msg.iMethod = 4;
|
|
msg.cbBuffer = sizeof(USHORT) + cInterfaceRefs*sizeof(REMINTERFACEREF);
|
|
hr = IRpcChannelBuffer_GetBuffer(This->chan, &msg, &IID_IRemUnknown);
|
|
if (SUCCEEDED(hr)) {
|
|
LPBYTE buf = msg.Buffer;
|
|
memcpy(buf, &cInterfaceRefs, sizeof(USHORT));
|
|
buf += sizeof(USHORT);
|
|
memcpy(buf, InterfaceRefs, cInterfaceRefs*sizeof(REMINTERFACEREF));
|
|
|
|
hr = IRpcChannelBuffer_SendReceive(This->chan, &msg, &status);
|
|
|
|
if (SUCCEEDED(hr)) {
|
|
buf = msg.Buffer;
|
|
memcpy(pResults, buf, cInterfaceRefs*sizeof(HRESULT));
|
|
}
|
|
|
|
IRpcChannelBuffer_FreeBuffer(This->chan, &msg);
|
|
}
|
|
|
|
return hr;
|
|
}
|
|
|
|
static HRESULT WINAPI RemUnkProxy_RemRelease(LPREMUNKNOWN iface,
|
|
USHORT cInterfaceRefs,
|
|
REMINTERFACEREF* InterfaceRefs)
|
|
{
|
|
RemUnkProxy *This = (RemUnkProxy *)iface;
|
|
RPCOLEMESSAGE msg;
|
|
HRESULT hr = S_OK;
|
|
ULONG status;
|
|
|
|
TRACE("(%p)->(%d,%p)\n",This,
|
|
cInterfaceRefs,InterfaceRefs);
|
|
|
|
memset(&msg, 0, sizeof(msg));
|
|
msg.iMethod = 5;
|
|
msg.cbBuffer = sizeof(USHORT) + cInterfaceRefs*sizeof(REMINTERFACEREF);
|
|
hr = IRpcChannelBuffer_GetBuffer(This->chan, &msg, &IID_IRemUnknown);
|
|
if (SUCCEEDED(hr)) {
|
|
LPBYTE buf = msg.Buffer;
|
|
memcpy(buf, &cInterfaceRefs, sizeof(USHORT));
|
|
buf += sizeof(USHORT);
|
|
memcpy(buf, InterfaceRefs, cInterfaceRefs*sizeof(REMINTERFACEREF));
|
|
|
|
hr = IRpcChannelBuffer_SendReceive(This->chan, &msg, &status);
|
|
|
|
IRpcChannelBuffer_FreeBuffer(This->chan, &msg);
|
|
}
|
|
|
|
return hr;
|
|
}
|
|
|
|
static const IRemUnknownVtbl RemUnkProxy_VTable =
|
|
{
|
|
RemUnkProxy_QueryInterface,
|
|
RemUnkProxy_AddRef,
|
|
RemUnkProxy_Release,
|
|
RemUnkProxy_RemQueryInterface,
|
|
RemUnkProxy_RemAddRef,
|
|
RemUnkProxy_RemRelease
|
|
};
|
|
|
|
|
|
static HRESULT WINAPI RURpcProxyBufferImpl_QueryInterface(LPRPCPROXYBUFFER iface,REFIID riid,LPVOID *ppv) {
|
|
*ppv = NULL;
|
|
if (IsEqualIID(riid,&IID_IRpcProxyBuffer)||IsEqualIID(riid,&IID_IUnknown)) {
|
|
IRpcProxyBuffer_AddRef(iface);
|
|
*ppv = (LPVOID)iface;
|
|
return S_OK;
|
|
}
|
|
FIXME("(%s), no interface.\n",debugstr_guid(riid));
|
|
return E_NOINTERFACE;
|
|
}
|
|
|
|
static ULONG WINAPI RURpcProxyBufferImpl_AddRef(LPRPCPROXYBUFFER iface) {
|
|
ICOM_THIS_MULTI(RemUnkProxy,lpvtbl_proxy,iface);
|
|
TRACE("%p, %ld\n", iface, This->refs + 1);
|
|
return InterlockedIncrement(&This->refs);
|
|
}
|
|
|
|
static ULONG WINAPI RURpcProxyBufferImpl_Release(LPRPCPROXYBUFFER iface) {
|
|
ICOM_THIS_MULTI(RemUnkProxy,lpvtbl_proxy,iface);
|
|
ULONG ref = InterlockedDecrement(&This->refs);
|
|
TRACE("%p, %ld\n", iface, ref);
|
|
if (!ref) {
|
|
IRpcProxyBuffer_Disconnect(iface);
|
|
HeapFree(GetProcessHeap(),0,This);
|
|
}
|
|
return ref;
|
|
}
|
|
|
|
static HRESULT WINAPI RURpcProxyBufferImpl_Connect(LPRPCPROXYBUFFER iface,IRpcChannelBuffer* pRpcChannelBuffer) {
|
|
ICOM_THIS_MULTI(RemUnkProxy,lpvtbl_proxy,iface);
|
|
|
|
TRACE("%p, %p\n", iface, pRpcChannelBuffer);
|
|
This->chan = pRpcChannelBuffer;
|
|
IRpcChannelBuffer_AddRef(This->chan);
|
|
return S_OK;
|
|
}
|
|
static void WINAPI RURpcProxyBufferImpl_Disconnect(LPRPCPROXYBUFFER iface) {
|
|
ICOM_THIS_MULTI(RemUnkProxy,lpvtbl_proxy,iface);
|
|
TRACE("%p, %p\n", iface, This->chan);
|
|
if (This->chan) {
|
|
IRpcChannelBuffer_Release(This->chan);
|
|
This->chan = NULL;
|
|
}
|
|
}
|
|
|
|
|
|
static const IRpcProxyBufferVtbl RURpcProxyBuffer_VTable = {
|
|
RURpcProxyBufferImpl_QueryInterface,
|
|
RURpcProxyBufferImpl_AddRef,
|
|
RURpcProxyBufferImpl_Release,
|
|
RURpcProxyBufferImpl_Connect,
|
|
RURpcProxyBufferImpl_Disconnect
|
|
};
|
|
|
|
static HRESULT
|
|
RemUnkProxy_Construct(IUnknown *pUnkOuter, LPVOID *ppv,LPVOID *ppProxy) {
|
|
RemUnkProxy *This;
|
|
|
|
This = HeapAlloc(GetProcessHeap(),HEAP_ZERO_MEMORY,sizeof(*This));
|
|
if (!This)
|
|
return E_OUTOFMEMORY;
|
|
|
|
This->lpvtbl_remunk = &RemUnkProxy_VTable;
|
|
This->lpvtbl_proxy = &RURpcProxyBuffer_VTable;
|
|
/* only one reference for the proxy buffer */
|
|
This->refs = 1;
|
|
This->outer_unknown = pUnkOuter;
|
|
*ppv = &(This->lpvtbl_remunk);
|
|
*ppProxy = &(This->lpvtbl_proxy);
|
|
/* and one reference for the object */
|
|
IUnknown_AddRef((IUnknown *)*ppv);
|
|
return S_OK;
|
|
}
|
|
|
|
|
|
/********************* OLE Proxy/Stub Factory ********************************/
|
|
static HRESULT WINAPI
|
|
PSFacBuf_QueryInterface(LPPSFACTORYBUFFER iface, REFIID iid, LPVOID *ppv) {
|
|
if (IsEqualIID(iid,&IID_IPSFactoryBuffer)||IsEqualIID(iid,&IID_IUnknown)) {
|
|
*ppv = (LPVOID)iface;
|
|
/* No ref counting, static class */
|
|
return S_OK;
|
|
}
|
|
FIXME("(%s) unknown IID?\n",debugstr_guid(iid));
|
|
return E_NOINTERFACE;
|
|
}
|
|
|
|
static ULONG WINAPI PSFacBuf_AddRef(LPPSFACTORYBUFFER iface) { return 2; }
|
|
static ULONG WINAPI PSFacBuf_Release(LPPSFACTORYBUFFER iface) { return 1; }
|
|
|
|
static HRESULT WINAPI
|
|
PSFacBuf_CreateProxy(
|
|
LPPSFACTORYBUFFER iface, IUnknown* pUnkOuter, REFIID riid,
|
|
IRpcProxyBuffer **ppProxy, LPVOID *ppv
|
|
) {
|
|
if (IsEqualIID(&IID_IClassFactory,riid))
|
|
return CFProxy_Construct(pUnkOuter, ppv,(LPVOID*)ppProxy);
|
|
else if (IsEqualIID(&IID_IRemUnknown,riid))
|
|
return RemUnkProxy_Construct(pUnkOuter, ppv,(LPVOID*)ppProxy);
|
|
FIXME("proxying not implemented for (%s) yet!\n",debugstr_guid(riid));
|
|
return E_FAIL;
|
|
}
|
|
|
|
static HRESULT WINAPI
|
|
PSFacBuf_CreateStub(
|
|
LPPSFACTORYBUFFER iface, REFIID riid,IUnknown *pUnkServer,
|
|
IRpcStubBuffer** ppStub
|
|
) {
|
|
HRESULT hres;
|
|
|
|
TRACE("(%s,%p,%p)\n",debugstr_guid(riid),pUnkServer,ppStub);
|
|
|
|
if (IsEqualIID(&IID_IClassFactory, riid) ||
|
|
IsEqualIID(&IID_IUnknown, riid) /* FIXME: fixup stub manager and remove this*/) {
|
|
hres = CFStub_Construct(ppStub);
|
|
if (!hres)
|
|
IRpcStubBuffer_Connect((*ppStub),pUnkServer);
|
|
return hres;
|
|
} else if (IsEqualIID(&IID_IRemUnknown,riid)) {
|
|
hres = RemUnkStub_Construct(ppStub);
|
|
if (!hres)
|
|
IRpcStubBuffer_Connect((*ppStub),pUnkServer);
|
|
return hres;
|
|
}
|
|
FIXME("stubbing not implemented for (%s) yet!\n",debugstr_guid(riid));
|
|
return E_FAIL;
|
|
}
|
|
|
|
static const IPSFactoryBufferVtbl psfacbufvtbl = {
|
|
PSFacBuf_QueryInterface,
|
|
PSFacBuf_AddRef,
|
|
PSFacBuf_Release,
|
|
PSFacBuf_CreateProxy,
|
|
PSFacBuf_CreateStub
|
|
};
|
|
|
|
/* This is the whole PSFactoryBuffer object, just the vtableptr */
|
|
static const IPSFactoryBufferVtbl *lppsfac = &psfacbufvtbl;
|
|
|
|
/***********************************************************************
|
|
* DllGetClassObject [OLE32.@]
|
|
*/
|
|
HRESULT WINAPI DllGetClassObject(REFCLSID rclsid, REFIID iid,LPVOID *ppv)
|
|
{
|
|
*ppv = NULL;
|
|
if (IsEqualIID(rclsid, &CLSID_PSFactoryBuffer))
|
|
return IPSFactoryBuffer_QueryInterface((IPSFactoryBuffer *)&lppsfac, iid, ppv);
|
|
if (IsEqualIID(rclsid,&CLSID_DfMarshal)&&(
|
|
IsEqualIID(iid,&IID_IClassFactory) ||
|
|
IsEqualIID(iid,&IID_IUnknown)
|
|
)
|
|
)
|
|
return MARSHAL_GetStandardMarshalCF(ppv);
|
|
if (IsEqualIID(rclsid,&CLSID_StdGlobalInterfaceTable) && (IsEqualIID(iid,&IID_IClassFactory) || IsEqualIID(iid,&IID_IUnknown)))
|
|
return StdGlobalInterfaceTable_GetFactory(ppv);
|
|
if (IsEqualCLSID(rclsid, &CLSID_FileMoniker))
|
|
return FileMonikerCF_Create(iid, ppv);
|
|
if (IsEqualCLSID(rclsid, &CLSID_ItemMoniker))
|
|
return ItemMonikerCF_Create(iid, ppv);
|
|
if (IsEqualCLSID(rclsid, &CLSID_AntiMoniker))
|
|
return AntiMonikerCF_Create(iid, ppv);
|
|
|
|
FIXME("\n\tCLSID:\t%s,\n\tIID:\t%s\n",debugstr_guid(rclsid),debugstr_guid(iid));
|
|
return CLASS_E_CLASSNOTAVAILABLE;
|
|
}
|