2247 lines
57 KiB
C
2247 lines
57 KiB
C
/*
|
|
* TYPELIB Marshaler
|
|
*
|
|
* Copyright 2002 Marcus Meissner
|
|
*
|
|
* The olerelay debug channel allows you to see calls marshalled by
|
|
* the typelib marshaller. It is not a generic COM relaying system.
|
|
*
|
|
* 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
|
|
*/
|
|
|
|
#include "config.h"
|
|
|
|
#include <assert.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
#include <stdarg.h>
|
|
#include <stdio.h>
|
|
#include <ctype.h>
|
|
|
|
#define COBJMACROS
|
|
#define NONAMELESSUNION
|
|
#define NONAMELESSSTRUCT
|
|
|
|
#include "winerror.h"
|
|
#include "windef.h"
|
|
#include "winbase.h"
|
|
#include "winnls.h"
|
|
#include "winreg.h"
|
|
#include "winuser.h"
|
|
|
|
#include "ole2.h"
|
|
#include "typelib.h"
|
|
#include "wine/debug.h"
|
|
|
|
static const WCHAR riidW[5] = {'r','i','i','d',0};
|
|
static const WCHAR pdispparamsW[] = {'p','d','i','s','p','p','a','r','a','m','s',0};
|
|
static const WCHAR ppvObjectW[] = {'p','p','v','O','b','j','e','c','t',0};
|
|
static const WCHAR IDispatchW[] = { 'I','D','i','s','p','a','t','c','h',0};
|
|
static const WCHAR GetIDsOfNamesW[] = { 'G','e','t','I','D','s','O','f','N','a','m','e','s',0};
|
|
|
|
WINE_DEFAULT_DEBUG_CHANNEL(ole);
|
|
WINE_DECLARE_DEBUG_CHANNEL(olerelay);
|
|
|
|
#define ICOM_THIS_MULTI(impl,field,iface) impl* const This=(impl*)((char*)(iface) - offsetof(impl,field))
|
|
|
|
typedef struct _marshal_state {
|
|
LPBYTE base;
|
|
int size;
|
|
int curoff;
|
|
|
|
BOOL thisisiid;
|
|
IID iid; /* HACK: for VT_VOID */
|
|
} marshal_state;
|
|
|
|
/* used in the olerelay code to avoid having the L"" stuff added by debugstr_w */
|
|
static char *relaystr(WCHAR *in) {
|
|
char *tmp = (char *)debugstr_w(in);
|
|
tmp += 2;
|
|
tmp[strlen(tmp)-1] = '\0';
|
|
return tmp;
|
|
}
|
|
|
|
static HRESULT
|
|
xbuf_add(marshal_state *buf, LPBYTE stuff, DWORD size) {
|
|
while (buf->size - buf->curoff < size) {
|
|
if (buf->base) {
|
|
buf->size += 100;
|
|
buf->base = HeapReAlloc(GetProcessHeap(),HEAP_ZERO_MEMORY,buf->base,buf->size);
|
|
if (!buf->base)
|
|
return E_OUTOFMEMORY;
|
|
} else {
|
|
buf->base = HeapAlloc(GetProcessHeap(),HEAP_ZERO_MEMORY,32);
|
|
buf->size = 32;
|
|
if (!buf->base)
|
|
return E_OUTOFMEMORY;
|
|
}
|
|
}
|
|
memcpy(buf->base+buf->curoff,stuff,size);
|
|
buf->curoff += size;
|
|
return S_OK;
|
|
}
|
|
|
|
static HRESULT
|
|
xbuf_get(marshal_state *buf, LPBYTE stuff, DWORD size) {
|
|
if (buf->size < buf->curoff+size) return E_FAIL;
|
|
memcpy(stuff,buf->base+buf->curoff,size);
|
|
buf->curoff += size;
|
|
return S_OK;
|
|
}
|
|
|
|
static HRESULT
|
|
xbuf_skip(marshal_state *buf, DWORD size) {
|
|
if (buf->size < buf->curoff+size) return E_FAIL;
|
|
buf->curoff += size;
|
|
return S_OK;
|
|
}
|
|
|
|
static HRESULT
|
|
_unmarshal_interface(marshal_state *buf, REFIID riid, LPUNKNOWN *pUnk) {
|
|
IStream *pStm;
|
|
ULARGE_INTEGER newpos;
|
|
LARGE_INTEGER seekto;
|
|
ULONG res;
|
|
HRESULT hres;
|
|
DWORD xsize;
|
|
|
|
TRACE("...%s...\n",debugstr_guid(riid));
|
|
|
|
*pUnk = NULL;
|
|
hres = xbuf_get(buf,(LPBYTE)&xsize,sizeof(xsize));
|
|
if (hres) {
|
|
ERR("xbuf_get failed\n");
|
|
return hres;
|
|
}
|
|
|
|
if (xsize == 0) return S_OK;
|
|
|
|
hres = CreateStreamOnHGlobal(0,TRUE,&pStm);
|
|
if (hres) {
|
|
ERR("Stream create failed %lx\n",hres);
|
|
return hres;
|
|
}
|
|
|
|
hres = IStream_Write(pStm,buf->base+buf->curoff,xsize,&res);
|
|
if (hres) {
|
|
ERR("stream write %lx\n",hres);
|
|
return hres;
|
|
}
|
|
|
|
memset(&seekto,0,sizeof(seekto));
|
|
hres = IStream_Seek(pStm,seekto,SEEK_SET,&newpos);
|
|
if (hres) {
|
|
ERR("Failed Seek %lx\n",hres);
|
|
return hres;
|
|
}
|
|
|
|
hres = CoUnmarshalInterface(pStm,riid,(LPVOID*)pUnk);
|
|
if (hres) {
|
|
ERR("Unmarshalling interface %s failed with %lx\n",debugstr_guid(riid),hres);
|
|
return hres;
|
|
}
|
|
|
|
IStream_Release(pStm);
|
|
return xbuf_skip(buf,xsize);
|
|
}
|
|
|
|
static HRESULT
|
|
_marshal_interface(marshal_state *buf, REFIID riid, LPUNKNOWN pUnk) {
|
|
LPUNKNOWN newiface = NULL;
|
|
LPBYTE tempbuf = NULL;
|
|
IStream *pStm = NULL;
|
|
STATSTG ststg;
|
|
ULARGE_INTEGER newpos;
|
|
LARGE_INTEGER seekto;
|
|
ULONG res;
|
|
DWORD xsize;
|
|
HRESULT hres;
|
|
|
|
hres = E_FAIL;
|
|
if (!pUnk) {
|
|
ERR("pUnk is NULL?\n");
|
|
goto fail;
|
|
}
|
|
|
|
TRACE("...%s...\n",debugstr_guid(riid));
|
|
hres = IUnknown_QueryInterface(pUnk,riid,(LPVOID*)&newiface);
|
|
if (hres) {
|
|
WARN("%p does not support iface %s\n",pUnk,debugstr_guid(riid));
|
|
goto fail;
|
|
}
|
|
|
|
hres = CreateStreamOnHGlobal(0,TRUE,&pStm);
|
|
if (hres) {
|
|
ERR("Stream create failed %lx\n",hres);
|
|
goto fail;
|
|
}
|
|
|
|
hres = CoMarshalInterface(pStm,riid,newiface,0,NULL,0);
|
|
if (hres) {
|
|
ERR("Marshalling interface %s failed with %lx\n", debugstr_guid(riid), hres);
|
|
goto fail;
|
|
}
|
|
|
|
hres = IStream_Stat(pStm,&ststg,0);
|
|
if (hres) {
|
|
ERR("Stream stat failed\n");
|
|
goto fail;
|
|
}
|
|
|
|
tempbuf = HeapAlloc(GetProcessHeap(), 0, ststg.cbSize.u.LowPart);
|
|
memset(&seekto,0,sizeof(seekto));
|
|
hres = IStream_Seek(pStm,seekto,SEEK_SET,&newpos);
|
|
if (hres) {
|
|
ERR("Failed Seek %lx\n",hres);
|
|
goto fail;
|
|
}
|
|
|
|
hres = IStream_Read(pStm,tempbuf,ststg.cbSize.u.LowPart,&res);
|
|
if (hres) {
|
|
ERR("Failed Read %lx\n",hres);
|
|
goto fail;
|
|
}
|
|
|
|
xsize = ststg.cbSize.u.LowPart;
|
|
xbuf_add(buf,(LPBYTE)&xsize,sizeof(xsize));
|
|
hres = xbuf_add(buf,tempbuf,ststg.cbSize.u.LowPart);
|
|
|
|
HeapFree(GetProcessHeap(),0,tempbuf);
|
|
IUnknown_Release(newiface);
|
|
IStream_Release(pStm);
|
|
|
|
return hres;
|
|
|
|
fail:
|
|
xsize = 0;
|
|
xbuf_add(buf,(LPBYTE)&xsize,sizeof(xsize));
|
|
if (pStm) IUnknown_Release(pStm);
|
|
if (newiface) IUnknown_Release(newiface);
|
|
HeapFree(GetProcessHeap(), 0, tempbuf);
|
|
return hres;
|
|
}
|
|
|
|
/********************* 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
|
|
_get_typeinfo_for_iid(REFIID riid, ITypeInfo**ti) {
|
|
HRESULT hres;
|
|
HKEY ikey;
|
|
char tlguid[200],typelibkey[300],interfacekey[300],ver[100];
|
|
char tlfn[260];
|
|
OLECHAR tlfnW[260];
|
|
DWORD tlguidlen, verlen, type, tlfnlen;
|
|
ITypeLib *tl;
|
|
|
|
sprintf( interfacekey, "Interface\\{%08lx-%04x-%04x-%02x%02x-%02x%02x%02x%02x%02x%02x}\\Typelib",
|
|
riid->Data1, riid->Data2, riid->Data3,
|
|
riid->Data4[0], riid->Data4[1], riid->Data4[2], riid->Data4[3],
|
|
riid->Data4[4], riid->Data4[5], riid->Data4[6], riid->Data4[7]
|
|
);
|
|
|
|
if (RegOpenKeyA(HKEY_CLASSES_ROOT,interfacekey,&ikey)) {
|
|
ERR("No %s key found.\n",interfacekey);
|
|
return E_FAIL;
|
|
}
|
|
type = (1<<REG_SZ);
|
|
tlguidlen = sizeof(tlguid);
|
|
if (RegQueryValueExA(ikey,NULL,NULL,&type,tlguid,&tlguidlen)) {
|
|
ERR("Getting typelib guid failed.\n");
|
|
RegCloseKey(ikey);
|
|
return E_FAIL;
|
|
}
|
|
type = (1<<REG_SZ);
|
|
verlen = sizeof(ver);
|
|
if (RegQueryValueExA(ikey,"Version",NULL,&type,ver,&verlen)) {
|
|
ERR("Could not get version value?\n");
|
|
RegCloseKey(ikey);
|
|
return E_FAIL;
|
|
}
|
|
RegCloseKey(ikey);
|
|
sprintf(typelibkey,"Typelib\\%s\\%s\\0\\win32",tlguid,ver);
|
|
tlfnlen = sizeof(tlfn);
|
|
if (RegQueryValueA(HKEY_CLASSES_ROOT,typelibkey,tlfn,&tlfnlen)) {
|
|
ERR("Could not get typelib fn?\n");
|
|
return E_FAIL;
|
|
}
|
|
MultiByteToWideChar(CP_ACP, 0, tlfn, -1, tlfnW, -1);
|
|
hres = LoadTypeLib(tlfnW,&tl);
|
|
if (hres) {
|
|
ERR("Failed to load typelib for %s, but it should be there.\n",debugstr_guid(riid));
|
|
return hres;
|
|
}
|
|
hres = ITypeLib_GetTypeInfoOfGuid(tl,riid,ti);
|
|
if (hres) {
|
|
ERR("typelib does not contain info for %s?\n",debugstr_guid(riid));
|
|
ITypeLib_Release(tl);
|
|
return hres;
|
|
}
|
|
/* FIXME: do this? ITypeLib_Release(tl); */
|
|
return hres;
|
|
}
|
|
|
|
/* Determine nr of functions. Since we use the toplevel interface and all
|
|
* inherited ones have lower numbers, we are ok to not to descent into
|
|
* the inheritance tree I think.
|
|
*/
|
|
static int _nroffuncs(ITypeInfo *tinfo) {
|
|
int n, max = 0;
|
|
FUNCDESC *fdesc;
|
|
HRESULT hres;
|
|
|
|
n=0;
|
|
while (1) {
|
|
hres = ITypeInfo_GetFuncDesc(tinfo,n,&fdesc);
|
|
if (hres)
|
|
return max+1;
|
|
if (fdesc->oVft/4 > max)
|
|
max = fdesc->oVft/4;
|
|
n++;
|
|
}
|
|
/*NOTREACHED*/
|
|
}
|
|
|
|
#ifdef __i386__
|
|
|
|
#include "pshpack1.h"
|
|
|
|
typedef struct _TMAsmProxy {
|
|
BYTE popleax;
|
|
BYTE pushlval;
|
|
BYTE nr;
|
|
BYTE pushleax;
|
|
BYTE lcall;
|
|
DWORD xcall;
|
|
BYTE lret;
|
|
WORD bytestopop;
|
|
} TMAsmProxy;
|
|
|
|
#include "poppack.h"
|
|
|
|
#else /* __i386__ */
|
|
# error You need to implement stubless proxies for your architecture
|
|
#endif
|
|
|
|
typedef struct _TMProxyImpl {
|
|
LPVOID *lpvtbl;
|
|
IRpcProxyBufferVtbl *lpvtbl2;
|
|
ULONG ref;
|
|
|
|
TMAsmProxy *asmstubs;
|
|
ITypeInfo* tinfo;
|
|
IRpcChannelBuffer* chanbuf;
|
|
IID iid;
|
|
CRITICAL_SECTION crit;
|
|
IUnknown *outerunknown;
|
|
} TMProxyImpl;
|
|
|
|
static HRESULT WINAPI
|
|
TMProxyImpl_QueryInterface(LPRPCPROXYBUFFER iface, REFIID riid, LPVOID *ppv)
|
|
{
|
|
TRACE("()\n");
|
|
if (IsEqualIID(riid,&IID_IUnknown)||IsEqualIID(riid,&IID_IRpcProxyBuffer)) {
|
|
*ppv = (LPVOID)iface;
|
|
IRpcProxyBuffer_AddRef(iface);
|
|
return S_OK;
|
|
}
|
|
FIXME("no interface for %s\n",debugstr_guid(riid));
|
|
return E_NOINTERFACE;
|
|
}
|
|
|
|
static ULONG WINAPI
|
|
TMProxyImpl_AddRef(LPRPCPROXYBUFFER iface)
|
|
{
|
|
ICOM_THIS_MULTI(TMProxyImpl,lpvtbl2,iface);
|
|
ULONG refCount = InterlockedIncrement(&This->ref);
|
|
|
|
TRACE("(%p)->(ref before=%lu)\n",This, refCount - 1);
|
|
|
|
return refCount;
|
|
}
|
|
|
|
static ULONG WINAPI
|
|
TMProxyImpl_Release(LPRPCPROXYBUFFER iface)
|
|
{
|
|
ICOM_THIS_MULTI(TMProxyImpl,lpvtbl2,iface);
|
|
ULONG refCount = InterlockedDecrement(&This->ref);
|
|
|
|
TRACE("(%p)->(ref before=%lu)\n",This, refCount + 1);
|
|
|
|
if (!refCount)
|
|
{
|
|
DeleteCriticalSection(&This->crit);
|
|
if (This->chanbuf) IRpcChannelBuffer_Release(This->chanbuf);
|
|
VirtualFree(This->asmstubs, 0, MEM_RELEASE);
|
|
CoTaskMemFree(This);
|
|
}
|
|
return refCount;
|
|
}
|
|
|
|
static HRESULT WINAPI
|
|
TMProxyImpl_Connect(
|
|
LPRPCPROXYBUFFER iface,IRpcChannelBuffer* pRpcChannelBuffer)
|
|
{
|
|
ICOM_THIS_MULTI(TMProxyImpl, lpvtbl2, iface);
|
|
|
|
TRACE("(%p)\n", pRpcChannelBuffer);
|
|
|
|
EnterCriticalSection(&This->crit);
|
|
|
|
IRpcChannelBuffer_AddRef(pRpcChannelBuffer);
|
|
This->chanbuf = pRpcChannelBuffer;
|
|
|
|
LeaveCriticalSection(&This->crit);
|
|
|
|
return S_OK;
|
|
}
|
|
|
|
static void WINAPI
|
|
TMProxyImpl_Disconnect(LPRPCPROXYBUFFER iface)
|
|
{
|
|
ICOM_THIS_MULTI(TMProxyImpl, lpvtbl2, iface);
|
|
|
|
TRACE("()\n");
|
|
|
|
EnterCriticalSection(&This->crit);
|
|
|
|
IRpcChannelBuffer_Release(This->chanbuf);
|
|
This->chanbuf = NULL;
|
|
|
|
LeaveCriticalSection(&This->crit);
|
|
}
|
|
|
|
|
|
static IRpcProxyBufferVtbl tmproxyvtable = {
|
|
TMProxyImpl_QueryInterface,
|
|
TMProxyImpl_AddRef,
|
|
TMProxyImpl_Release,
|
|
TMProxyImpl_Connect,
|
|
TMProxyImpl_Disconnect
|
|
};
|
|
|
|
/* how much space do we use on stack in DWORD steps. */
|
|
int
|
|
_argsize(DWORD vt) {
|
|
switch (vt) {
|
|
case VT_DATE:
|
|
return sizeof(DATE)/sizeof(DWORD);
|
|
case VT_VARIANT:
|
|
return (sizeof(VARIANT)+3)/sizeof(DWORD);
|
|
default:
|
|
return 1;
|
|
}
|
|
}
|
|
|
|
static int
|
|
_xsize(TYPEDESC *td) {
|
|
switch (td->vt) {
|
|
case VT_DATE:
|
|
return sizeof(DATE);
|
|
case VT_VARIANT:
|
|
return sizeof(VARIANT)+3;
|
|
case VT_CARRAY: {
|
|
int i, arrsize = 1;
|
|
ARRAYDESC *adesc = td->u.lpadesc;
|
|
|
|
for (i=0;i<adesc->cDims;i++)
|
|
arrsize *= adesc->rgbounds[i].cElements;
|
|
return arrsize*_xsize(&adesc->tdescElem);
|
|
}
|
|
case VT_UI2:
|
|
case VT_I2:
|
|
return 2;
|
|
case VT_UI1:
|
|
case VT_I1:
|
|
return 1;
|
|
default:
|
|
return 4;
|
|
}
|
|
}
|
|
|
|
static HRESULT
|
|
serialize_param(
|
|
ITypeInfo *tinfo,
|
|
BOOL writeit,
|
|
BOOL debugout,
|
|
BOOL dealloc,
|
|
TYPEDESC *tdesc,
|
|
DWORD *arg,
|
|
marshal_state *buf)
|
|
{
|
|
HRESULT hres = S_OK;
|
|
|
|
TRACE("(tdesc.vt %d)\n",tdesc->vt);
|
|
|
|
switch (tdesc->vt) {
|
|
case VT_EMPTY: /* nothing. empty variant for instance */
|
|
return S_OK;
|
|
case VT_BOOL:
|
|
case VT_ERROR:
|
|
case VT_UINT:
|
|
case VT_I4:
|
|
case VT_R4:
|
|
case VT_UI4:
|
|
hres = S_OK;
|
|
if (debugout) TRACE_(olerelay)("%lx",*arg);
|
|
if (writeit)
|
|
hres = xbuf_add(buf,(LPBYTE)arg,sizeof(DWORD));
|
|
return hres;
|
|
case VT_I2:
|
|
case VT_UI2:
|
|
hres = S_OK;
|
|
if (debugout) TRACE_(olerelay)("%04lx",*arg & 0xffff);
|
|
if (writeit)
|
|
hres = xbuf_add(buf,(LPBYTE)arg,sizeof(DWORD));
|
|
return hres;
|
|
case VT_I1:
|
|
case VT_UI1:
|
|
hres = S_OK;
|
|
if (debugout) TRACE_(olerelay)("%02lx",*arg & 0xff);
|
|
if (writeit)
|
|
hres = xbuf_add(buf,(LPBYTE)arg,sizeof(DWORD));
|
|
return hres;
|
|
case VT_VARIANT: {
|
|
TYPEDESC tdesc2;
|
|
VARIANT *vt = (VARIANT*)arg;
|
|
DWORD vttype = V_VT(vt);
|
|
|
|
if (debugout) TRACE_(olerelay)("Vt(%ld)(",vttype);
|
|
tdesc2.vt = vttype;
|
|
if (writeit) {
|
|
hres = xbuf_add(buf,(LPBYTE)&vttype,sizeof(vttype));
|
|
if (hres) return hres;
|
|
}
|
|
/* need to recurse since we need to free the stuff */
|
|
hres = serialize_param(tinfo,writeit,debugout,dealloc,&tdesc2,&(V_I4(vt)),buf);
|
|
if (debugout) TRACE_(olerelay)(")");
|
|
return hres;
|
|
}
|
|
case VT_BSTR|VT_BYREF: {
|
|
if (debugout) TRACE_(olerelay)("[byref]'%s'", *arg ? relaystr(*((BSTR*)*arg)) : "<bstr NULL>");
|
|
if (writeit) {
|
|
/* ptr to ptr to magic widestring, basically */
|
|
BSTR *bstr = (BSTR *) *arg;
|
|
if (!bstr) {
|
|
/* -1 means "null string" which is equivalent to empty string */
|
|
DWORD fakelen = -1;
|
|
xbuf_add(buf, (LPBYTE)&fakelen,4);
|
|
} else {
|
|
/* BSTRs store the length behind the first character */
|
|
DWORD *len = ((DWORD *)(*bstr))-1;
|
|
hres = xbuf_add(buf, (LPBYTE) len, *len + 4);
|
|
if (hres) return hres;
|
|
}
|
|
}
|
|
|
|
if (dealloc && arg) {
|
|
BSTR *str = *((BSTR **)arg);
|
|
SysFreeString(*str);
|
|
}
|
|
return S_OK;
|
|
}
|
|
|
|
case VT_BSTR: {
|
|
if (debugout) {
|
|
if (*arg)
|
|
TRACE_(olerelay)("%s",relaystr((WCHAR*)*arg));
|
|
else
|
|
TRACE_(olerelay)("<bstr NULL>");
|
|
}
|
|
if (writeit) {
|
|
if (!*arg) {
|
|
DWORD fakelen = -1;
|
|
hres = xbuf_add(buf,(LPBYTE)&fakelen,4);
|
|
if (hres)
|
|
return hres;
|
|
} else {
|
|
DWORD *bstr = ((DWORD*)(*arg))-1;
|
|
|
|
hres = xbuf_add(buf,(LPBYTE)bstr,bstr[0]+4);
|
|
if (hres)
|
|
return hres;
|
|
}
|
|
}
|
|
|
|
if (dealloc && arg)
|
|
SysFreeString((BSTR)*arg);
|
|
return S_OK;
|
|
}
|
|
case VT_PTR: {
|
|
DWORD cookie;
|
|
|
|
if (debugout) TRACE_(olerelay)("*");
|
|
/* Write always, so the other side knows when it gets a NULL pointer.
|
|
*/
|
|
cookie = *arg ? 0x42424242 : 0;
|
|
hres = xbuf_add(buf,(LPBYTE)&cookie,sizeof(cookie));
|
|
if (hres)
|
|
return hres;
|
|
if (!*arg) {
|
|
if (debugout) TRACE_(olerelay)("NULL");
|
|
return S_OK;
|
|
}
|
|
hres = serialize_param(tinfo,writeit,debugout,dealloc,tdesc->u.lptdesc,(DWORD*)*arg,buf);
|
|
if (dealloc) HeapFree(GetProcessHeap(),0,(LPVOID)arg);
|
|
return hres;
|
|
}
|
|
case VT_UNKNOWN:
|
|
if (debugout) TRACE_(olerelay)("unk(0x%lx)",*arg);
|
|
if (writeit)
|
|
hres = _marshal_interface(buf,&IID_IUnknown,(LPUNKNOWN)*arg);
|
|
return hres;
|
|
case VT_DISPATCH:
|
|
if (debugout) TRACE_(olerelay)("idisp(0x%lx)",*arg);
|
|
if (writeit)
|
|
hres = _marshal_interface(buf,&IID_IDispatch,(LPUNKNOWN)*arg);
|
|
return hres;
|
|
case VT_VOID:
|
|
if (debugout) TRACE_(olerelay)("<void>");
|
|
return S_OK;
|
|
case VT_USERDEFINED: {
|
|
ITypeInfo *tinfo2;
|
|
TYPEATTR *tattr;
|
|
|
|
hres = ITypeInfo_GetRefTypeInfo(tinfo,tdesc->u.hreftype,&tinfo2);
|
|
if (hres) {
|
|
ERR("Could not get typeinfo of hreftype %lx for VT_USERDEFINED.\n",tdesc->u.hreftype);
|
|
return hres;
|
|
}
|
|
ITypeInfo_GetTypeAttr(tinfo2,&tattr);
|
|
switch (tattr->typekind) {
|
|
case TKIND_DISPATCH:
|
|
case TKIND_INTERFACE:
|
|
if (writeit)
|
|
hres=_marshal_interface(buf,&(tattr->guid),(LPUNKNOWN)arg);
|
|
break;
|
|
case TKIND_RECORD: {
|
|
int i;
|
|
if (debugout) TRACE_(olerelay)("{");
|
|
for (i=0;i<tattr->cVars;i++) {
|
|
VARDESC *vdesc;
|
|
ELEMDESC *elem2;
|
|
TYPEDESC *tdesc2;
|
|
|
|
hres = ITypeInfo2_GetVarDesc(tinfo2, i, &vdesc);
|
|
if (hres) {
|
|
ERR("Could not get vardesc of %d\n",i);
|
|
return hres;
|
|
}
|
|
/* Need them for hack below */
|
|
/*
|
|
memset(names,0,sizeof(names));
|
|
hres = ITypeInfo_GetNames(tinfo2,vdesc->memid,names,sizeof(names)/sizeof(names[0]),&nrofnames);
|
|
if (nrofnames > sizeof(names)/sizeof(names[0])) {
|
|
ERR("Need more names!\n");
|
|
}
|
|
if (!hres && debugout)
|
|
TRACE_(olerelay)("%s=",relaystr(names[0]));
|
|
*/
|
|
elem2 = &vdesc->elemdescVar;
|
|
tdesc2 = &elem2->tdesc;
|
|
hres = serialize_param(
|
|
tinfo2,
|
|
writeit,
|
|
debugout,
|
|
dealloc,
|
|
tdesc2,
|
|
(DWORD*)(((LPBYTE)arg)+vdesc->u.oInst),
|
|
buf
|
|
);
|
|
if (hres!=S_OK)
|
|
return hres;
|
|
if (debugout && (i<(tattr->cVars-1)))
|
|
TRACE_(olerelay)(",");
|
|
}
|
|
if (buf->thisisiid && (tattr->cbSizeInstance==sizeof(GUID)))
|
|
memcpy(&(buf->iid),arg,sizeof(buf->iid));
|
|
if (debugout) TRACE_(olerelay)("}");
|
|
break;
|
|
}
|
|
default:
|
|
FIXME("Unhandled typekind %d\n",tattr->typekind);
|
|
hres = E_FAIL;
|
|
break;
|
|
}
|
|
ITypeInfo_Release(tinfo2);
|
|
return hres;
|
|
}
|
|
case VT_CARRAY: {
|
|
ARRAYDESC *adesc = tdesc->u.lpadesc;
|
|
int i, arrsize = 1;
|
|
|
|
if (debugout) TRACE_(olerelay)("carr");
|
|
for (i=0;i<adesc->cDims;i++) {
|
|
if (debugout) TRACE_(olerelay)("[%ld]",adesc->rgbounds[i].cElements);
|
|
arrsize *= adesc->rgbounds[i].cElements;
|
|
}
|
|
if (debugout) TRACE_(olerelay)("(vt %d)",adesc->tdescElem.vt);
|
|
if (debugout) TRACE_(olerelay)("[");
|
|
for (i=0;i<arrsize;i++) {
|
|
hres = serialize_param(tinfo, writeit, debugout, dealloc, &adesc->tdescElem, (DWORD*)((LPBYTE)arg+i*_xsize(&adesc->tdescElem)), buf);
|
|
if (hres)
|
|
return hres;
|
|
if (debugout && (i<arrsize-1)) TRACE_(olerelay)(",");
|
|
}
|
|
if (debugout) TRACE_(olerelay)("]");
|
|
return S_OK;
|
|
}
|
|
default:
|
|
ERR("Unhandled marshal type %d.\n",tdesc->vt);
|
|
return S_OK;
|
|
}
|
|
}
|
|
|
|
/* IDL desc:
|
|
* HRESULT GetIDsOfNames(
|
|
* [in] REFIID riid, args[1]
|
|
* [in, size_is(cNames)] LPOLESTR *rgszNames, args[2]
|
|
* [in] UINT cNames, args[3]
|
|
* [in] LCID lcid, args[4]
|
|
* [out, size_is(cNames)] DISPID *rgDispId); args[5]
|
|
*
|
|
* line format:
|
|
* IID iid;
|
|
* DWORD cNames;
|
|
* LPOLESTR rgszNames[cNames];
|
|
* DWORD bytestrlen (incl 0)
|
|
* BYTE data[bytestrlen] (incl 0)
|
|
* LCID
|
|
*/
|
|
static HRESULT
|
|
serialize_IDispatch_GetIDsOfNames(
|
|
BOOL inputparams,
|
|
BOOL debugout,
|
|
DWORD *args,
|
|
marshal_state *buf)
|
|
{
|
|
HRESULT hres;
|
|
DWORD cNames = args[2];
|
|
LPOLESTR *rgszNames = (LPOLESTR*)args[1];
|
|
int i;
|
|
|
|
if (inputparams) {
|
|
if (debugout) TRACE_(olerelay)("riid=%s,",debugstr_guid((REFIID)args[0]));
|
|
hres = xbuf_add(buf, (LPBYTE)args[0], sizeof(IID));
|
|
if (hres) {
|
|
FIXME("serialize of IID failed.\n");
|
|
return hres;
|
|
}
|
|
if (debugout) TRACE_(olerelay)("cNames=%ld,",cNames);
|
|
hres = xbuf_add(buf, (LPBYTE)&cNames, sizeof(DWORD));
|
|
if (hres) {
|
|
FIXME("serialize of cNames failed.\n");
|
|
return hres;
|
|
}
|
|
if (debugout) TRACE_(olerelay)("rgszNames=[");
|
|
for (i=0;i<cNames;i++) {
|
|
DWORD len = 2*(lstrlenW(rgszNames[i])+1);
|
|
|
|
if (debugout) TRACE_(olerelay)("%s,",relaystr(rgszNames[i]));
|
|
hres = xbuf_add(buf, (LPBYTE)&len, sizeof(DWORD));
|
|
if (hres) {
|
|
FIXME("serialize of len failed.\n");
|
|
return hres;
|
|
}
|
|
hres = xbuf_add(buf, (LPBYTE)rgszNames[i], len);
|
|
if (hres) {
|
|
FIXME("serialize of rgszNames[i] failed.\n");
|
|
return hres;
|
|
}
|
|
}
|
|
if (debugout) TRACE_(olerelay)("],lcid=%04lx)",args[3]);
|
|
hres = xbuf_add(buf, (LPBYTE)&args[3], sizeof(DWORD));
|
|
if (hres) {
|
|
FIXME("serialize of lcid failed.\n");
|
|
return hres;
|
|
}
|
|
} else {
|
|
DISPID *rgDispId = (DISPID*)args[4];
|
|
|
|
hres = xbuf_add(buf, (LPBYTE)rgDispId, sizeof(DISPID) * cNames);
|
|
if (hres) {
|
|
FIXME("serialize of rgDispId failed.\n");
|
|
return hres;
|
|
}
|
|
if (debugout) {
|
|
TRACE_(olerelay)("riid=[in],rgszNames=[in],cNames=[in],rgDispId=[");
|
|
for (i=0;i<cNames;i++)
|
|
TRACE_(olerelay)("%08lx,",rgDispId[i]);
|
|
TRACE_(olerelay)("])");
|
|
}
|
|
HeapFree(GetProcessHeap(),0,(IID*)args[0]);
|
|
rgszNames = (LPOLESTR*)args[1];
|
|
for (i=0;i<cNames;i++) HeapFree(GetProcessHeap(),0,rgszNames[i]);
|
|
HeapFree(GetProcessHeap(),0,rgszNames);
|
|
HeapFree(GetProcessHeap(),0,rgDispId);
|
|
}
|
|
return S_OK;
|
|
}
|
|
|
|
static HRESULT
|
|
deserialize_IDispatch_GetIDsOfNames(
|
|
BOOL inputparams,
|
|
BOOL debugout,
|
|
DWORD *args,
|
|
marshal_state *buf)
|
|
{
|
|
HRESULT hres;
|
|
DWORD cNames;
|
|
LPOLESTR *rgszNames;
|
|
int i;
|
|
|
|
if (inputparams) {
|
|
args[0] = (DWORD)HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(IID));
|
|
if (!args[0]) return E_FAIL;
|
|
hres = xbuf_get(buf, (LPBYTE)args[0], sizeof(IID));
|
|
if (hres) {
|
|
FIXME("deserialize of IID failed.\n");
|
|
return hres;
|
|
}
|
|
if (debugout) TRACE_(olerelay)("riid=%s,",debugstr_guid((REFIID)args[0]));
|
|
|
|
hres = xbuf_get(buf, (LPBYTE)&cNames, sizeof(DWORD));
|
|
if (hres) {
|
|
FIXME("deserialize of cNames failed.\n");
|
|
return hres;
|
|
}
|
|
args[2] = cNames;
|
|
if (debugout) TRACE_(olerelay)("cNames=%ld,",cNames);
|
|
if (debugout) TRACE_(olerelay)("rgszNames=[");
|
|
rgszNames = HeapAlloc(GetProcessHeap(),HEAP_ZERO_MEMORY,sizeof(LPOLESTR) * cNames);
|
|
if (!rgszNames) return E_FAIL;
|
|
args[1] = (DWORD)rgszNames;
|
|
for (i=0;i<cNames;i++) {
|
|
DWORD len;
|
|
|
|
hres = xbuf_get(buf, (LPBYTE)&len, sizeof(DWORD));
|
|
if (hres) {
|
|
FIXME("serialize of len failed.\n");
|
|
return hres;
|
|
}
|
|
rgszNames[i] = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, len);
|
|
if (!rgszNames[i]) {
|
|
FIXME("heapalloc of %ld bytes failed\n", len);
|
|
return E_FAIL;
|
|
}
|
|
hres = xbuf_get(buf, (LPBYTE)rgszNames[i], len);
|
|
if (hres) {
|
|
FIXME("serialize of rgszNames[i] failed.\n");
|
|
return hres;
|
|
}
|
|
if (debugout) TRACE_(olerelay)("%s,",relaystr(rgszNames[i]));
|
|
}
|
|
hres = xbuf_get(buf, (LPBYTE)&args[3], sizeof(DWORD));
|
|
if (hres) {
|
|
FIXME("deserialize of lcid failed.\n");
|
|
return hres;
|
|
}
|
|
if (debugout) TRACE_(olerelay)("],lcid=%04lx,rgDispId=[out])",args[3]);
|
|
args[4] = (DWORD)HeapAlloc(GetProcessHeap(),HEAP_ZERO_MEMORY,sizeof(DISPID) * cNames);
|
|
} else {
|
|
hres = xbuf_get(buf, (LPBYTE)args[4], sizeof(DISPID) * args[2]);
|
|
if (hres) {
|
|
FIXME("serialize of rgDispId failed.\n");
|
|
return hres;
|
|
}
|
|
if (debugout) {
|
|
TRACE_(olerelay)("dispid=[");
|
|
for (i=0;i<args[2];i++)
|
|
TRACE_(olerelay)("%08lx,",((DISPID*)args[4])[i]);
|
|
TRACE_(olerelay)("])");
|
|
}
|
|
}
|
|
return S_OK;
|
|
}
|
|
|
|
static HRESULT
|
|
serialize_LPVOID_ptr(
|
|
ITypeInfo *tinfo,
|
|
BOOL writeit,
|
|
BOOL debugout,
|
|
BOOL dealloc,
|
|
TYPEDESC *tdesc,
|
|
DWORD *arg,
|
|
marshal_state *buf)
|
|
{
|
|
HRESULT hres;
|
|
DWORD cookie;
|
|
|
|
if ((tdesc->vt != VT_PTR) ||
|
|
(tdesc->u.lptdesc->vt != VT_PTR) ||
|
|
(tdesc->u.lptdesc->u.lptdesc->vt != VT_VOID)
|
|
) {
|
|
FIXME("ppvObject not expressed as VT_PTR -> VT_PTR -> VT_VOID?\n");
|
|
return E_FAIL;
|
|
}
|
|
cookie = (*(DWORD*)*arg) ? 0x42424242: 0x0;
|
|
if (writeit) {
|
|
hres = xbuf_add(buf, (LPVOID)&cookie, sizeof(cookie));
|
|
if (hres)
|
|
return hres;
|
|
}
|
|
if (!*(DWORD*)*arg) {
|
|
if (debugout) TRACE_(olerelay)("<lpvoid NULL>");
|
|
return S_OK;
|
|
}
|
|
if (debugout)
|
|
TRACE_(olerelay)("ppv(%p)",*(LPUNKNOWN*)*arg);
|
|
if (writeit) {
|
|
hres = _marshal_interface(buf,&(buf->iid),*(LPUNKNOWN*)*arg);
|
|
if (hres)
|
|
return hres;
|
|
}
|
|
if (dealloc)
|
|
HeapFree(GetProcessHeap(),0,(LPVOID)*arg);
|
|
return S_OK;
|
|
}
|
|
|
|
static HRESULT
|
|
serialize_DISPPARAM_ptr(
|
|
ITypeInfo *tinfo,
|
|
BOOL writeit,
|
|
BOOL debugout,
|
|
BOOL dealloc,
|
|
TYPEDESC *tdesc,
|
|
DWORD *arg,
|
|
marshal_state *buf)
|
|
{
|
|
DWORD cookie;
|
|
HRESULT hres;
|
|
DISPPARAMS *disp;
|
|
int i;
|
|
|
|
if ((tdesc->vt != VT_PTR) || (tdesc->u.lptdesc->vt != VT_USERDEFINED)) {
|
|
FIXME("DISPPARAMS not expressed as VT_PTR -> VT_USERDEFINED?\n");
|
|
return E_FAIL;
|
|
}
|
|
|
|
cookie = *arg ? 0x42424242 : 0x0;
|
|
if (writeit) {
|
|
hres = xbuf_add(buf,(LPBYTE)&cookie,sizeof(cookie));
|
|
if (hres)
|
|
return hres;
|
|
}
|
|
if (!*arg) {
|
|
if (debugout) TRACE_(olerelay)("<DISPPARAMS NULL>");
|
|
return S_OK;
|
|
}
|
|
disp = (DISPPARAMS*)*arg;
|
|
if (writeit) {
|
|
hres = xbuf_add(buf,(LPBYTE)&disp->cArgs,sizeof(disp->cArgs));
|
|
if (hres)
|
|
return hres;
|
|
}
|
|
if (debugout) TRACE_(olerelay)("D{");
|
|
for (i=0;i<disp->cArgs;i++) {
|
|
TYPEDESC vtdesc;
|
|
|
|
vtdesc.vt = VT_VARIANT;
|
|
serialize_param(
|
|
tinfo,
|
|
writeit,
|
|
debugout,
|
|
dealloc,
|
|
&vtdesc,
|
|
(DWORD*)(disp->rgvarg+i),
|
|
buf
|
|
);
|
|
if (debugout && (i<disp->cArgs-1))
|
|
TRACE_(olerelay)(",");
|
|
}
|
|
if (dealloc)
|
|
HeapFree(GetProcessHeap(),0,disp->rgvarg);
|
|
if (writeit) {
|
|
hres = xbuf_add(buf,(LPBYTE)&disp->cNamedArgs,sizeof(disp->cNamedArgs));
|
|
if (hres)
|
|
return hres;
|
|
}
|
|
if (debugout) TRACE_(olerelay)("}{");
|
|
for (i=0;i<disp->cNamedArgs;i++) {
|
|
TYPEDESC vtdesc;
|
|
|
|
vtdesc.vt = VT_UINT;
|
|
serialize_param(
|
|
tinfo,
|
|
writeit,
|
|
debugout,
|
|
dealloc,
|
|
&vtdesc,
|
|
(DWORD*)(disp->rgdispidNamedArgs+i),
|
|
buf
|
|
);
|
|
if (debugout && (i<disp->cNamedArgs-1))
|
|
TRACE_(olerelay)(",");
|
|
}
|
|
if (debugout) TRACE_(olerelay)("}");
|
|
if (dealloc) {
|
|
HeapFree(GetProcessHeap(),0,disp->rgdispidNamedArgs);
|
|
HeapFree(GetProcessHeap(),0,disp);
|
|
}
|
|
return S_OK;
|
|
}
|
|
|
|
static HRESULT
|
|
deserialize_param(
|
|
ITypeInfo *tinfo,
|
|
BOOL readit,
|
|
BOOL debugout,
|
|
BOOL alloc,
|
|
TYPEDESC *tdesc,
|
|
DWORD *arg,
|
|
marshal_state *buf)
|
|
{
|
|
HRESULT hres = S_OK;
|
|
|
|
TRACE("vt %d at %p\n",tdesc->vt,arg);
|
|
|
|
while (1) {
|
|
switch (tdesc->vt) {
|
|
case VT_EMPTY:
|
|
if (debugout) TRACE_(olerelay)("<empty>");
|
|
return S_OK;
|
|
case VT_NULL:
|
|
if (debugout) TRACE_(olerelay)("<null>");
|
|
return S_OK;
|
|
case VT_VARIANT: {
|
|
VARIANT *vt = (VARIANT*)arg;
|
|
|
|
if (readit) {
|
|
DWORD vttype;
|
|
TYPEDESC tdesc2;
|
|
hres = xbuf_get(buf,(LPBYTE)&vttype,sizeof(vttype));
|
|
if (hres) {
|
|
FIXME("vt type not read?\n");
|
|
return hres;
|
|
}
|
|
memset(&tdesc2,0,sizeof(tdesc2));
|
|
tdesc2.vt = vttype;
|
|
V_VT(vt) = vttype;
|
|
if (debugout) TRACE_(olerelay)("Vt(%ld)(",vttype);
|
|
hres = deserialize_param(tinfo, readit, debugout, alloc, &tdesc2, &(V_I4(vt)), buf);
|
|
TRACE_(olerelay)(")");
|
|
return hres;
|
|
} else {
|
|
VariantInit(vt);
|
|
return S_OK;
|
|
}
|
|
}
|
|
case VT_ERROR:
|
|
case VT_BOOL:
|
|
case VT_I4:
|
|
case VT_UINT:
|
|
case VT_R4:
|
|
case VT_UI4:
|
|
if (readit) {
|
|
hres = xbuf_get(buf,(LPBYTE)arg,sizeof(DWORD));
|
|
if (hres) ERR("Failed to read integer 4 byte\n");
|
|
}
|
|
if (debugout) TRACE_(olerelay)("%lx",*arg);
|
|
return hres;
|
|
case VT_I2:
|
|
case VT_UI2:
|
|
if (readit) {
|
|
DWORD x;
|
|
hres = xbuf_get(buf,(LPBYTE)&x,sizeof(DWORD));
|
|
if (hres) ERR("Failed to read integer 4 byte\n");
|
|
memcpy(arg,&x,2);
|
|
}
|
|
if (debugout) TRACE_(olerelay)("%04lx",*arg & 0xffff);
|
|
return hres;
|
|
case VT_I1:
|
|
case VT_UI1:
|
|
if (readit) {
|
|
DWORD x;
|
|
hres = xbuf_get(buf,(LPBYTE)&x,sizeof(DWORD));
|
|
if (hres) ERR("Failed to read integer 4 byte\n");
|
|
memcpy(arg,&x,1);
|
|
}
|
|
if (debugout) TRACE_(olerelay)("%02lx",*arg & 0xff);
|
|
return hres;
|
|
case VT_BSTR|VT_BYREF: {
|
|
BSTR **bstr = (BSTR **)arg;
|
|
WCHAR *str;
|
|
DWORD len;
|
|
|
|
if (readit) {
|
|
hres = xbuf_get(buf,(LPBYTE)&len,sizeof(DWORD));
|
|
if (hres) {
|
|
ERR("failed to read bstr klen\n");
|
|
return hres;
|
|
}
|
|
if (len == -1) {
|
|
*bstr = NULL;
|
|
if (debugout) TRACE_(olerelay)("<bstr NULL>");
|
|
} else {
|
|
str = HeapAlloc(GetProcessHeap(),HEAP_ZERO_MEMORY,len+sizeof(WCHAR));
|
|
hres = xbuf_get(buf,(LPBYTE)str,len);
|
|
if (hres) {
|
|
ERR("Failed to read BSTR.\n");
|
|
return hres;
|
|
}
|
|
*bstr = CoTaskMemAlloc(sizeof(BSTR *));
|
|
**bstr = SysAllocStringLen(str,len);
|
|
if (debugout) TRACE_(olerelay)("%s",relaystr(str));
|
|
HeapFree(GetProcessHeap(),0,str);
|
|
}
|
|
} else {
|
|
*bstr = NULL;
|
|
}
|
|
return S_OK;
|
|
}
|
|
case VT_BSTR: {
|
|
WCHAR *str;
|
|
DWORD len;
|
|
|
|
if (readit) {
|
|
hres = xbuf_get(buf,(LPBYTE)&len,sizeof(DWORD));
|
|
if (hres) {
|
|
ERR("failed to read bstr klen\n");
|
|
return hres;
|
|
}
|
|
if (len == -1) {
|
|
*arg = 0;
|
|
if (debugout) TRACE_(olerelay)("<bstr NULL>");
|
|
} else {
|
|
str = HeapAlloc(GetProcessHeap(),HEAP_ZERO_MEMORY,len+sizeof(WCHAR));
|
|
hres = xbuf_get(buf,(LPBYTE)str,len);
|
|
if (hres) {
|
|
ERR("Failed to read BSTR.\n");
|
|
return hres;
|
|
}
|
|
*arg = (DWORD)SysAllocStringLen(str,len);
|
|
if (debugout) TRACE_(olerelay)("%s",relaystr(str));
|
|
HeapFree(GetProcessHeap(),0,str);
|
|
}
|
|
} else {
|
|
*arg = 0;
|
|
}
|
|
return S_OK;
|
|
}
|
|
case VT_PTR: {
|
|
DWORD cookie;
|
|
BOOL derefhere = 0;
|
|
|
|
derefhere = (tdesc->u.lptdesc->vt != VT_USERDEFINED);
|
|
/* read it in all cases, we need to know if we have
|
|
* NULL pointer or not.
|
|
*/
|
|
hres = xbuf_get(buf,(LPBYTE)&cookie,sizeof(cookie));
|
|
if (hres) {
|
|
ERR("Failed to load pointer cookie.\n");
|
|
return hres;
|
|
}
|
|
if (cookie != 0x42424242) {
|
|
/* we read a NULL ptr from the remote side */
|
|
if (debugout) TRACE_(olerelay)("NULL");
|
|
*arg = 0;
|
|
return S_OK;
|
|
}
|
|
if (debugout) TRACE_(olerelay)("*");
|
|
if (alloc) {
|
|
/* Allocate space for the referenced struct */
|
|
if (derefhere)
|
|
*arg=(DWORD)HeapAlloc(GetProcessHeap(),HEAP_ZERO_MEMORY,_xsize(tdesc->u.lptdesc));
|
|
}
|
|
if (derefhere)
|
|
return deserialize_param(tinfo, readit, debugout, alloc, tdesc->u.lptdesc, (LPDWORD)*arg, buf);
|
|
else
|
|
return deserialize_param(tinfo, readit, debugout, alloc, tdesc->u.lptdesc, arg, buf);
|
|
}
|
|
case VT_UNKNOWN:
|
|
/* FIXME: UNKNOWN is unknown ..., but allocate 4 byte for it */
|
|
if (alloc)
|
|
*arg=(DWORD)HeapAlloc(GetProcessHeap(),HEAP_ZERO_MEMORY,sizeof(DWORD));
|
|
hres = S_OK;
|
|
if (readit)
|
|
hres = _unmarshal_interface(buf,&IID_IUnknown,(LPUNKNOWN*)arg);
|
|
if (debugout)
|
|
TRACE_(olerelay)("unk(%p)",arg);
|
|
return hres;
|
|
case VT_DISPATCH:
|
|
hres = S_OK;
|
|
if (readit)
|
|
hres = _unmarshal_interface(buf,&IID_IDispatch,(LPUNKNOWN*)arg);
|
|
if (debugout)
|
|
TRACE_(olerelay)("idisp(%p)",arg);
|
|
return hres;
|
|
case VT_VOID:
|
|
if (debugout) TRACE_(olerelay)("<void>");
|
|
return S_OK;
|
|
case VT_USERDEFINED: {
|
|
ITypeInfo *tinfo2;
|
|
TYPEATTR *tattr;
|
|
|
|
hres = ITypeInfo_GetRefTypeInfo(tinfo,tdesc->u.hreftype,&tinfo2);
|
|
if (hres) {
|
|
ERR("Could not get typeinfo of hreftype %lx for VT_USERDEFINED.\n",tdesc->u.hreftype);
|
|
return hres;
|
|
}
|
|
hres = ITypeInfo_GetTypeAttr(tinfo2,&tattr);
|
|
if (hres) {
|
|
ERR("Could not get typeattr in VT_USERDEFINED.\n");
|
|
} else {
|
|
switch (tattr->typekind) {
|
|
case TKIND_DISPATCH:
|
|
case TKIND_INTERFACE:
|
|
if (readit)
|
|
hres = _unmarshal_interface(buf,&(tattr->guid),(LPUNKNOWN*)arg);
|
|
break;
|
|
case TKIND_RECORD: {
|
|
int i;
|
|
|
|
if (alloc)
|
|
*arg = (DWORD)HeapAlloc(GetProcessHeap(),HEAP_ZERO_MEMORY,tattr->cbSizeInstance);
|
|
|
|
if (debugout) TRACE_(olerelay)("{");
|
|
for (i=0;i<tattr->cVars;i++) {
|
|
VARDESC *vdesc;
|
|
|
|
hres = ITypeInfo2_GetVarDesc(tinfo2, i, &vdesc);
|
|
if (hres) {
|
|
ERR("Could not get vardesc of %d\n",i);
|
|
return hres;
|
|
}
|
|
hres = deserialize_param(
|
|
tinfo2,
|
|
readit,
|
|
debugout,
|
|
alloc,
|
|
&vdesc->elemdescVar.tdesc,
|
|
(DWORD*)(((LPBYTE)*arg)+vdesc->u.oInst),
|
|
buf
|
|
);
|
|
if (debugout && (i<tattr->cVars-1)) TRACE_(olerelay)(",");
|
|
}
|
|
if (buf->thisisiid && (tattr->cbSizeInstance==sizeof(GUID)))
|
|
memcpy(&(buf->iid),(LPBYTE)*arg,sizeof(buf->iid));
|
|
if (debugout) TRACE_(olerelay)("}");
|
|
break;
|
|
}
|
|
default:
|
|
ERR("Unhandled typekind %d\n",tattr->typekind);
|
|
hres = E_FAIL;
|
|
break;
|
|
}
|
|
}
|
|
if (hres)
|
|
ERR("failed to stuballoc in TKIND_RECORD.\n");
|
|
ITypeInfo_Release(tinfo2);
|
|
return hres;
|
|
}
|
|
case VT_CARRAY: {
|
|
/* arg is pointing to the start of the array. */
|
|
ARRAYDESC *adesc = tdesc->u.lpadesc;
|
|
int arrsize,i;
|
|
arrsize = 1;
|
|
if (adesc->cDims > 1) FIXME("cDims > 1 in VT_CARRAY. Does it work?\n");
|
|
for (i=0;i<adesc->cDims;i++)
|
|
arrsize *= adesc->rgbounds[i].cElements;
|
|
for (i=0;i<arrsize;i++)
|
|
deserialize_param(
|
|
tinfo,
|
|
readit,
|
|
debugout,
|
|
alloc,
|
|
&adesc->tdescElem,
|
|
(DWORD*)((LPBYTE)(arg)+i*_xsize(&adesc->tdescElem)),
|
|
buf
|
|
);
|
|
return S_OK;
|
|
}
|
|
default:
|
|
ERR("No handler for VT type %d!\n",tdesc->vt);
|
|
return S_OK;
|
|
}
|
|
}
|
|
}
|
|
|
|
static HRESULT
|
|
deserialize_LPVOID_ptr(
|
|
ITypeInfo *tinfo,
|
|
BOOL readit,
|
|
BOOL debugout,
|
|
BOOL alloc,
|
|
TYPEDESC *tdesc,
|
|
DWORD *arg,
|
|
marshal_state *buf
|
|
) {
|
|
HRESULT hres;
|
|
DWORD cookie;
|
|
|
|
if ((tdesc->vt != VT_PTR) ||
|
|
(tdesc->u.lptdesc->vt != VT_PTR) ||
|
|
(tdesc->u.lptdesc->u.lptdesc->vt != VT_VOID)
|
|
) {
|
|
FIXME("ppvObject not expressed as VT_PTR -> VT_PTR -> VT_VOID?\n");
|
|
return E_FAIL;
|
|
}
|
|
if (alloc)
|
|
*arg=(DWORD)HeapAlloc(GetProcessHeap(),HEAP_ZERO_MEMORY,sizeof(LPVOID));
|
|
if (readit) {
|
|
hres = xbuf_get(buf, (LPVOID)&cookie, sizeof(cookie));
|
|
if (hres)
|
|
return hres;
|
|
if (cookie != 0x42424242) {
|
|
*(DWORD*)*arg = 0;
|
|
if (debugout) TRACE_(olerelay)("<lpvoid NULL>");
|
|
return S_OK;
|
|
}
|
|
}
|
|
if (readit) {
|
|
hres = _unmarshal_interface(buf,&buf->iid,(LPUNKNOWN*)*arg);
|
|
if (hres) {
|
|
FIXME("_unmarshal_interface of %s , %p failed with %lx\n", debugstr_guid(&buf->iid), (LPUNKNOWN*)*arg, hres);
|
|
return hres;
|
|
}
|
|
}
|
|
if (debugout) TRACE_(olerelay)("ppv(%p)",(LPVOID)*arg);
|
|
return S_OK;
|
|
}
|
|
|
|
static HRESULT
|
|
deserialize_DISPPARAM_ptr(
|
|
ITypeInfo *tinfo,
|
|
BOOL readit,
|
|
BOOL debugout,
|
|
BOOL alloc,
|
|
TYPEDESC *tdesc,
|
|
DWORD *arg,
|
|
marshal_state *buf)
|
|
{
|
|
DWORD cookie;
|
|
DISPPARAMS *disps;
|
|
HRESULT hres;
|
|
int i;
|
|
|
|
if ((tdesc->vt != VT_PTR) || (tdesc->u.lptdesc->vt != VT_USERDEFINED)) {
|
|
FIXME("DISPPARAMS not expressed as VT_PTR -> VT_USERDEFINED?\n");
|
|
return E_FAIL;
|
|
}
|
|
if (readit) {
|
|
hres = xbuf_get(buf,(LPBYTE)&cookie,sizeof(cookie));
|
|
if (hres)
|
|
return hres;
|
|
if (cookie == 0) {
|
|
*arg = 0;
|
|
if (debugout) TRACE_(olerelay)("<DISPPARAMS NULL>");
|
|
return S_OK;
|
|
}
|
|
}
|
|
if (alloc)
|
|
*arg = (DWORD)HeapAlloc(GetProcessHeap(),HEAP_ZERO_MEMORY,sizeof(DISPPARAMS));
|
|
disps = (DISPPARAMS*)*arg;
|
|
if (!readit)
|
|
return S_OK;
|
|
hres = xbuf_get(buf, (LPBYTE)&disps->cArgs, sizeof(disps->cArgs));
|
|
if (hres)
|
|
return hres;
|
|
if (alloc)
|
|
disps->rgvarg = HeapAlloc(GetProcessHeap(),HEAP_ZERO_MEMORY,sizeof(VARIANT)*disps->cArgs);
|
|
if (debugout) TRACE_(olerelay)("D{");
|
|
for (i=0; i< disps->cArgs; i++) {
|
|
TYPEDESC vdesc;
|
|
|
|
vdesc.vt = VT_VARIANT;
|
|
hres = deserialize_param(
|
|
tinfo,
|
|
readit,
|
|
debugout,
|
|
alloc,
|
|
&vdesc,
|
|
(DWORD*)(disps->rgvarg+i),
|
|
buf
|
|
);
|
|
}
|
|
if (debugout) TRACE_(olerelay)("}{");
|
|
hres = xbuf_get(buf, (LPBYTE)&disps->cNamedArgs, sizeof(disps->cNamedArgs));
|
|
if (hres)
|
|
return hres;
|
|
if (disps->cNamedArgs) {
|
|
if (alloc)
|
|
disps->rgdispidNamedArgs = HeapAlloc(GetProcessHeap(),HEAP_ZERO_MEMORY,sizeof(DISPID)*disps->cNamedArgs);
|
|
for (i=0; i< disps->cNamedArgs; i++) {
|
|
TYPEDESC vdesc;
|
|
|
|
vdesc.vt = VT_UINT;
|
|
hres = deserialize_param(
|
|
tinfo,
|
|
readit,
|
|
debugout,
|
|
alloc,
|
|
&vdesc,
|
|
(DWORD*)(disps->rgdispidNamedArgs+i),
|
|
buf
|
|
);
|
|
if (debugout && i<(disps->cNamedArgs-1)) TRACE_(olerelay)(",");
|
|
}
|
|
}
|
|
if (debugout) TRACE_(olerelay)("}");
|
|
return S_OK;
|
|
}
|
|
|
|
/* Searches function, also in inherited interfaces */
|
|
static HRESULT
|
|
_get_funcdesc(
|
|
ITypeInfo *tinfo, int iMethod, FUNCDESC **fdesc, BSTR *iname, BSTR *fname)
|
|
{
|
|
int i = 0, j = 0;
|
|
HRESULT hres;
|
|
|
|
if (fname) *fname = NULL;
|
|
if (iname) *iname = NULL;
|
|
|
|
while (1) {
|
|
hres = ITypeInfo_GetFuncDesc(tinfo, i, fdesc);
|
|
if (hres) {
|
|
ITypeInfo *tinfo2;
|
|
HREFTYPE href;
|
|
TYPEATTR *attr;
|
|
|
|
hres = ITypeInfo_GetTypeAttr(tinfo, &attr);
|
|
if (hres) {
|
|
ERR("GetTypeAttr failed with %lx\n",hres);
|
|
return hres;
|
|
}
|
|
/* Not found, so look in inherited ifaces. */
|
|
for (j=0;j<attr->cImplTypes;j++) {
|
|
hres = ITypeInfo_GetRefTypeOfImplType(tinfo, j, &href);
|
|
if (hres) {
|
|
ERR("Did not find a reftype for interface offset %d?\n",j);
|
|
break;
|
|
}
|
|
hres = ITypeInfo_GetRefTypeInfo(tinfo, href, &tinfo2);
|
|
if (hres) {
|
|
ERR("Did not find a typeinfo for reftype %ld?\n",href);
|
|
continue;
|
|
}
|
|
hres = _get_funcdesc(tinfo2,iMethod,fdesc,iname,fname);
|
|
ITypeInfo_Release(tinfo2);
|
|
if (!hres) return S_OK;
|
|
}
|
|
return hres;
|
|
}
|
|
if (((*fdesc)->oVft/4) == iMethod) {
|
|
if (fname)
|
|
ITypeInfo_GetDocumentation(tinfo,(*fdesc)->memid,fname,NULL,NULL,NULL);
|
|
if (iname)
|
|
ITypeInfo_GetDocumentation(tinfo,-1,iname,NULL,NULL,NULL);
|
|
return S_OK;
|
|
}
|
|
i++;
|
|
}
|
|
}
|
|
|
|
static DWORD
|
|
xCall(LPVOID retptr, int method, TMProxyImpl *tpinfo /*, args */)
|
|
{
|
|
DWORD *args = ((DWORD*)&tpinfo)+1, *xargs;
|
|
FUNCDESC *fdesc;
|
|
HRESULT hres;
|
|
int i, relaydeb = TRACE_ON(olerelay);
|
|
marshal_state buf;
|
|
RPCOLEMESSAGE msg;
|
|
ULONG status;
|
|
BSTR fname,iname;
|
|
BSTR names[10];
|
|
int nrofnames;
|
|
int is_idispatch_getidsofnames = 0;
|
|
DWORD remoteresult = 0;
|
|
|
|
EnterCriticalSection(&tpinfo->crit);
|
|
|
|
hres = _get_funcdesc(tpinfo->tinfo,method,&fdesc,&iname,&fname);
|
|
if (hres) {
|
|
ERR("Did not find typeinfo/funcdesc entry for method %d!\n",method);
|
|
LeaveCriticalSection(&tpinfo->crit);
|
|
return E_FAIL;
|
|
}
|
|
|
|
if (!tpinfo->chanbuf)
|
|
{
|
|
WARN("Tried to use disconnected proxy\n");
|
|
LeaveCriticalSection(&tpinfo->crit);
|
|
return RPC_E_DISCONNECTED;
|
|
}
|
|
|
|
if (relaydeb) {
|
|
TRACE_(olerelay)("->");
|
|
if (iname)
|
|
TRACE_(olerelay)("%s:",relaystr(iname));
|
|
if (fname)
|
|
TRACE_(olerelay)("%s(%d)",relaystr(fname),method);
|
|
else
|
|
TRACE_(olerelay)("%d",method);
|
|
TRACE_(olerelay)("(");
|
|
}
|
|
if (iname && fname && !lstrcmpW(iname,IDispatchW) && !lstrcmpW(fname,GetIDsOfNamesW))
|
|
is_idispatch_getidsofnames = 1;
|
|
|
|
if (iname) SysFreeString(iname);
|
|
if (fname) SysFreeString(fname);
|
|
|
|
memset(&buf,0,sizeof(buf));
|
|
buf.iid = IID_IUnknown;
|
|
|
|
/* Special IDispatch::GetIDsOfNames() serializer */
|
|
if (is_idispatch_getidsofnames) {
|
|
hres = serialize_IDispatch_GetIDsOfNames(TRUE,relaydeb,args,&buf);
|
|
if (hres != S_OK) {
|
|
FIXME("serialize of IDispatch::GetIDsOfNames failed!\n");
|
|
return hres;
|
|
}
|
|
goto afterserialize;
|
|
}
|
|
|
|
/* special QueryInterface serialize */
|
|
if (method == 0) {
|
|
xbuf_add(&buf,(LPBYTE)args[0],sizeof(IID));
|
|
if (relaydeb) TRACE_(olerelay)("riid=%s,[out])",debugstr_guid((REFIID)args[0]));
|
|
goto afterserialize;
|
|
}
|
|
|
|
/* normal typelib driven serializing */
|
|
|
|
/* Need them for hack below */
|
|
memset(names,0,sizeof(names));
|
|
if (ITypeInfo_GetNames(tpinfo->tinfo,fdesc->memid,names,sizeof(names)/sizeof(names[0]),&nrofnames))
|
|
nrofnames = 0;
|
|
if (nrofnames > sizeof(names)/sizeof(names[0]))
|
|
ERR("Need more names!\n");
|
|
|
|
xargs = args;
|
|
for (i=0;i<fdesc->cParams;i++) {
|
|
ELEMDESC *elem = fdesc->lprgelemdescParam+i;
|
|
BOOL isserialized = FALSE;
|
|
if (relaydeb) {
|
|
if (i) TRACE_(olerelay)(",");
|
|
if (i+1<nrofnames && names[i+1])
|
|
TRACE_(olerelay)("%s=",relaystr(names[i+1]));
|
|
}
|
|
/* No need to marshal other data than FIN and any VT_PTR. */
|
|
if (!(elem->u.paramdesc.wParamFlags & PARAMFLAG_FIN) && (elem->tdesc.vt != VT_PTR)) {
|
|
xargs+=_argsize(elem->tdesc.vt);
|
|
if (relaydeb) TRACE_(olerelay)("[out]");
|
|
continue;
|
|
}
|
|
if (((i+1)<nrofnames) && !IsBadStringPtrW(names[i+1],1)) {
|
|
/* If the parameter is 'riid', we use it as interface IID
|
|
* for a later ppvObject serialization.
|
|
*/
|
|
buf.thisisiid = !lstrcmpW(names[i+1],riidW);
|
|
|
|
/* DISPPARAMS* needs special serializer */
|
|
if (!lstrcmpW(names[i+1],pdispparamsW)) {
|
|
hres = serialize_DISPPARAM_ptr(
|
|
tpinfo->tinfo,
|
|
elem->u.paramdesc.wParamFlags & PARAMFLAG_FIN,
|
|
relaydeb,
|
|
FALSE,
|
|
&elem->tdesc,
|
|
xargs,
|
|
&buf
|
|
);
|
|
isserialized = TRUE;
|
|
}
|
|
if (!lstrcmpW(names[i+1],ppvObjectW)) {
|
|
hres = serialize_LPVOID_ptr(
|
|
tpinfo->tinfo,
|
|
elem->u.paramdesc.wParamFlags & PARAMFLAG_FIN,
|
|
relaydeb,
|
|
FALSE,
|
|
&elem->tdesc,
|
|
xargs,
|
|
&buf
|
|
);
|
|
if (hres == S_OK)
|
|
isserialized = TRUE;
|
|
}
|
|
}
|
|
if (!isserialized)
|
|
hres = serialize_param(
|
|
tpinfo->tinfo,
|
|
elem->u.paramdesc.wParamFlags & PARAMFLAG_FIN,
|
|
relaydeb,
|
|
FALSE,
|
|
&elem->tdesc,
|
|
xargs,
|
|
&buf
|
|
);
|
|
|
|
if (hres) {
|
|
ERR("Failed to serialize param, hres %lx\n",hres);
|
|
break;
|
|
}
|
|
xargs+=_argsize(elem->tdesc.vt);
|
|
}
|
|
if (relaydeb) TRACE_(olerelay)(")");
|
|
|
|
afterserialize:
|
|
memset(&msg,0,sizeof(msg));
|
|
msg.cbBuffer = buf.curoff;
|
|
msg.iMethod = method;
|
|
hres = IRpcChannelBuffer_GetBuffer(tpinfo->chanbuf,&msg,&(tpinfo->iid));
|
|
if (hres) {
|
|
ERR("RpcChannelBuffer GetBuffer failed, %lx\n",hres);
|
|
LeaveCriticalSection(&tpinfo->crit);
|
|
return hres;
|
|
}
|
|
memcpy(msg.Buffer,buf.base,buf.curoff);
|
|
if (relaydeb) TRACE_(olerelay)("\n");
|
|
hres = IRpcChannelBuffer_SendReceive(tpinfo->chanbuf,&msg,&status);
|
|
if (hres) {
|
|
ERR("RpcChannelBuffer SendReceive failed, %lx\n",hres);
|
|
LeaveCriticalSection(&tpinfo->crit);
|
|
return hres;
|
|
}
|
|
|
|
if (relaydeb) TRACE_(olerelay)(" status = %08lx (",status);
|
|
if (buf.base)
|
|
buf.base = HeapReAlloc(GetProcessHeap(),0,buf.base,msg.cbBuffer);
|
|
else
|
|
buf.base = HeapAlloc(GetProcessHeap(),0,msg.cbBuffer);
|
|
buf.size = msg.cbBuffer;
|
|
memcpy(buf.base,msg.Buffer,buf.size);
|
|
buf.curoff = 0;
|
|
|
|
/* Special IDispatch::GetIDsOfNames() deserializer */
|
|
if (is_idispatch_getidsofnames) {
|
|
hres = deserialize_IDispatch_GetIDsOfNames(FALSE,relaydeb,args,&buf);
|
|
if (hres != S_OK) {
|
|
FIXME("deserialize of IDispatch::GetIDsOfNames failed!\n");
|
|
return hres;
|
|
}
|
|
goto after_deserialize;
|
|
}
|
|
/* Special QueryInterface deserializer */
|
|
if (method == 0) {
|
|
_unmarshal_interface(&buf,(REFIID)args[0],(LPUNKNOWN*)args[1]);
|
|
if (relaydeb) TRACE_(olerelay)("[in],%p",*((DWORD**)args[1]));
|
|
goto after_deserialize;
|
|
}
|
|
|
|
/* generic deserializer using typelib description */
|
|
xargs = args;
|
|
status = S_OK;
|
|
for (i=0;i<fdesc->cParams;i++) {
|
|
ELEMDESC *elem = fdesc->lprgelemdescParam+i;
|
|
BOOL isdeserialized = FALSE;
|
|
|
|
if (relaydeb) {
|
|
if (i) TRACE_(olerelay)(",");
|
|
if (i+1<nrofnames && names[i+1]) TRACE_(olerelay)("%s=",relaystr(names[i+1]));
|
|
}
|
|
/* No need to marshal other data than FOUT and any VT_PTR */
|
|
if (!(elem->u.paramdesc.wParamFlags & PARAMFLAG_FOUT) && (elem->tdesc.vt != VT_PTR)) {
|
|
xargs += _argsize(elem->tdesc.vt);
|
|
if (relaydeb) TRACE_(olerelay)("[in]");
|
|
continue;
|
|
}
|
|
if (((i+1)<nrofnames) && !IsBadStringPtrW(names[i+1],1)) {
|
|
/* If the parameter is 'riid', we use it as interface IID
|
|
* for a later ppvObject serialization.
|
|
*/
|
|
buf.thisisiid = !lstrcmpW(names[i+1],riidW);
|
|
|
|
/* deserialize DISPPARAM */
|
|
if (!lstrcmpW(names[i+1],pdispparamsW)) {
|
|
hres = deserialize_DISPPARAM_ptr(
|
|
tpinfo->tinfo,
|
|
elem->u.paramdesc.wParamFlags & PARAMFLAG_FOUT,
|
|
relaydeb,
|
|
FALSE,
|
|
&(elem->tdesc),
|
|
xargs,
|
|
&buf
|
|
);
|
|
if (hres) {
|
|
ERR("Failed to deserialize DISPPARAM*, hres %lx\n",hres);
|
|
break;
|
|
}
|
|
isdeserialized = TRUE;
|
|
}
|
|
if (!lstrcmpW(names[i+1],ppvObjectW)) {
|
|
hres = deserialize_LPVOID_ptr(
|
|
tpinfo->tinfo,
|
|
elem->u.paramdesc.wParamFlags & PARAMFLAG_FOUT,
|
|
relaydeb,
|
|
FALSE,
|
|
&elem->tdesc,
|
|
xargs,
|
|
&buf
|
|
);
|
|
if (hres == S_OK)
|
|
isdeserialized = TRUE;
|
|
}
|
|
}
|
|
if (!isdeserialized)
|
|
hres = deserialize_param(
|
|
tpinfo->tinfo,
|
|
elem->u.paramdesc.wParamFlags & PARAMFLAG_FOUT,
|
|
relaydeb,
|
|
FALSE,
|
|
&(elem->tdesc),
|
|
xargs,
|
|
&buf
|
|
);
|
|
if (hres) {
|
|
ERR("Failed to unmarshall param, hres %lx\n",hres);
|
|
status = hres;
|
|
break;
|
|
}
|
|
xargs += _argsize(elem->tdesc.vt);
|
|
}
|
|
after_deserialize:
|
|
hres = xbuf_get(&buf, (LPBYTE)&remoteresult, sizeof(DWORD));
|
|
if (hres != S_OK)
|
|
return hres;
|
|
if (relaydeb) TRACE_(olerelay)(") = %08lx\n", remoteresult);
|
|
|
|
if (status != S_OK) /* OLE/COM internal error */
|
|
return status;
|
|
|
|
HeapFree(GetProcessHeap(),0,buf.base);
|
|
LeaveCriticalSection(&tpinfo->crit);
|
|
return remoteresult;
|
|
}
|
|
|
|
HRESULT WINAPI ProxyIUnknown_QueryInterface(IUnknown *iface, REFIID riid, void **ppv)
|
|
{
|
|
TMProxyImpl *proxy = (TMProxyImpl *)iface;
|
|
|
|
TRACE("(%s, %p)\n", debugstr_guid(riid), ppv);
|
|
|
|
if (proxy->outerunknown)
|
|
return IUnknown_QueryInterface(proxy->outerunknown, riid, ppv);
|
|
|
|
FIXME("No interface\n");
|
|
return E_NOINTERFACE;
|
|
}
|
|
|
|
ULONG WINAPI ProxyIUnknown_AddRef(IUnknown *iface)
|
|
{
|
|
TMProxyImpl *proxy = (TMProxyImpl *)iface;
|
|
|
|
TRACE("\n");
|
|
|
|
if (proxy->outerunknown)
|
|
return IUnknown_AddRef(proxy->outerunknown);
|
|
|
|
return 2; /* FIXME */
|
|
}
|
|
|
|
ULONG WINAPI ProxyIUnknown_Release(IUnknown *iface)
|
|
{
|
|
TMProxyImpl *proxy = (TMProxyImpl *)iface;
|
|
|
|
TRACE("\n");
|
|
|
|
if (proxy->outerunknown)
|
|
return IUnknown_Release(proxy->outerunknown);
|
|
|
|
return 1; /* FIXME */
|
|
}
|
|
|
|
static HRESULT WINAPI
|
|
PSFacBuf_CreateProxy(
|
|
LPPSFACTORYBUFFER iface, IUnknown* pUnkOuter, REFIID riid,
|
|
IRpcProxyBuffer **ppProxy, LPVOID *ppv)
|
|
{
|
|
HRESULT hres;
|
|
ITypeInfo *tinfo;
|
|
int i, nroffuncs;
|
|
FUNCDESC *fdesc;
|
|
TMProxyImpl *proxy;
|
|
|
|
TRACE("(...%s...)\n",debugstr_guid(riid));
|
|
hres = _get_typeinfo_for_iid(riid,&tinfo);
|
|
if (hres) {
|
|
ERR("No typeinfo for %s?\n",debugstr_guid(riid));
|
|
return hres;
|
|
}
|
|
nroffuncs = _nroffuncs(tinfo);
|
|
proxy = CoTaskMemAlloc(sizeof(TMProxyImpl));
|
|
if (!proxy) return E_OUTOFMEMORY;
|
|
|
|
assert(sizeof(TMAsmProxy) == 12);
|
|
|
|
proxy->outerunknown = pUnkOuter;
|
|
proxy->asmstubs = VirtualAlloc(NULL, sizeof(TMAsmProxy) * nroffuncs, MEM_COMMIT, PAGE_EXECUTE_READWRITE);
|
|
if (!proxy->asmstubs) {
|
|
ERR("Could not commit pages for proxy thunks\n");
|
|
CoTaskMemFree(proxy);
|
|
return E_OUTOFMEMORY;
|
|
}
|
|
|
|
InitializeCriticalSection(&proxy->crit);
|
|
|
|
proxy->lpvtbl = HeapAlloc(GetProcessHeap(),0,sizeof(LPBYTE)*nroffuncs);
|
|
for (i=0;i<nroffuncs;i++) {
|
|
TMAsmProxy *xasm = proxy->asmstubs+i;
|
|
|
|
switch (i) {
|
|
case 0:
|
|
proxy->lpvtbl[i] = ProxyIUnknown_QueryInterface;
|
|
break;
|
|
case 1:
|
|
proxy->lpvtbl[i] = ProxyIUnknown_AddRef;
|
|
break;
|
|
case 2:
|
|
proxy->lpvtbl[i] = ProxyIUnknown_Release;
|
|
break;
|
|
default: {
|
|
int j;
|
|
/* nrofargs without This */
|
|
int nrofargs;
|
|
hres = _get_funcdesc(tinfo,i,&fdesc,NULL,NULL);
|
|
if (hres) {
|
|
ERR("GetFuncDesc %lx should not fail here.\n",hres);
|
|
return hres;
|
|
}
|
|
/* some args take more than 4 byte on the stack */
|
|
nrofargs = 0;
|
|
for (j=0;j<fdesc->cParams;j++)
|
|
nrofargs += _argsize(fdesc->lprgelemdescParam[j].tdesc.vt);
|
|
|
|
if (fdesc->callconv != CC_STDCALL) {
|
|
ERR("calling convention is not stdcall????\n");
|
|
return E_FAIL;
|
|
}
|
|
/* popl %eax - return ptr
|
|
* pushl <nr>
|
|
* pushl %eax
|
|
* call xCall
|
|
* lret <nr> (+4)
|
|
*
|
|
*
|
|
* arg3 arg2 arg1 <method> <returnptr>
|
|
*/
|
|
xasm->popleax = 0x58;
|
|
xasm->pushlval = 0x6a;
|
|
xasm->nr = i;
|
|
xasm->pushleax = 0x50;
|
|
xasm->lcall = 0xe8; /* relative jump */
|
|
xasm->xcall = (DWORD)xCall;
|
|
xasm->xcall -= (DWORD)&(xasm->lret);
|
|
xasm->lret = 0xc2;
|
|
xasm->bytestopop= (nrofargs+2)*4; /* pop args, This, iMethod */
|
|
proxy->lpvtbl[i] = xasm;
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
proxy->lpvtbl2 = &tmproxyvtable;
|
|
/* 1 reference for the proxy and 1 for the object */
|
|
proxy->ref = 2;
|
|
proxy->tinfo = tinfo;
|
|
memcpy(&proxy->iid,riid,sizeof(*riid));
|
|
proxy->chanbuf = 0;
|
|
*ppv = (LPVOID)proxy;
|
|
*ppProxy = (IRpcProxyBuffer *)&(proxy->lpvtbl2);
|
|
return S_OK;
|
|
}
|
|
|
|
typedef struct _TMStubImpl {
|
|
IRpcStubBufferVtbl *lpvtbl;
|
|
ULONG ref;
|
|
|
|
LPUNKNOWN pUnk;
|
|
ITypeInfo *tinfo;
|
|
IID iid;
|
|
} TMStubImpl;
|
|
|
|
static HRESULT WINAPI
|
|
TMStubImpl_QueryInterface(LPRPCSTUBBUFFER iface, REFIID riid, LPVOID *ppv)
|
|
{
|
|
if (IsEqualIID(riid,&IID_IRpcStubBuffer)||IsEqualIID(riid,&IID_IUnknown)){
|
|
*ppv = (LPVOID)iface;
|
|
IRpcStubBuffer_AddRef(iface);
|
|
return S_OK;
|
|
}
|
|
FIXME("%s, not supported IID.\n",debugstr_guid(riid));
|
|
return E_NOINTERFACE;
|
|
}
|
|
|
|
static ULONG WINAPI
|
|
TMStubImpl_AddRef(LPRPCSTUBBUFFER iface)
|
|
{
|
|
TMStubImpl *This = (TMStubImpl *)iface;
|
|
ULONG refCount = InterlockedIncrement(&This->ref);
|
|
|
|
TRACE("(%p)->(ref before=%lu)\n", This, refCount - 1);
|
|
|
|
return refCount;
|
|
}
|
|
|
|
static ULONG WINAPI
|
|
TMStubImpl_Release(LPRPCSTUBBUFFER iface)
|
|
{
|
|
TMStubImpl *This = (TMStubImpl *)iface;
|
|
ULONG refCount = InterlockedDecrement(&This->ref);
|
|
|
|
TRACE("(%p)->(ref before=%lu)\n", This, refCount + 1);
|
|
|
|
if (!refCount)
|
|
{
|
|
IRpcStubBuffer_Disconnect(iface);
|
|
CoTaskMemFree(This);
|
|
}
|
|
return refCount;
|
|
}
|
|
|
|
static HRESULT WINAPI
|
|
TMStubImpl_Connect(LPRPCSTUBBUFFER iface, LPUNKNOWN pUnkServer)
|
|
{
|
|
TMStubImpl *This = (TMStubImpl *)iface;
|
|
|
|
TRACE("(%p)->(%p)\n", This, pUnkServer);
|
|
|
|
IUnknown_AddRef(pUnkServer);
|
|
This->pUnk = pUnkServer;
|
|
return S_OK;
|
|
}
|
|
|
|
static void WINAPI
|
|
TMStubImpl_Disconnect(LPRPCSTUBBUFFER iface)
|
|
{
|
|
TMStubImpl *This = (TMStubImpl *)iface;
|
|
|
|
TRACE("(%p)->()\n", This);
|
|
|
|
IUnknown_Release(This->pUnk);
|
|
This->pUnk = NULL;
|
|
return;
|
|
}
|
|
|
|
static HRESULT WINAPI
|
|
TMStubImpl_Invoke(
|
|
LPRPCSTUBBUFFER iface, RPCOLEMESSAGE* xmsg,IRpcChannelBuffer*rpcchanbuf)
|
|
{
|
|
int i;
|
|
FUNCDESC *fdesc;
|
|
TMStubImpl *This = (TMStubImpl *)iface;
|
|
HRESULT hres;
|
|
DWORD *args, res, *xargs, nrofargs;
|
|
marshal_state buf;
|
|
int nrofnames;
|
|
BSTR names[10];
|
|
BSTR fname = NULL,iname = NULL;
|
|
BOOL is_idispatch_getidsofnames = 0;
|
|
|
|
memset(&buf,0,sizeof(buf));
|
|
buf.size = xmsg->cbBuffer;
|
|
buf.base = xmsg->Buffer;
|
|
buf.curoff = 0;
|
|
buf.iid = IID_IUnknown;
|
|
|
|
TRACE("...\n");
|
|
if (xmsg->iMethod == 0) { /* QI */
|
|
IID xiid;
|
|
/* in: IID, out: <iface> */
|
|
|
|
xbuf_get(&buf,(LPBYTE)&xiid,sizeof(xiid));
|
|
buf.curoff = 0;
|
|
hres = _marshal_interface(&buf,&xiid,This->pUnk);
|
|
xmsg->Buffer = buf.base; /* Might have been reallocated */
|
|
xmsg->cbBuffer = buf.size;
|
|
return hres;
|
|
}
|
|
hres = _get_funcdesc(This->tinfo,xmsg->iMethod,&fdesc,&iname,&fname);
|
|
if (hres) {
|
|
ERR("GetFuncDesc on method %ld failed with %lx\n",xmsg->iMethod,hres);
|
|
return hres;
|
|
}
|
|
|
|
if (iname && fname && !lstrcmpW(iname, IDispatchW) && !lstrcmpW(fname, GetIDsOfNamesW))
|
|
is_idispatch_getidsofnames = 1;
|
|
|
|
if (iname) SysFreeString (iname);
|
|
if (fname) SysFreeString (fname);
|
|
|
|
/* Need them for hack below */
|
|
memset(names,0,sizeof(names));
|
|
ITypeInfo_GetNames(This->tinfo,fdesc->memid,names,sizeof(names)/sizeof(names[0]),&nrofnames);
|
|
if (nrofnames > sizeof(names)/sizeof(names[0])) {
|
|
ERR("Need more names!\n");
|
|
}
|
|
|
|
/*dump_FUNCDESC(fdesc);*/
|
|
nrofargs = 0;
|
|
for (i=0;i<fdesc->cParams;i++)
|
|
nrofargs += _argsize(fdesc->lprgelemdescParam[i].tdesc.vt);
|
|
args = HeapAlloc(GetProcessHeap(),0,(nrofargs+1)*sizeof(DWORD));
|
|
if (!args) return E_OUTOFMEMORY;
|
|
|
|
if (is_idispatch_getidsofnames) {
|
|
hres = deserialize_IDispatch_GetIDsOfNames(TRUE,FALSE,args+1,&buf);
|
|
if (hres != S_OK) {
|
|
FIXME("deserialize_IDispatch_GetIDsOfNames failed!\n");
|
|
return hres;
|
|
}
|
|
xargs = args+1+5;
|
|
goto afterdeserialize;
|
|
}
|
|
|
|
/* Allocate all stuff used by call. */
|
|
xargs = args+1;
|
|
for (i=0;i<fdesc->cParams;i++) {
|
|
ELEMDESC *elem = fdesc->lprgelemdescParam+i;
|
|
BOOL isdeserialized = FALSE;
|
|
|
|
if (((i+1)<nrofnames) && !IsBadStringPtrW(names[i+1],1)) {
|
|
/* If the parameter is 'riid', we use it as interface IID
|
|
* for a later ppvObject serialization.
|
|
*/
|
|
buf.thisisiid = !lstrcmpW(names[i+1],riidW);
|
|
|
|
/* deserialize DISPPARAM */
|
|
if (!lstrcmpW(names[i+1],pdispparamsW)) {
|
|
hres = deserialize_DISPPARAM_ptr(
|
|
This->tinfo,
|
|
elem->u.paramdesc.wParamFlags & PARAMFLAG_FIN,
|
|
FALSE,
|
|
TRUE,
|
|
&(elem->tdesc),
|
|
xargs,
|
|
&buf
|
|
);
|
|
if (hres) {
|
|
ERR("Failed to deserialize DISPPARAM*, hres %lx\n",hres);
|
|
break;
|
|
}
|
|
isdeserialized = TRUE;
|
|
}
|
|
if (!lstrcmpW(names[i+1],ppvObjectW)) {
|
|
hres = deserialize_LPVOID_ptr(
|
|
This->tinfo,
|
|
elem->u.paramdesc.wParamFlags & PARAMFLAG_FIN,
|
|
FALSE,
|
|
TRUE,
|
|
&elem->tdesc,
|
|
xargs,
|
|
&buf
|
|
);
|
|
if (hres == S_OK)
|
|
isdeserialized = TRUE;
|
|
}
|
|
}
|
|
if (!isdeserialized)
|
|
hres = deserialize_param(
|
|
This->tinfo,
|
|
elem->u.paramdesc.wParamFlags & PARAMFLAG_FIN,
|
|
FALSE,
|
|
TRUE,
|
|
&(elem->tdesc),
|
|
xargs,
|
|
&buf
|
|
);
|
|
xargs += _argsize(elem->tdesc.vt);
|
|
if (hres) {
|
|
ERR("Failed to deserialize param %s, hres %lx\n",relaystr(names[i+1]),hres);
|
|
break;
|
|
}
|
|
}
|
|
afterdeserialize:
|
|
hres = IUnknown_QueryInterface(This->pUnk,&(This->iid),(LPVOID*)&(args[0]));
|
|
if (hres) {
|
|
ERR("Does not support iface %s, returning %lx\n",debugstr_guid(&(This->iid)), hres);
|
|
return hres;
|
|
}
|
|
res = _invoke(
|
|
(*((FARPROC**)args[0]))[fdesc->oVft/4],
|
|
fdesc->callconv,
|
|
(xargs-args),
|
|
args
|
|
);
|
|
IUnknown_Release((LPUNKNOWN)args[0]);
|
|
buf.curoff = 0;
|
|
|
|
/* special IDispatch::GetIDsOfNames serializer */
|
|
if (is_idispatch_getidsofnames) {
|
|
hres = serialize_IDispatch_GetIDsOfNames(FALSE,FALSE,args+1,&buf);
|
|
if (hres != S_OK) {
|
|
FIXME("serialize of IDispatch::GetIDsOfNames failed!\n");
|
|
return hres;
|
|
}
|
|
goto afterserialize;
|
|
}
|
|
xargs = args+1;
|
|
for (i=0;i<fdesc->cParams;i++) {
|
|
ELEMDESC *elem = fdesc->lprgelemdescParam+i;
|
|
BOOL isserialized = FALSE;
|
|
|
|
if (((i+1)<nrofnames) && !IsBadStringPtrW(names[i+1],1)) {
|
|
/* If the parameter is 'riid', we use it as interface IID
|
|
* for a later ppvObject serialization.
|
|
*/
|
|
buf.thisisiid = !lstrcmpW(names[i+1],riidW);
|
|
|
|
/* DISPPARAMS* needs special serializer */
|
|
if (!lstrcmpW(names[i+1],pdispparamsW)) {
|
|
hres = serialize_DISPPARAM_ptr(
|
|
This->tinfo,
|
|
elem->u.paramdesc.wParamFlags & PARAMFLAG_FOUT,
|
|
FALSE,
|
|
TRUE,
|
|
&elem->tdesc,
|
|
xargs,
|
|
&buf
|
|
);
|
|
isserialized = TRUE;
|
|
}
|
|
if (!lstrcmpW(names[i+1],ppvObjectW)) {
|
|
hres = serialize_LPVOID_ptr(
|
|
This->tinfo,
|
|
elem->u.paramdesc.wParamFlags & PARAMFLAG_FOUT,
|
|
FALSE,
|
|
TRUE,
|
|
&elem->tdesc,
|
|
xargs,
|
|
&buf
|
|
);
|
|
if (hres == S_OK)
|
|
isserialized = TRUE;
|
|
}
|
|
}
|
|
if (!isserialized)
|
|
hres = serialize_param(
|
|
This->tinfo,
|
|
elem->u.paramdesc.wParamFlags & PARAMFLAG_FOUT,
|
|
FALSE,
|
|
TRUE,
|
|
&elem->tdesc,
|
|
xargs,
|
|
&buf
|
|
);
|
|
xargs += _argsize(elem->tdesc.vt);
|
|
if (hres) {
|
|
ERR("Failed to stuballoc param, hres %lx\n",hres);
|
|
break;
|
|
}
|
|
}
|
|
afterserialize:
|
|
hres = xbuf_add (&buf, (LPBYTE)&res, sizeof(DWORD));
|
|
if (hres != S_OK)
|
|
return hres;
|
|
|
|
/* might need to use IRpcChannelBuffer_GetBuffer ? */
|
|
xmsg->cbBuffer = buf.curoff;
|
|
xmsg->Buffer = buf.base;
|
|
HeapFree(GetProcessHeap(),0,args);
|
|
return S_OK;
|
|
}
|
|
|
|
static LPRPCSTUBBUFFER WINAPI
|
|
TMStubImpl_IsIIDSupported(LPRPCSTUBBUFFER iface, REFIID riid) {
|
|
FIXME("Huh (%s)?\n",debugstr_guid(riid));
|
|
return NULL;
|
|
}
|
|
|
|
static ULONG WINAPI
|
|
TMStubImpl_CountRefs(LPRPCSTUBBUFFER iface) {
|
|
TMStubImpl *This = (TMStubImpl *)iface;
|
|
|
|
return This->ref; /*FIXME? */
|
|
}
|
|
|
|
static HRESULT WINAPI
|
|
TMStubImpl_DebugServerQueryInterface(LPRPCSTUBBUFFER iface, LPVOID *ppv) {
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
static void WINAPI
|
|
TMStubImpl_DebugServerRelease(LPRPCSTUBBUFFER iface, LPVOID ppv) {
|
|
return;
|
|
}
|
|
|
|
IRpcStubBufferVtbl tmstubvtbl = {
|
|
TMStubImpl_QueryInterface,
|
|
TMStubImpl_AddRef,
|
|
TMStubImpl_Release,
|
|
TMStubImpl_Connect,
|
|
TMStubImpl_Disconnect,
|
|
TMStubImpl_Invoke,
|
|
TMStubImpl_IsIIDSupported,
|
|
TMStubImpl_CountRefs,
|
|
TMStubImpl_DebugServerQueryInterface,
|
|
TMStubImpl_DebugServerRelease
|
|
};
|
|
|
|
static HRESULT WINAPI
|
|
PSFacBuf_CreateStub(
|
|
LPPSFACTORYBUFFER iface, REFIID riid,IUnknown *pUnkServer,
|
|
IRpcStubBuffer** ppStub
|
|
) {
|
|
HRESULT hres;
|
|
ITypeInfo *tinfo;
|
|
TMStubImpl *stub;
|
|
|
|
TRACE("(%s,%p,%p)\n",debugstr_guid(riid),pUnkServer,ppStub);
|
|
hres = _get_typeinfo_for_iid(riid,&tinfo);
|
|
if (hres) {
|
|
ERR("No typeinfo for %s?\n",debugstr_guid(riid));
|
|
return hres;
|
|
}
|
|
stub = CoTaskMemAlloc(sizeof(TMStubImpl));
|
|
if (!stub)
|
|
return E_OUTOFMEMORY;
|
|
stub->lpvtbl = &tmstubvtbl;
|
|
stub->ref = 1;
|
|
stub->tinfo = tinfo;
|
|
memcpy(&(stub->iid),riid,sizeof(*riid));
|
|
hres = IRpcStubBuffer_Connect((LPRPCSTUBBUFFER)stub,pUnkServer);
|
|
*ppStub = (LPRPCSTUBBUFFER)stub;
|
|
TRACE("IRpcStubBuffer: %p\n", stub);
|
|
if (hres)
|
|
ERR("Connect to pUnkServer failed?\n");
|
|
return hres;
|
|
}
|
|
|
|
static IPSFactoryBufferVtbl psfacbufvtbl = {
|
|
PSFacBuf_QueryInterface,
|
|
PSFacBuf_AddRef,
|
|
PSFacBuf_Release,
|
|
PSFacBuf_CreateProxy,
|
|
PSFacBuf_CreateStub
|
|
};
|
|
|
|
/* This is the whole PSFactoryBuffer object, just the vtableptr */
|
|
static IPSFactoryBufferVtbl *lppsfac = &psfacbufvtbl;
|
|
|
|
/***********************************************************************
|
|
* DllGetClassObject [OLE32.63]
|
|
*/
|
|
HRESULT WINAPI
|
|
TypeLibFac_DllGetClassObject(REFCLSID rclsid, REFIID iid,LPVOID *ppv)
|
|
{
|
|
if (IsEqualIID(iid,&IID_IPSFactoryBuffer)) {
|
|
*ppv = &lppsfac;
|
|
return S_OK;
|
|
}
|
|
return E_NOINTERFACE;
|
|
}
|