2002-02-05 19:11:17 +01:00
|
|
|
/*
|
2003-05-13 02:41:57 +02:00
|
|
|
* Marshalling library
|
2002-02-05 19:11:17 +01:00
|
|
|
*
|
|
|
|
* Copyright 2002 Marcus Meissner
|
2002-03-10 00:29:33 +01:00
|
|
|
*
|
|
|
|
* 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
|
2002-02-05 19:11:17 +01:00
|
|
|
*/
|
|
|
|
|
|
|
|
#include "config.h"
|
|
|
|
|
|
|
|
#include <stdlib.h>
|
2003-09-06 01:08:26 +02:00
|
|
|
#include <stdarg.h>
|
2002-02-05 19:11:17 +01:00
|
|
|
#include <stdio.h>
|
|
|
|
#include <string.h>
|
|
|
|
#include <assert.h>
|
|
|
|
|
|
|
|
#include "windef.h"
|
2003-09-06 01:08:26 +02:00
|
|
|
#include "winbase.h"
|
2003-09-09 21:39:31 +02:00
|
|
|
#include "winuser.h"
|
2002-02-05 19:11:17 +01:00
|
|
|
#include "objbase.h"
|
|
|
|
#include "ole2.h"
|
|
|
|
#include "ole2ver.h"
|
|
|
|
#include "rpc.h"
|
|
|
|
#include "winerror.h"
|
|
|
|
#include "winreg.h"
|
|
|
|
#include "wownt32.h"
|
|
|
|
#include "wtypes.h"
|
|
|
|
#include "wine/unicode.h"
|
|
|
|
#include "wine/winbase16.h"
|
|
|
|
#include "compobj_private.h"
|
|
|
|
#include "ifs.h"
|
|
|
|
|
2002-03-10 00:29:33 +01:00
|
|
|
#include "wine/debug.h"
|
2002-02-05 19:11:17 +01:00
|
|
|
|
2002-03-10 00:29:33 +01:00
|
|
|
WINE_DEFAULT_DEBUG_CHANNEL(ole);
|
2002-02-05 19:11:17 +01:00
|
|
|
|
2002-12-19 23:16:35 +01:00
|
|
|
extern const CLSID CLSID_DfMarshal;
|
|
|
|
|
2003-05-13 02:41:57 +02:00
|
|
|
/* Marshalling just passes a unique identifier to the remote client,
|
2002-02-05 19:11:17 +01:00
|
|
|
* that makes it possible to find the passed interface again.
|
|
|
|
*
|
|
|
|
* So basically we need a set of values that make it unique.
|
|
|
|
*
|
|
|
|
* Process Identifier, Object IUnknown ptr, IID
|
|
|
|
*
|
|
|
|
* Note that the IUnknown_QI(ob,xiid,&ppv) always returns the SAME ppv value!
|
|
|
|
*/
|
|
|
|
|
2004-07-23 01:44:54 +02:00
|
|
|
inline static HRESULT
|
|
|
|
get_facbuf_for_iid(REFIID riid,IPSFactoryBuffer **facbuf) {
|
|
|
|
HRESULT hres;
|
|
|
|
CLSID pxclsid;
|
|
|
|
|
|
|
|
if ((hres = CoGetPSClsid(riid,&pxclsid)))
|
|
|
|
return hres;
|
|
|
|
return CoGetClassObject(&pxclsid,CLSCTX_INPROC_SERVER,NULL,&IID_IPSFactoryBuffer,(LPVOID*)facbuf);
|
|
|
|
}
|
|
|
|
|
|
|
|
typedef struct _wine_marshal_data {
|
|
|
|
DWORD dwDestContext;
|
|
|
|
DWORD mshlflags;
|
|
|
|
} wine_marshal_data;
|
|
|
|
|
2002-02-05 19:11:17 +01:00
|
|
|
typedef struct _mid2unknown {
|
|
|
|
wine_marshal_id mid;
|
|
|
|
LPUNKNOWN pUnk;
|
|
|
|
} mid2unknown;
|
|
|
|
|
|
|
|
typedef struct _mid2stub {
|
|
|
|
wine_marshal_id mid;
|
|
|
|
IRpcStubBuffer *stub;
|
|
|
|
LPUNKNOWN pUnkServer;
|
2004-07-22 22:34:14 +02:00
|
|
|
BOOL valid;
|
2002-02-05 19:11:17 +01:00
|
|
|
} mid2stub;
|
|
|
|
|
|
|
|
static mid2stub *stubs = NULL;
|
|
|
|
static int nrofstubs = 0;
|
|
|
|
|
|
|
|
static mid2unknown *proxies = NULL;
|
|
|
|
static int nrofproxies = 0;
|
|
|
|
|
2004-07-30 01:58:12 +02:00
|
|
|
void MARSHAL_Invalidate_Stub_From_MID(wine_marshal_id *mid) {
|
|
|
|
int i;
|
|
|
|
|
|
|
|
for (i=0;i<nrofstubs;i++) {
|
|
|
|
if (!stubs[i].valid) continue;
|
|
|
|
|
|
|
|
if (MARSHAL_Compare_Mids(mid,&(stubs[i].mid))) {
|
|
|
|
stubs[i].valid = FALSE;
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
2002-02-05 19:11:17 +01:00
|
|
|
HRESULT
|
|
|
|
MARSHAL_Find_Stub_Buffer(wine_marshal_id *mid,IRpcStubBuffer **stub) {
|
|
|
|
int i;
|
|
|
|
|
|
|
|
for (i=0;i<nrofstubs;i++) {
|
2004-07-22 22:34:14 +02:00
|
|
|
if (!stubs[i].valid) continue;
|
|
|
|
|
2002-02-05 19:11:17 +01:00
|
|
|
if (MARSHAL_Compare_Mids(mid,&(stubs[i].mid))) {
|
|
|
|
*stub = stubs[i].stub;
|
|
|
|
IUnknown_AddRef((*stub));
|
|
|
|
return S_OK;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return E_FAIL;
|
|
|
|
}
|
|
|
|
|
2004-08-02 20:47:31 +02:00
|
|
|
static HRESULT
|
2002-02-05 19:11:17 +01:00
|
|
|
MARSHAL_Find_Stub(wine_marshal_id *mid,LPUNKNOWN *pUnk) {
|
|
|
|
int i;
|
|
|
|
|
|
|
|
for (i=0;i<nrofstubs;i++) {
|
2004-07-22 22:34:14 +02:00
|
|
|
if (!stubs[i].valid) continue;
|
|
|
|
|
2002-02-05 19:11:17 +01:00
|
|
|
if (MARSHAL_Compare_Mids(mid,&(stubs[i].mid))) {
|
|
|
|
*pUnk = stubs[i].pUnkServer;
|
|
|
|
IUnknown_AddRef((*pUnk));
|
|
|
|
return S_OK;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return E_FAIL;
|
|
|
|
}
|
|
|
|
|
2004-08-02 20:47:31 +02:00
|
|
|
static HRESULT
|
2002-02-05 19:11:17 +01:00
|
|
|
MARSHAL_Register_Stub(wine_marshal_id *mid,LPUNKNOWN pUnk,IRpcStubBuffer *stub) {
|
|
|
|
LPUNKNOWN xPunk;
|
|
|
|
if (!MARSHAL_Find_Stub(mid,&xPunk)) {
|
|
|
|
FIXME("Already have entry for (%lx/%s)!\n",mid->objectid,debugstr_guid(&(mid->iid)));
|
|
|
|
return S_OK;
|
|
|
|
}
|
|
|
|
if (nrofstubs)
|
|
|
|
stubs=HeapReAlloc(GetProcessHeap(),0,stubs,sizeof(stubs[0])*(nrofstubs+1));
|
|
|
|
else
|
|
|
|
stubs=HeapAlloc(GetProcessHeap(),0,sizeof(stubs[0]));
|
|
|
|
if (!stubs) return E_OUTOFMEMORY;
|
|
|
|
stubs[nrofstubs].stub = stub;
|
|
|
|
stubs[nrofstubs].pUnkServer = pUnk;
|
|
|
|
memcpy(&(stubs[nrofstubs].mid),mid,sizeof(*mid));
|
2004-07-22 22:34:14 +02:00
|
|
|
stubs[nrofstubs].valid = TRUE; /* set to false when released by ReleaseMarshalData */
|
2002-02-05 19:11:17 +01:00
|
|
|
nrofstubs++;
|
|
|
|
return S_OK;
|
|
|
|
}
|
|
|
|
|
2004-08-02 20:28:29 +02:00
|
|
|
HRESULT
|
|
|
|
MARSHAL_Disconnect_Proxies() {
|
|
|
|
int i;
|
|
|
|
|
|
|
|
TRACE("Disconnecting %d proxies\n", nrofproxies);
|
|
|
|
|
|
|
|
for (i = 0; i < nrofproxies; i++)
|
|
|
|
IRpcProxyBuffer_Disconnect((IRpcProxyBuffer*)proxies[i].pUnk);
|
|
|
|
|
|
|
|
return S_OK;
|
|
|
|
}
|
|
|
|
|
2004-08-02 20:47:31 +02:00
|
|
|
static HRESULT
|
2002-02-05 19:11:17 +01:00
|
|
|
MARSHAL_Register_Proxy(wine_marshal_id *mid,LPUNKNOWN punk) {
|
|
|
|
int i;
|
|
|
|
|
|
|
|
for (i=0;i<nrofproxies;i++) {
|
|
|
|
if (MARSHAL_Compare_Mids(mid,&(proxies[i].mid))) {
|
|
|
|
ERR("Already have mid?\n");
|
|
|
|
return E_FAIL;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
if (nrofproxies)
|
|
|
|
proxies = HeapReAlloc(GetProcessHeap(),0,proxies,sizeof(proxies[0])*(nrofproxies+1));
|
|
|
|
else
|
|
|
|
proxies = HeapAlloc(GetProcessHeap(),0,sizeof(proxies[0]));
|
|
|
|
memcpy(&(proxies[nrofproxies].mid),mid,sizeof(*mid));
|
|
|
|
proxies[nrofproxies].pUnk = punk;
|
|
|
|
nrofproxies++;
|
|
|
|
IUnknown_AddRef(punk);
|
|
|
|
return S_OK;
|
|
|
|
}
|
|
|
|
|
|
|
|
/********************** StdMarshal implementation ****************************/
|
|
|
|
typedef struct _StdMarshalImpl {
|
|
|
|
ICOM_VTABLE(IMarshal) *lpvtbl;
|
|
|
|
DWORD ref;
|
|
|
|
|
|
|
|
IID iid;
|
|
|
|
DWORD dwDestContext;
|
|
|
|
LPVOID pvDestContext;
|
|
|
|
DWORD mshlflags;
|
|
|
|
} StdMarshalImpl;
|
|
|
|
|
2004-08-02 20:47:31 +02:00
|
|
|
static HRESULT WINAPI
|
2002-02-05 19:11:17 +01:00
|
|
|
StdMarshalImpl_QueryInterface(LPMARSHAL iface,REFIID riid,LPVOID *ppv) {
|
|
|
|
*ppv = NULL;
|
|
|
|
if (IsEqualIID(&IID_IUnknown,riid) || IsEqualIID(&IID_IMarshal,riid)) {
|
|
|
|
*ppv = iface;
|
|
|
|
IUnknown_AddRef(iface);
|
|
|
|
return S_OK;
|
|
|
|
}
|
|
|
|
FIXME("No interface for %s.\n",debugstr_guid(riid));
|
|
|
|
return E_NOINTERFACE;
|
|
|
|
}
|
|
|
|
|
2004-08-02 20:47:31 +02:00
|
|
|
static ULONG WINAPI
|
2002-02-05 19:11:17 +01:00
|
|
|
StdMarshalImpl_AddRef(LPMARSHAL iface) {
|
|
|
|
ICOM_THIS(StdMarshalImpl,iface);
|
|
|
|
This->ref++;
|
|
|
|
return This->ref;
|
|
|
|
}
|
|
|
|
|
2004-08-02 20:47:31 +02:00
|
|
|
static ULONG WINAPI
|
2002-02-05 19:11:17 +01:00
|
|
|
StdMarshalImpl_Release(LPMARSHAL iface) {
|
|
|
|
ICOM_THIS(StdMarshalImpl,iface);
|
|
|
|
This->ref--;
|
|
|
|
|
|
|
|
if (This->ref)
|
|
|
|
return This->ref;
|
|
|
|
HeapFree(GetProcessHeap(),0,This);
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
2004-08-02 20:47:31 +02:00
|
|
|
static HRESULT WINAPI
|
2002-02-05 19:11:17 +01:00
|
|
|
StdMarshalImpl_GetUnmarshalClass(
|
|
|
|
LPMARSHAL iface, REFIID riid, void* pv, DWORD dwDestContext,
|
|
|
|
void* pvDestContext, DWORD mshlflags, CLSID* pCid
|
|
|
|
) {
|
|
|
|
memcpy(pCid,&CLSID_DfMarshal,sizeof(CLSID_DfMarshal));
|
|
|
|
return S_OK;
|
|
|
|
}
|
|
|
|
|
2004-08-02 20:47:31 +02:00
|
|
|
static HRESULT WINAPI
|
2002-02-05 19:11:17 +01:00
|
|
|
StdMarshalImpl_GetMarshalSizeMax(
|
|
|
|
LPMARSHAL iface, REFIID riid, void* pv, DWORD dwDestContext,
|
|
|
|
void* pvDestContext, DWORD mshlflags, DWORD* pSize
|
|
|
|
) {
|
|
|
|
*pSize = sizeof(wine_marshal_id)+sizeof(wine_marshal_data);
|
|
|
|
return S_OK;
|
|
|
|
}
|
|
|
|
|
2004-08-02 20:47:31 +02:00
|
|
|
static HRESULT WINAPI
|
2002-02-05 19:11:17 +01:00
|
|
|
StdMarshalImpl_MarshalInterface(
|
|
|
|
LPMARSHAL iface, IStream *pStm,REFIID riid, void* pv, DWORD dwDestContext,
|
|
|
|
void* pvDestContext, DWORD mshlflags
|
|
|
|
) {
|
|
|
|
wine_marshal_id mid;
|
|
|
|
wine_marshal_data md;
|
|
|
|
IUnknown *pUnk;
|
|
|
|
ULONG res;
|
|
|
|
HRESULT hres;
|
|
|
|
IRpcStubBuffer *stub;
|
|
|
|
IPSFactoryBuffer *psfacbuf;
|
|
|
|
|
|
|
|
TRACE("(...,%s,...)\n",debugstr_guid(riid));
|
2004-07-23 21:10:13 +02:00
|
|
|
|
2004-05-21 22:52:57 +02:00
|
|
|
IUnknown_QueryInterface((LPUNKNOWN)pv,&IID_IUnknown,(LPVOID*)&pUnk);
|
2002-02-05 19:11:17 +01:00
|
|
|
mid.processid = GetCurrentProcessId();
|
|
|
|
mid.objectid = (DWORD)pUnk; /* FIXME */
|
|
|
|
IUnknown_Release(pUnk);
|
|
|
|
memcpy(&mid.iid,riid,sizeof(mid.iid));
|
|
|
|
md.dwDestContext = dwDestContext;
|
|
|
|
md.mshlflags = mshlflags;
|
|
|
|
hres = IStream_Write(pStm,&mid,sizeof(mid),&res);
|
|
|
|
if (hres) return hres;
|
|
|
|
hres = IStream_Write(pStm,&md,sizeof(md),&res);
|
|
|
|
if (hres) return hres;
|
|
|
|
|
2004-07-29 04:43:59 +02:00
|
|
|
if (SUCCEEDED(MARSHAL_Find_Stub_Buffer(&mid,&stub))) {
|
|
|
|
/* Find_Stub_Buffer gives us a ref but we want to keep it, as if we'd created a new one */
|
|
|
|
TRACE("Found RpcStubBuffer %p\n", stub);
|
2002-02-05 19:11:17 +01:00
|
|
|
return S_OK;
|
|
|
|
}
|
|
|
|
hres = get_facbuf_for_iid(riid,&psfacbuf);
|
|
|
|
if (hres) return hres;
|
2004-07-22 22:34:14 +02:00
|
|
|
|
2002-02-05 19:11:17 +01:00
|
|
|
hres = IPSFactoryBuffer_CreateStub(psfacbuf,riid,pv,&stub);
|
|
|
|
IPSFactoryBuffer_Release(psfacbuf);
|
|
|
|
if (hres) {
|
|
|
|
FIXME("Failed to create a stub for %s\n",debugstr_guid(riid));
|
|
|
|
return hres;
|
|
|
|
}
|
2004-05-21 22:52:57 +02:00
|
|
|
IUnknown_QueryInterface((LPUNKNOWN)pv,riid,(LPVOID*)&pUnk);
|
2002-02-05 19:11:17 +01:00
|
|
|
MARSHAL_Register_Stub(&mid,pUnk,stub);
|
|
|
|
IUnknown_Release(pUnk);
|
|
|
|
return S_OK;
|
|
|
|
}
|
|
|
|
|
2004-08-02 20:47:31 +02:00
|
|
|
static HRESULT WINAPI
|
2002-02-05 19:11:17 +01:00
|
|
|
StdMarshalImpl_UnmarshalInterface(
|
|
|
|
LPMARSHAL iface, IStream *pStm, REFIID riid, void **ppv
|
|
|
|
) {
|
|
|
|
wine_marshal_id mid;
|
|
|
|
wine_marshal_data md;
|
|
|
|
ULONG res;
|
|
|
|
HRESULT hres;
|
|
|
|
IPSFactoryBuffer *psfacbuf;
|
|
|
|
IRpcProxyBuffer *rpcproxy;
|
|
|
|
IRpcChannelBuffer *chanbuf;
|
|
|
|
|
|
|
|
TRACE("(...,%s,....)\n",debugstr_guid(riid));
|
|
|
|
hres = IStream_Read(pStm,&mid,sizeof(mid),&res);
|
|
|
|
if (hres) return hres;
|
|
|
|
hres = IStream_Read(pStm,&md,sizeof(md),&res);
|
|
|
|
if (hres) return hres;
|
|
|
|
if (SUCCEEDED(MARSHAL_Find_Stub(&mid,(LPUNKNOWN*)ppv))) {
|
|
|
|
FIXME("Calling back to ourselves for %s!\n",debugstr_guid(riid));
|
|
|
|
return S_OK;
|
|
|
|
}
|
2004-07-30 02:02:17 +02:00
|
|
|
if (IsEqualIID(riid, &IID_IUnknown) || IsEqualIID(riid, &IID_NULL)) {
|
|
|
|
/* should return proxy manager IUnknown object */
|
|
|
|
FIXME("Special treatment required for IID of %s\n", debugstr_guid(riid));
|
|
|
|
}
|
2002-02-05 19:11:17 +01:00
|
|
|
hres = get_facbuf_for_iid(riid,&psfacbuf);
|
|
|
|
if (hres) return hres;
|
|
|
|
hres = IPSFactoryBuffer_CreateProxy(psfacbuf,NULL,riid,&rpcproxy,ppv);
|
|
|
|
if (hres) {
|
|
|
|
FIXME("Failed to create a proxy for %s\n",debugstr_guid(riid));
|
|
|
|
return hres;
|
|
|
|
}
|
2004-08-02 20:28:29 +02:00
|
|
|
|
|
|
|
MARSHAL_Register_Proxy(&mid, (LPUNKNOWN) rpcproxy);
|
|
|
|
|
2002-02-05 19:11:17 +01:00
|
|
|
hres = PIPE_GetNewPipeBuf(&mid,&chanbuf);
|
2004-06-14 19:58:03 +02:00
|
|
|
IPSFactoryBuffer_Release(psfacbuf);
|
2002-11-04 23:38:26 +01:00
|
|
|
if (hres) {
|
|
|
|
ERR("Failed to get an rpc channel buffer for %s\n",debugstr_guid(riid));
|
|
|
|
} else {
|
2004-06-14 19:58:03 +02:00
|
|
|
/* Connect the channel buffer to the proxy and release the no longer
|
|
|
|
* needed proxy.
|
|
|
|
* NOTE: The proxy should have taken an extra reference because it also
|
|
|
|
* aggregates the object, so we can safely release our reference to it. */
|
2002-11-04 23:38:26 +01:00
|
|
|
IRpcProxyBuffer_Connect(rpcproxy,chanbuf);
|
2004-06-14 19:58:03 +02:00
|
|
|
IRpcProxyBuffer_Release(rpcproxy);
|
|
|
|
/* IRpcProxyBuffer takes a reference on the channel buffer and
|
|
|
|
* we no longer need it, so release it */
|
|
|
|
IRpcChannelBuffer_Release(chanbuf);
|
2002-11-04 23:38:26 +01:00
|
|
|
}
|
|
|
|
return hres;
|
2002-02-05 19:11:17 +01:00
|
|
|
}
|
|
|
|
|
2004-08-02 20:47:31 +02:00
|
|
|
static HRESULT WINAPI
|
2002-02-05 19:11:17 +01:00
|
|
|
StdMarshalImpl_ReleaseMarshalData(LPMARSHAL iface, IStream *pStm) {
|
2004-07-22 22:34:14 +02:00
|
|
|
wine_marshal_id mid;
|
|
|
|
ULONG res;
|
|
|
|
HRESULT hres;
|
|
|
|
IRpcStubBuffer *stub = NULL;
|
|
|
|
int i;
|
|
|
|
|
|
|
|
hres = IStream_Read(pStm,&mid,sizeof(mid),&res);
|
|
|
|
if (hres) return hres;
|
|
|
|
|
|
|
|
for (i=0; i < nrofstubs; i++)
|
|
|
|
{
|
|
|
|
if (!stubs[i].valid) continue;
|
|
|
|
|
|
|
|
if (MARSHAL_Compare_Mids(&mid, &(stubs[i].mid)))
|
|
|
|
{
|
|
|
|
stub = stubs[i].stub;
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
if (!stub)
|
|
|
|
{
|
|
|
|
FIXME("Could not map MID to stub??\n");
|
|
|
|
return E_FAIL;
|
|
|
|
}
|
|
|
|
|
|
|
|
res = IRpcStubBuffer_Release(stub);
|
|
|
|
stubs[i].valid = FALSE;
|
|
|
|
TRACE("stub refcount of %p is %ld\n", stub, res);
|
|
|
|
|
2002-02-05 19:11:17 +01:00
|
|
|
return S_OK;
|
|
|
|
}
|
|
|
|
|
2004-08-02 20:47:31 +02:00
|
|
|
static HRESULT WINAPI
|
2002-02-05 19:11:17 +01:00
|
|
|
StdMarshalImpl_DisconnectObject(LPMARSHAL iface, DWORD dwReserved) {
|
|
|
|
FIXME("(), stub!\n");
|
|
|
|
return S_OK;
|
|
|
|
}
|
|
|
|
|
|
|
|
ICOM_VTABLE(IMarshal) stdmvtbl = {
|
|
|
|
ICOM_MSVTABLE_COMPAT_DummyRTTIVALUE
|
|
|
|
StdMarshalImpl_QueryInterface,
|
|
|
|
StdMarshalImpl_AddRef,
|
|
|
|
StdMarshalImpl_Release,
|
|
|
|
StdMarshalImpl_GetUnmarshalClass,
|
|
|
|
StdMarshalImpl_GetMarshalSizeMax,
|
|
|
|
StdMarshalImpl_MarshalInterface,
|
|
|
|
StdMarshalImpl_UnmarshalInterface,
|
|
|
|
StdMarshalImpl_ReleaseMarshalData,
|
|
|
|
StdMarshalImpl_DisconnectObject
|
|
|
|
};
|
|
|
|
|
|
|
|
/***********************************************************************
|
2003-09-11 05:06:25 +02:00
|
|
|
* CoGetStandardMarshal [OLE32.@]
|
2002-02-05 19:11:17 +01:00
|
|
|
*
|
2003-05-13 02:41:57 +02:00
|
|
|
* When the COM library in the client process receives a marshalled
|
2002-02-05 19:11:17 +01:00
|
|
|
* interface pointer, it looks for a CLSID to be used in creating a proxy
|
2003-05-13 02:41:57 +02:00
|
|
|
* for the purposes of unmarshalling the packet. If the packet does not
|
2002-02-05 19:11:17 +01:00
|
|
|
* contain a CLSID for the proxy, COM calls CoGetStandardMarshal, passing a
|
|
|
|
* NULL pUnk value.
|
|
|
|
* This function creates a standard proxy in the client process and returns
|
|
|
|
* a pointer to that proxy's implementation of IMarshal.
|
|
|
|
* COM uses this pointer to call CoUnmarshalInterface to retrieve the pointer
|
|
|
|
* to the requested interface.
|
|
|
|
*/
|
|
|
|
HRESULT WINAPI
|
|
|
|
CoGetStandardMarshal(
|
|
|
|
REFIID riid,IUnknown *pUnk,DWORD dwDestContext,LPVOID pvDestContext,
|
|
|
|
DWORD mshlflags, LPMARSHAL *pMarshal
|
|
|
|
) {
|
|
|
|
StdMarshalImpl *dm;
|
|
|
|
|
|
|
|
if (pUnk == NULL) {
|
|
|
|
FIXME("(%s,NULL,%lx,%p,%lx,%p), unimplemented yet.\n",
|
|
|
|
debugstr_guid(riid),dwDestContext,pvDestContext,mshlflags,pMarshal
|
|
|
|
);
|
|
|
|
return E_FAIL;
|
|
|
|
}
|
|
|
|
TRACE("(%s,%p,%lx,%p,%lx,%p)\n",
|
|
|
|
debugstr_guid(riid),pUnk,dwDestContext,pvDestContext,mshlflags,pMarshal
|
|
|
|
);
|
2004-04-30 20:32:36 +02:00
|
|
|
*pMarshal = HeapAlloc(GetProcessHeap(),0,sizeof(StdMarshalImpl));
|
|
|
|
dm = (StdMarshalImpl*) *pMarshal;
|
2002-02-05 19:11:17 +01:00
|
|
|
if (!dm) return E_FAIL;
|
|
|
|
dm->lpvtbl = &stdmvtbl;
|
|
|
|
dm->ref = 1;
|
|
|
|
|
|
|
|
memcpy(&dm->iid,riid,sizeof(dm->iid));
|
|
|
|
dm->dwDestContext = dwDestContext;
|
|
|
|
dm->pvDestContext = pvDestContext;
|
|
|
|
dm->mshlflags = mshlflags;
|
|
|
|
return S_OK;
|
|
|
|
}
|
|
|
|
|
|
|
|
/* Helper function for getting Marshaler */
|
|
|
|
static HRESULT WINAPI
|
|
|
|
_GetMarshaller(REFIID riid, IUnknown *pUnk,DWORD dwDestContext,
|
|
|
|
void *pvDestContext, DWORD mshlFlags, LPMARSHAL *pMarshal
|
|
|
|
) {
|
|
|
|
HRESULT hres;
|
|
|
|
|
|
|
|
if (!pUnk)
|
|
|
|
return E_POINTER;
|
|
|
|
hres = IUnknown_QueryInterface(pUnk,&IID_IMarshal,(LPVOID*)pMarshal);
|
|
|
|
if (hres)
|
|
|
|
hres = CoGetStandardMarshal(riid,pUnk,dwDestContext,pvDestContext,mshlFlags,pMarshal);
|
|
|
|
return hres;
|
|
|
|
}
|
|
|
|
|
|
|
|
/***********************************************************************
|
2003-09-11 05:06:25 +02:00
|
|
|
* CoGetMarshalSizeMax [OLE32.@]
|
2002-02-05 19:11:17 +01:00
|
|
|
*/
|
|
|
|
HRESULT WINAPI
|
|
|
|
CoGetMarshalSizeMax(ULONG *pulSize, REFIID riid, IUnknown *pUnk,
|
|
|
|
DWORD dwDestContext, void *pvDestContext, DWORD mshlFlags
|
|
|
|
) {
|
|
|
|
HRESULT hres;
|
|
|
|
LPMARSHAL pMarshal;
|
|
|
|
|
|
|
|
hres = _GetMarshaller(riid,pUnk,dwDestContext,pvDestContext,mshlFlags,&pMarshal);
|
|
|
|
if (hres)
|
|
|
|
return hres;
|
|
|
|
hres = IMarshal_GetMarshalSizeMax(pMarshal,riid,pUnk,dwDestContext,pvDestContext,mshlFlags,pulSize);
|
|
|
|
*pulSize += sizeof(wine_marshal_id)+sizeof(wine_marshal_data)+sizeof(CLSID);
|
|
|
|
IMarshal_Release(pMarshal);
|
|
|
|
return hres;
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
/***********************************************************************
|
2003-09-11 05:06:25 +02:00
|
|
|
* CoMarshalInterface [OLE32.@]
|
2002-02-05 19:11:17 +01:00
|
|
|
*/
|
|
|
|
HRESULT WINAPI
|
|
|
|
CoMarshalInterface( IStream *pStm, REFIID riid, IUnknown *pUnk,
|
|
|
|
DWORD dwDestContext, void *pvDestContext, DWORD mshlflags
|
|
|
|
) {
|
|
|
|
HRESULT hres;
|
|
|
|
LPMARSHAL pMarshal;
|
|
|
|
CLSID xclsid;
|
|
|
|
ULONG writeres;
|
|
|
|
wine_marshal_id mid;
|
|
|
|
wine_marshal_data md;
|
|
|
|
ULONG res;
|
|
|
|
IUnknown *pUnknown;
|
|
|
|
|
|
|
|
TRACE("(%p, %s, %p, %lx, %p, %lx)\n",
|
|
|
|
pStm,debugstr_guid(riid),pUnk,dwDestContext,pvDestContext,mshlflags
|
|
|
|
);
|
2004-06-30 20:14:00 +02:00
|
|
|
|
|
|
|
if (pUnk == NULL)
|
|
|
|
return E_INVALIDARG;
|
|
|
|
|
2002-02-05 19:11:17 +01:00
|
|
|
STUBMGR_Start(); /* Just to be sure we have one running. */
|
|
|
|
mid.processid = GetCurrentProcessId();
|
2004-05-21 22:52:57 +02:00
|
|
|
IUnknown_QueryInterface(pUnk,&IID_IUnknown,(LPVOID*)&pUnknown);
|
2002-02-05 19:11:17 +01:00
|
|
|
mid.objectid = (DWORD)pUnknown;
|
|
|
|
IUnknown_Release(pUnknown);
|
|
|
|
memcpy(&mid.iid,riid,sizeof(mid.iid));
|
|
|
|
md.dwDestContext = dwDestContext;
|
|
|
|
md.mshlflags = mshlflags;
|
|
|
|
hres = IStream_Write(pStm,&mid,sizeof(mid),&res);
|
|
|
|
if (hres) return hres;
|
|
|
|
hres = IStream_Write(pStm,&md,sizeof(md),&res);
|
|
|
|
if (hres) return hres;
|
|
|
|
hres = _GetMarshaller(riid,pUnk,dwDestContext,pvDestContext,mshlflags,&pMarshal);
|
|
|
|
if (hres) {
|
|
|
|
FIXME("Failed to get marshaller, %lx?\n",hres);
|
|
|
|
return hres;
|
|
|
|
}
|
|
|
|
hres = IMarshal_GetUnmarshalClass(pMarshal,riid,pUnk,dwDestContext,pvDestContext,mshlflags,&xclsid);
|
|
|
|
if (hres) {
|
|
|
|
FIXME("IMarshal:GetUnmarshalClass failed, %lx\n",hres);
|
|
|
|
goto release_marshal;
|
|
|
|
}
|
|
|
|
hres = IStream_Write(pStm,&xclsid,sizeof(xclsid),&writeres);
|
|
|
|
if (hres) {
|
|
|
|
FIXME("Stream write failed, %lx\n",hres);
|
|
|
|
goto release_marshal;
|
|
|
|
}
|
2004-07-23 21:10:13 +02:00
|
|
|
|
|
|
|
TRACE("Calling IMarshal::MarshalInterace\n");
|
2002-02-05 19:11:17 +01:00
|
|
|
hres = IMarshal_MarshalInterface(pMarshal,pStm,riid,pUnk,dwDestContext,pvDestContext,mshlflags);
|
2004-07-23 21:10:13 +02:00
|
|
|
|
2002-02-05 19:11:17 +01:00
|
|
|
if (hres) {
|
2004-06-17 21:55:24 +02:00
|
|
|
if (IsEqualGUID(riid,&IID_IOleObject)) {
|
|
|
|
ERR("WINE currently cannot marshal IOleObject interfaces. This means you cannot embed/link OLE objects between applications.\n");
|
2002-05-16 22:26:40 +02:00
|
|
|
} else {
|
2004-06-17 21:55:24 +02:00
|
|
|
FIXME("Failed to marshal the interface %s, %lx?\n",debugstr_guid(riid),hres);
|
2002-05-16 22:26:40 +02:00
|
|
|
}
|
2002-02-05 19:11:17 +01:00
|
|
|
}
|
|
|
|
release_marshal:
|
|
|
|
IMarshal_Release(pMarshal);
|
|
|
|
return hres;
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
/***********************************************************************
|
2003-09-11 05:06:25 +02:00
|
|
|
* CoUnmarshalInterface [OLE32.@]
|
2002-02-05 19:11:17 +01:00
|
|
|
*/
|
|
|
|
HRESULT WINAPI
|
|
|
|
CoUnmarshalInterface(IStream *pStm, REFIID riid, LPVOID *ppv) {
|
|
|
|
HRESULT hres;
|
|
|
|
wine_marshal_id mid;
|
|
|
|
wine_marshal_data md;
|
|
|
|
ULONG res;
|
|
|
|
LPMARSHAL pMarshal;
|
|
|
|
LPUNKNOWN pUnk;
|
|
|
|
CLSID xclsid;
|
|
|
|
|
|
|
|
TRACE("(%p,%s,%p)\n",pStm,debugstr_guid(riid),ppv);
|
|
|
|
|
|
|
|
hres = IStream_Read(pStm,&mid,sizeof(mid),&res);
|
|
|
|
if (hres) {
|
|
|
|
FIXME("Stream read 1 failed, %lx, (%ld of %d)\n",hres,res,sizeof(mid));
|
|
|
|
return hres;
|
|
|
|
}
|
|
|
|
hres = IStream_Read(pStm,&md,sizeof(md),&res);
|
|
|
|
if (hres) {
|
|
|
|
FIXME("Stream read 2 failed, %lx, (%ld of %d)\n",hres,res,sizeof(md));
|
|
|
|
return hres;
|
|
|
|
}
|
|
|
|
hres = IStream_Read(pStm,&xclsid,sizeof(xclsid),&res);
|
|
|
|
if (hres) {
|
|
|
|
FIXME("Stream read 3 failed, %lx, (%ld of %d)\n",hres,res,sizeof(xclsid));
|
|
|
|
return hres;
|
|
|
|
}
|
2004-05-21 22:52:57 +02:00
|
|
|
hres=CoCreateInstance(&xclsid,NULL,CLSCTX_INPROC_SERVER | CLSCTX_INPROC_HANDLER | CLSCTX_LOCAL_SERVER,&IID_IMarshal,(void**)&pUnk);
|
2002-02-05 19:11:17 +01:00
|
|
|
if (hres) {
|
|
|
|
FIXME("Failed to create instance of unmarshaller %s.\n",debugstr_guid(&xclsid));
|
|
|
|
return hres;
|
|
|
|
}
|
|
|
|
hres = _GetMarshaller(riid,pUnk,md.dwDestContext,NULL,md.mshlflags,&pMarshal);
|
|
|
|
if (hres) {
|
|
|
|
FIXME("Failed to get unmarshaller, %lx?\n",hres);
|
|
|
|
return hres;
|
|
|
|
}
|
|
|
|
hres = IMarshal_UnmarshalInterface(pMarshal,pStm,riid,ppv);
|
|
|
|
if (hres) {
|
|
|
|
FIXME("Failed to Unmarshal the interface, %lx?\n",hres);
|
|
|
|
goto release_marshal;
|
|
|
|
}
|
|
|
|
release_marshal:
|
|
|
|
IMarshal_Release(pMarshal);
|
|
|
|
return hres;
|
|
|
|
}
|
|
|
|
|
2004-05-02 06:20:37 +02:00
|
|
|
/***********************************************************************
|
|
|
|
* CoReleaseMarshalData [OLE32.@]
|
|
|
|
*/
|
|
|
|
HRESULT WINAPI
|
|
|
|
CoReleaseMarshalData(IStream *pStm) {
|
|
|
|
HRESULT hres;
|
|
|
|
wine_marshal_id mid;
|
|
|
|
wine_marshal_data md;
|
|
|
|
ULONG res;
|
|
|
|
LPMARSHAL pMarshal;
|
|
|
|
LPUNKNOWN pUnk;
|
|
|
|
CLSID xclsid;
|
|
|
|
|
|
|
|
TRACE("(%p)\n",pStm);
|
|
|
|
|
|
|
|
hres = IStream_Read(pStm,&mid,sizeof(mid),&res);
|
|
|
|
if (hres) {
|
|
|
|
FIXME("Stream read 1 failed, %lx, (%ld of %d)\n",hres,res,sizeof(mid));
|
|
|
|
return hres;
|
|
|
|
}
|
|
|
|
hres = IStream_Read(pStm,&md,sizeof(md),&res);
|
|
|
|
if (hres) {
|
|
|
|
FIXME("Stream read 2 failed, %lx, (%ld of %d)\n",hres,res,sizeof(md));
|
|
|
|
return hres;
|
|
|
|
}
|
|
|
|
hres = IStream_Read(pStm,&xclsid,sizeof(xclsid),&res);
|
|
|
|
if (hres) {
|
|
|
|
FIXME("Stream read 3 failed, %lx, (%ld of %d)\n",hres,res,sizeof(xclsid));
|
|
|
|
return hres;
|
|
|
|
}
|
|
|
|
hres=CoCreateInstance(&xclsid,NULL,CLSCTX_INPROC_SERVER | CLSCTX_INPROC_HANDLER | CLSCTX_LOCAL_SERVER,&IID_IMarshal,(void**)(char*)&pUnk);
|
|
|
|
if (hres) {
|
|
|
|
FIXME("Failed to create instance of unmarshaller %s.\n",debugstr_guid(&xclsid));
|
|
|
|
return hres;
|
|
|
|
}
|
|
|
|
hres = IUnknown_QueryInterface(pUnk,&IID_IMarshal,(LPVOID*)(char*)&pMarshal);
|
|
|
|
if (hres) {
|
|
|
|
FIXME("Failed to get IMarshal iface, %lx?\n",hres);
|
|
|
|
return hres;
|
|
|
|
}
|
|
|
|
hres = IMarshal_ReleaseMarshalData(pMarshal,pStm);
|
|
|
|
if (hres) {
|
|
|
|
FIXME("Failed to releasemarshaldata the interface, %lx?\n",hres);
|
|
|
|
}
|
|
|
|
IMarshal_Release(pMarshal);
|
|
|
|
IUnknown_Release(pUnk);
|
|
|
|
return hres;
|
|
|
|
}
|
|
|
|
|
|
|
|
|
2002-02-05 19:11:17 +01:00
|
|
|
/***********************************************************************
|
2003-09-11 05:06:25 +02:00
|
|
|
* CoMarshalInterThreadInterfaceInStream [OLE32.@]
|
2002-02-05 19:11:17 +01:00
|
|
|
*
|
2004-05-29 02:22:55 +02:00
|
|
|
* Marshal an interface across threads in the same process.
|
|
|
|
*
|
|
|
|
* PARAMS
|
|
|
|
* riid [I] Identifier of the interface to be marshalled.
|
|
|
|
* pUnk [I] Pointer to IUnknown-derived interface that will be marshalled.
|
|
|
|
* ppStm [O] Pointer to IStream object that is created and then used to store the marshalled inteface.
|
|
|
|
*
|
|
|
|
* RETURNS
|
|
|
|
* Success: S_OK
|
|
|
|
* Failure: E_OUTOFMEMORY and other COM error codes
|
|
|
|
*
|
|
|
|
* SEE
|
|
|
|
* CoMarshalInterface(), CoUnmarshalInterface() and CoGetInterfaceAndReleaseStream()
|
2002-02-05 19:11:17 +01:00
|
|
|
*/
|
|
|
|
HRESULT WINAPI
|
|
|
|
CoMarshalInterThreadInterfaceInStream(
|
2004-05-29 02:22:55 +02:00
|
|
|
REFIID riid, LPUNKNOWN pUnk, LPSTREAM * ppStm)
|
|
|
|
{
|
|
|
|
ULARGE_INTEGER xpos;
|
|
|
|
LARGE_INTEGER seekto;
|
|
|
|
HRESULT hres;
|
2002-02-05 19:11:17 +01:00
|
|
|
|
2004-05-29 02:22:55 +02:00
|
|
|
TRACE("(%s, %p, %p)\n",debugstr_guid(riid), pUnk, ppStm);
|
|
|
|
|
|
|
|
hres = CreateStreamOnHGlobal(0, TRUE, ppStm);
|
|
|
|
if (FAILED(hres)) return hres;
|
|
|
|
hres = CoMarshalInterface(*ppStm, riid, pUnk, MSHCTX_INPROC, NULL, MSHLFLAGS_NORMAL);
|
|
|
|
|
|
|
|
/* FIXME: is this needed? */
|
|
|
|
memset(&seekto,0,sizeof(seekto));
|
|
|
|
IStream_Seek(*ppStm,seekto,SEEK_SET,&xpos);
|
|
|
|
|
|
|
|
return hres;
|
2002-02-05 19:11:17 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
/***********************************************************************
|
2003-09-11 05:06:25 +02:00
|
|
|
* CoGetInterfaceAndReleaseStream [OLE32.@]
|
2004-05-29 02:22:55 +02:00
|
|
|
*
|
|
|
|
* Unmarshalls an inteface from a stream and then releases the stream.
|
|
|
|
*
|
|
|
|
* PARAMS
|
|
|
|
* pStm [I] Stream that contains the marshalled inteface.
|
|
|
|
* riid [I] Interface identifier of the object to unmarshall.
|
|
|
|
* ppv [O] Address of pointer where the requested interface object will be stored.
|
|
|
|
*
|
|
|
|
* RETURNS
|
|
|
|
* Success: S_OK
|
|
|
|
* Failure: A COM error code
|
|
|
|
*
|
|
|
|
* SEE
|
|
|
|
* CoMarshalInterThreadInterfaceInStream() and CoUnmarshalInteface()
|
2002-02-05 19:11:17 +01:00
|
|
|
*/
|
|
|
|
HRESULT WINAPI
|
2004-05-29 02:22:55 +02:00
|
|
|
CoGetInterfaceAndReleaseStream(LPSTREAM pStm,REFIID riid, LPVOID *ppv)
|
|
|
|
{
|
|
|
|
HRESULT hres;
|
2002-02-05 19:11:17 +01:00
|
|
|
|
2004-05-29 02:22:55 +02:00
|
|
|
TRACE("(%p, %s, %p)\n", pStm, debugstr_guid(riid), ppv);
|
|
|
|
|
|
|
|
hres = CoUnmarshalInterface(pStm, riid, ppv);
|
|
|
|
IStream_Release(pStm);
|
|
|
|
return hres;
|
2002-02-05 19:11:17 +01:00
|
|
|
}
|
|
|
|
|
2002-02-22 22:22:06 +01:00
|
|
|
static HRESULT WINAPI
|
2002-02-05 19:11:17 +01:00
|
|
|
SMCF_QueryInterface(LPCLASSFACTORY iface,REFIID riid, LPVOID *ppv) {
|
|
|
|
*ppv = NULL;
|
|
|
|
if (IsEqualIID(riid,&IID_IUnknown) || IsEqualIID(riid,&IID_IClassFactory)) {
|
|
|
|
*ppv = (LPVOID)iface;
|
|
|
|
return S_OK;
|
|
|
|
}
|
|
|
|
return E_NOINTERFACE;
|
|
|
|
}
|
2002-02-22 22:22:06 +01:00
|
|
|
static ULONG WINAPI SMCF_AddRef(LPCLASSFACTORY iface) { return 2; }
|
|
|
|
static ULONG WINAPI SMCF_Release(LPCLASSFACTORY iface) { return 1; }
|
2002-02-05 19:11:17 +01:00
|
|
|
|
2002-02-22 22:22:06 +01:00
|
|
|
static HRESULT WINAPI
|
2002-02-05 19:11:17 +01:00
|
|
|
SMCF_CreateInstance(
|
|
|
|
LPCLASSFACTORY iface, LPUNKNOWN pUnk, REFIID riid, LPVOID *ppv
|
|
|
|
) {
|
|
|
|
if (IsEqualIID(riid,&IID_IMarshal)) {
|
|
|
|
StdMarshalImpl *dm;
|
|
|
|
dm=(StdMarshalImpl*)HeapAlloc(GetProcessHeap(),0,sizeof(StdMarshalImpl));
|
2002-06-01 01:06:46 +02:00
|
|
|
if (!dm)
|
2002-02-05 19:11:17 +01:00
|
|
|
return E_FAIL;
|
|
|
|
dm->lpvtbl = &stdmvtbl;
|
|
|
|
dm->ref = 1;
|
|
|
|
*ppv = (LPVOID)dm;
|
|
|
|
return S_OK;
|
|
|
|
}
|
|
|
|
FIXME("(%s), not supported.\n",debugstr_guid(riid));
|
|
|
|
return E_NOINTERFACE;
|
|
|
|
}
|
|
|
|
|
2002-02-22 22:22:06 +01:00
|
|
|
static HRESULT WINAPI
|
2002-02-05 19:11:17 +01:00
|
|
|
SMCF_LockServer(LPCLASSFACTORY iface, BOOL fLock) {
|
|
|
|
FIXME("(%d), stub!\n",fLock);
|
|
|
|
return S_OK;
|
|
|
|
}
|
|
|
|
|
|
|
|
static ICOM_VTABLE(IClassFactory) dfmarshalcfvtbl = {
|
|
|
|
ICOM_MSVTABLE_COMPAT_DummyRTTIVALUE
|
|
|
|
SMCF_QueryInterface,
|
|
|
|
SMCF_AddRef,
|
|
|
|
SMCF_Release,
|
|
|
|
SMCF_CreateInstance,
|
|
|
|
SMCF_LockServer
|
|
|
|
};
|
|
|
|
static ICOM_VTABLE(IClassFactory) *pdfmarshalcfvtbl = &dfmarshalcfvtbl;
|
|
|
|
|
|
|
|
HRESULT
|
|
|
|
MARSHAL_GetStandardMarshalCF(LPVOID *ppv) {
|
|
|
|
*ppv = &pdfmarshalcfvtbl;
|
|
|
|
return S_OK;
|
|
|
|
}
|