4 * Copyright 2002 Marcus Meissner
6 * This library is free software; you can redistribute it and/or
7 * modify it under the terms of the GNU Lesser General Public
8 * License as published by the Free Software Foundation; either
9 * version 2.1 of the License, or (at your option) any later version.
11 * This library is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
14 * Lesser General Public License for more details.
16 * You should have received a copy of the GNU Lesser General Public
17 * License along with this library; if not, write to the Free Software
18 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
35 #include "wine/unicode.h"
36 #include "wine/obj_base.h"
37 #include "wine/obj_channel.h"
38 #include "wine/obj_storage.h"
42 #include "wine/debug.h"
45 static const WCHAR riidW[5] = {'r','i','i','d',0};
47 WINE_DEFAULT_DEBUG_CHANNEL(ole);
48 WINE_DECLARE_DEBUG_CHANNEL(olerelay);
50 typedef struct _marshal_state {
56 IID iid; /* HACK: for VT_VOID */
60 xbuf_add(marshal_state *buf, LPBYTE stuff, DWORD size) {
61 while (buf->size - buf->curoff < size) {
64 buf->base = HeapReAlloc(GetProcessHeap(),HEAP_ZERO_MEMORY,buf->base,buf->size);
68 buf->base = HeapAlloc(GetProcessHeap(),HEAP_ZERO_MEMORY,32);
74 memcpy(buf->base+buf->curoff,stuff,size);
80 xbuf_get(marshal_state *buf, LPBYTE stuff, DWORD size) {
81 if (buf->size < buf->curoff+size) return E_FAIL;
82 memcpy(stuff,buf->base+buf->curoff,size);
88 xbuf_skip(marshal_state *buf, DWORD size) {
89 if (buf->size < buf->curoff+size) return E_FAIL;
95 _unmarshal_interface(marshal_state *buf, REFIID riid, LPUNKNOWN *pUnk) {
97 ULARGE_INTEGER newpos;
103 TRACE("...%s...\n",debugstr_guid(riid));
105 hres = xbuf_get(buf,(LPBYTE)&xsize,sizeof(xsize));
106 if (hres) return hres;
107 if (xsize == 0) return S_OK;
108 hres = CreateStreamOnHGlobal(0,TRUE,&pStm);
110 FIXME("Stream create failed %lx\n",hres);
113 hres = IStream_Write(pStm,buf->base+buf->curoff,xsize,&res);
114 if (hres) { FIXME("stream write %lx\n",hres); return hres; }
115 memset(&seekto,0,sizeof(seekto));
116 hres = IStream_Seek(pStm,seekto,SEEK_SET,&newpos);
117 if (hres) { FIXME("Failed Seek %lx\n",hres); return hres;}
118 hres = CoUnmarshalInterface(pStm,riid,(LPVOID*)pUnk);
120 FIXME("Marshaling interface %s failed with %lx\n",debugstr_guid(riid),hres);
123 IStream_Release(pStm);
124 return xbuf_skip(buf,xsize);
128 _marshal_interface(marshal_state *buf, REFIID riid, LPUNKNOWN pUnk) {
133 ULARGE_INTEGER newpos;
134 LARGE_INTEGER seekto;
143 TRACE("...%s...\n",debugstr_guid(riid));
144 hres=IUnknown_QueryInterface(pUnk,riid,(LPVOID*)&newiface);
146 TRACE("%p does not support iface %s\n",pUnk,debugstr_guid(riid));
149 hres = CreateStreamOnHGlobal(0,TRUE,&pStm);
151 FIXME("Stream create failed %lx\n",hres);
154 hres = CoMarshalInterface(pStm,riid,newiface,0,NULL,0);
155 IUnknown_Release(newiface);
157 FIXME("Marshaling interface %s failed with %lx\n",
158 debugstr_guid(riid),hres
162 hres = IStream_Stat(pStm,&ststg,0);
163 tempbuf = HeapAlloc(GetProcessHeap(), 0, ststg.cbSize.s.LowPart);
164 memset(&seekto,0,sizeof(seekto));
165 hres = IStream_Seek(pStm,seekto,SEEK_SET,&newpos);
166 if (hres) { FIXME("Failed Seek %lx\n",hres); goto fail;}
167 hres = IStream_Read(pStm,tempbuf,ststg.cbSize.s.LowPart,&res);
168 if (hres) { FIXME("Failed Read %lx\n",hres); goto fail;}
169 IStream_Release(pStm);
170 xsize = ststg.cbSize.s.LowPart;
171 xbuf_add(buf,(LPBYTE)&xsize,sizeof(xsize));
172 hres = xbuf_add(buf,tempbuf,ststg.cbSize.s.LowPart);
173 HeapFree(GetProcessHeap(),0,tempbuf);
177 xbuf_add(buf,(LPBYTE)&xsize,sizeof(xsize));
181 /********************* OLE Proxy/Stub Factory ********************************/
182 static HRESULT WINAPI
183 PSFacBuf_QueryInterface(LPPSFACTORYBUFFER iface, REFIID iid, LPVOID *ppv) {
184 if (IsEqualIID(iid,&IID_IPSFactoryBuffer)||IsEqualIID(iid,&IID_IUnknown)) {
185 *ppv = (LPVOID)iface;
186 /* No ref counting, static class */
189 FIXME("(%s) unknown IID?\n",debugstr_guid(iid));
190 return E_NOINTERFACE;
193 static ULONG WINAPI PSFacBuf_AddRef(LPPSFACTORYBUFFER iface) { return 2; }
194 static ULONG WINAPI PSFacBuf_Release(LPPSFACTORYBUFFER iface) { return 1; }
197 _get_typeinfo_for_iid(REFIID riid, ITypeInfo**ti) {
200 char tlguid[200],typelibkey[300],interfacekey[300],ver[100];
203 DWORD tlguidlen, verlen, type, tlfnlen;
206 sprintf( interfacekey, "Interface\\{%08lx-%04x-%04x-%02x%02x-%02x%02x%02x%02x%02x%02x}\\Typelib",
207 riid->Data1, riid->Data2, riid->Data3,
208 riid->Data4[0], riid->Data4[1], riid->Data4[2], riid->Data4[3],
209 riid->Data4[4], riid->Data4[5], riid->Data4[6], riid->Data4[7]
212 if (RegOpenKeyA(HKEY_CLASSES_ROOT,interfacekey,&ikey)) {
213 FIXME("No %s key found.\n",interfacekey);
217 tlguidlen = sizeof(tlguid);
218 if (RegQueryValueExA(ikey,NULL,NULL,&type,tlguid,&tlguidlen)) {
219 FIXME("Getting typelib guid failed.\n");
224 verlen = sizeof(ver);
225 if (RegQueryValueExA(ikey,"Version",NULL,&type,ver,&verlen)) {
226 FIXME("Could not get version value?\n");
231 sprintf(typelibkey,"Typelib\\%s\\%s\\0\\win32",tlguid,ver);
232 tlfnlen = sizeof(tlfn);
233 if (RegQueryValueA(HKEY_CLASSES_ROOT,typelibkey,tlfn,&tlfnlen)) {
234 FIXME("Could not get typelib fn?\n");
237 MultiByteToWideChar(CP_ACP, 0, tlfn, -1, tlfnW, -1);
238 hres = LoadTypeLib(tlfnW,&tl);
240 ERR("Failed to load typelib for %s, but it should be there.\n",debugstr_guid(riid));
243 hres = ITypeLib_GetTypeInfoOfGuid(tl,riid,ti);
245 ERR("typelib does not contain info for %s?\n",debugstr_guid(riid));
246 ITypeLib_Release(tl);
249 /* FIXME: do this? ITypeLib_Release(tl); */
253 /* Determine nr of functions. Since we use the toplevel interface and all
254 * inherited ones have lower numbers, we are ok to not to descent into
255 * the inheritance tree I think.
257 static int _nroffuncs(ITypeInfo *tinfo) {
264 hres = ITypeInfo_GetFuncDesc(tinfo,n,&fdesc);
265 if (fdesc->oVft/4 > max)
274 typedef struct _TMAsmProxy {
283 } WINE_PACKED TMAsmProxy;
285 typedef struct _TMProxyImpl {
287 ICOM_VTABLE(IRpcProxyBuffer) *lpvtbl2;
290 TMAsmProxy *asmstubs;
292 IRpcChannelBuffer* chanbuf;
296 static HRESULT WINAPI
297 TMProxyImpl_QueryInterface(LPRPCPROXYBUFFER iface, REFIID riid, LPVOID *ppv) {
299 if (IsEqualIID(riid,&IID_IUnknown)||IsEqualIID(riid,&IID_IRpcProxyBuffer)) {
300 *ppv = (LPVOID)iface;
301 IRpcProxyBuffer_AddRef(iface);
304 FIXME("no interface for %s\n",debugstr_guid(riid));
305 return E_NOINTERFACE;
309 TMProxyImpl_AddRef(LPRPCPROXYBUFFER iface) {
310 ICOM_THIS_MULTI(TMProxyImpl,lpvtbl2,iface);
318 TMProxyImpl_Release(LPRPCPROXYBUFFER iface) {
319 ICOM_THIS_MULTI(TMProxyImpl,lpvtbl2,iface);
323 if (This->ref) return This->ref;
324 if (This->chanbuf) IRpcChannelBuffer_Release(This->chanbuf);
325 HeapFree(GetProcessHeap(),0,This);
329 static HRESULT WINAPI
331 LPRPCPROXYBUFFER iface,IRpcChannelBuffer* pRpcChannelBuffer
333 ICOM_THIS_MULTI(TMProxyImpl,lpvtbl2,iface);
335 TRACE("(%p)\n",pRpcChannelBuffer);
336 This->chanbuf = pRpcChannelBuffer;
337 IRpcChannelBuffer_AddRef(This->chanbuf);
342 TMProxyImpl_Disconnect(LPRPCPROXYBUFFER iface) {
343 ICOM_THIS_MULTI(TMProxyImpl,lpvtbl2,iface);
346 IRpcChannelBuffer_Release(This->chanbuf);
347 This->chanbuf = NULL;
351 static ICOM_VTABLE(IRpcProxyBuffer) tmproxyvtable = {
352 ICOM_MSVTABLE_COMPAT_DummyRTTIVALUE
353 TMProxyImpl_QueryInterface,
357 TMProxyImpl_Disconnect
362 ITypeInfo *tinfo, ELEMDESC *elem, TYPEDESC *tdesc, DWORD *arg, marshal_state *buf
364 int relaydeb = TRACE_ON(olerelay);
367 if (!tdesc) tdesc = &(elem->tdesc);
371 case VT_BSTR: { /* DWORD size, string data */
374 DWORD *bstr = ((DWORD*)(*arg))-1;
376 if (relaydeb) MESSAGE("%s",debugstr_w((LPWSTR)(bstr+1)));
377 return xbuf_add(buf,(LPBYTE)bstr,bstr[0]+4);
381 return xbuf_add(buf,(LPBYTE)&xnull,sizeof(xnull));
386 if (relaydeb) MESSAGE("%ld",*arg);
387 return xbuf_add(buf,(LPBYTE)arg,4);
389 /* We use ourselves to marshal the value further */
391 VARIANT *vt = (VARIANT*)arg;
392 DWORD vttype = V_VT(vt);
394 hres = xbuf_add(buf,(LPBYTE)&vttype,sizeof(vttype));
395 if (hres) return hres;
397 if (relaydeb) MESSAGE("Vt %ld ",vttype);
398 /* shield your eyes, bad pointer voodoo below */
399 return marshall_param(tinfo,elem,&tdesc2,(DWORD*)&(V_I4(vt)),buf);
402 return marshall_param(tinfo,elem,tdesc->u.lptdesc,(DWORD*)*arg,buf);
404 hres = _marshal_interface(buf,&(buf->iid),(LPUNKNOWN)arg);
406 FIXME("Failed unmarshaling VT_VOID with guid %s?\n",debugstr_guid(&(buf->iid)));
409 case VT_USERDEFINED: {
413 /*FIXME("VT_USERDEFINED arg is %p, *arg is %p\n",arg,*arg);*/
414 hres = ITypeInfo_GetRefTypeInfo(tinfo,tdesc->u.hreftype,&tinfo2);
416 FIXME("Could not get typeinfo of hreftype %lx for VT_USERDEFINED.\n",tdesc->u.hreftype);
419 ITypeInfo_GetTypeAttr(tinfo2,&tattr);
420 switch (tattr->typekind) {
421 case TKIND_INTERFACE:
422 if (relaydeb) MESSAGE("if(%p), vtbl %p",arg,(LPVOID)*arg);
423 hres = _marshal_interface(buf,&(tattr->guid),(LPUNKNOWN)arg);
426 if (relaydeb) MESSAGE("record %p",arg);
428 memcpy(&(buf->iid),arg,sizeof(buf->iid));
429 hres = xbuf_add(buf,(LPBYTE)arg,tattr->cbSizeInstance);
432 FIXME("Don't know how to marshal type kind %d\n",tattr->typekind);
436 ITypeInfo_Release(tinfo2);
440 ERR("Cannot marshal type %d\n",tdesc->vt);
441 /*dump_ELEMDESC(elem);*/
449 ITypeInfo *tinfo, ELEMDESC *elem, TYPEDESC *tdesc, DWORD *arg, marshal_state *buf
452 int relaydeb = TRACE_ON(olerelay);
454 if (!tdesc) tdesc = &(elem->tdesc);
458 xbuf_get(buf,(LPBYTE)&x,sizeof(x));
460 if (relaydeb) MESSAGE("%ld ",x);
464 if ((tdesc->u.lptdesc->vt != VT_USERDEFINED) &&
465 (tdesc->u.lptdesc->vt != VT_VOID)
467 hres = unmarshall_param(tinfo,elem,tdesc->u.lptdesc,(DWORD*)(*arg),buf);
469 hres = unmarshall_param(tinfo,elem,tdesc->u.lptdesc,arg,buf);
470 if (relaydeb) MESSAGE("%p ",(LPVOID)*arg);
472 case VT_USERDEFINED: {
476 if (relaydeb) MESSAGE("%p",arg);
477 hres = ITypeInfo_GetRefTypeInfo(tinfo,tdesc->u.hreftype,&tinfo2);
479 FIXME("Could not get typeinfo of hreftype %lx for VT_USERDEFINED.\n",tdesc->u.hreftype);
482 hres = ITypeInfo_GetTypeAttr(tinfo2,&tattr);
484 FIXME("Could not get typeattr in VT_USERDEFINED.\n");
487 switch (tattr->typekind) {
488 case TKIND_INTERFACE:
489 hres = _unmarshal_interface(buf,&(tattr->guid),(LPUNKNOWN*)arg);
492 hres = xbuf_get(buf,(LPBYTE)arg,tattr->cbSizeInstance);
496 FIXME("Don't know how to marshal type kind %d\n",tattr->typekind);
498 ITypeInfo_Release(tinfo2);
502 /* Hacky. If we are LPVOID* we apparently have to guess the IID
503 * for the interface. This sucks pretty badly. */
504 return _unmarshal_interface(buf,&(buf->iid),(LPUNKNOWN*)arg);
505 default: ERR("Cannot unmarshal type %d\n",tdesc->vt);
511 /* Searches function, also in inherited interfaces */
514 ITypeInfo *tinfo, int iMethod, FUNCDESC **fdesc,
515 BSTR *iname, BSTR *fname
520 if (fname) *fname = NULL;
521 if (iname) *iname = NULL;
524 hres = ITypeInfo_GetFuncDesc(tinfo, i, fdesc);
530 hres = ITypeInfo_GetTypeAttr(tinfo, &attr);
532 FIXME("GetTypeAttr failed with %lx\n",hres);
535 /* Not found, so look in inherited ifaces. */
536 for (j=0;j<attr->cImplTypes;j++) {
537 hres = ITypeInfo_GetRefTypeOfImplType(tinfo, j, &href);
539 FIXME("Did not find a reftype for interface offset %d?\n",j);
542 hres = ITypeInfo_GetRefTypeInfo(tinfo, href, &tinfo2);
544 FIXME("Did not find a typeinfo for reftype %ld?\n",href);
547 hres = _get_funcdesc(tinfo2,iMethod,fdesc,iname,fname);
548 ITypeInfo_Release(tinfo2);
549 if (!hres) return S_OK;
553 if (((*fdesc)->oVft/4) == iMethod) {
555 ITypeInfo_GetDocumentation(tinfo,(*fdesc)->memid,fname,NULL,NULL,NULL);
557 ITypeInfo_GetDocumentation(tinfo,-1,iname,NULL,NULL,NULL);
565 /* how much space do we use on stack in DWORD steps. */
567 _argsize(DWORD vt_type) {
570 return (sizeof(VARIANT)+3)/sizeof(DWORD);
577 xCall(LPVOID retptr, int method, TMProxyImpl *tpinfo /*, args */) {
578 DWORD *args = ((DWORD*)&tpinfo)+1, *xargs;
581 int i, relaydeb = TRACE_ON(olerelay);
589 hres = _get_funcdesc(tpinfo->tinfo,method,&fdesc,&iname,&fname);
591 ERR("Did not find typeinfo/funcdesc entry for method %d!\n",method);
595 /*dump_FUNCDESC(fdesc);*/
597 TRACE_(olerelay)("->");
599 MESSAGE("%s:",debugstr_w(iname));
601 MESSAGE("%s(%d)",debugstr_w(fname),method);
603 MESSAGE("%d",method);
605 if (iname) SysFreeString(iname);
606 if (fname) SysFreeString(fname);
608 /* Need them for hack below */
609 memset(names,0,sizeof(names));
610 ITypeInfo_GetNames(tpinfo->tinfo,fdesc->memid,names,sizeof(names)/sizeof(names[0]),&nrofnames);
611 if (nrofnames > sizeof(names)/sizeof(names[0])) {
612 ERR("Need more names!\n");
614 memset(&buf,0,sizeof(buf));
615 buf.iid = IID_IUnknown;
617 xbuf_add(&buf,(LPBYTE)args[0],sizeof(IID));
618 if (relaydeb) MESSAGE("riid=%s,[out]",debugstr_guid((REFIID)args[0]));
621 for (i=0;i<fdesc->cParams;i++) {
622 ELEMDESC *elem = fdesc->lprgelemdescParam+i;
625 if (i+1<nrofnames && names[i+1])
626 MESSAGE("%s=",debugstr_w(names[i+1]));
628 if (((i+1)<nrofnames) && !lstrcmpW(names[i+1],riidW)) {
629 buf.thisisiid = TRUE;
631 buf.thisisiid = FALSE;
633 /* No need to marshal other data than FIN */
634 if (!(elem->u.paramdesc.wParamFlags & PARAMFLAG_FIN)) {
635 xargs+=_argsize(elem->tdesc.vt);
636 if (relaydeb) MESSAGE("[out]");
639 hres = marshall_param(tpinfo->tinfo,elem,NULL,xargs,&buf);
640 xargs+=_argsize(elem->tdesc.vt);
642 FIXME("Failed to marshall param, hres %lx\n",hres);
647 if (relaydeb) MESSAGE(")");
648 memset(&msg,0,sizeof(msg));
649 msg.cbBuffer = buf.curoff;
650 msg.iMethod = method;
651 hres = IRpcChannelBuffer_GetBuffer(tpinfo->chanbuf,&msg,&(tpinfo->iid));
653 FIXME("RpcChannelBuffer GetBuffer failed, %lx\n",hres);
656 memcpy(msg.Buffer,buf.base,buf.curoff);
657 hres = IRpcChannelBuffer_SendReceive(tpinfo->chanbuf,&msg,&status);
659 FIXME("RpcChannelBuffer SendReceive failed, %lx\n",hres);
662 if (relaydeb) MESSAGE(" = %08lx (",status);
664 buf.base = HeapReAlloc(GetProcessHeap(),0,buf.base,msg.cbBuffer);
666 buf.base = HeapAlloc(GetProcessHeap(),0,msg.cbBuffer);
667 buf.size = msg.cbBuffer;
668 memcpy(buf.base,msg.Buffer,buf.size);
671 _unmarshal_interface(&buf,(REFIID)args[0],(LPUNKNOWN*)args[1]);
672 if (relaydeb) MESSAGE("[in],%p",*((DWORD**)args[1]));
675 for (i=0;i<fdesc->cParams;i++) {
676 ELEMDESC *elem = fdesc->lprgelemdescParam+i;
680 if (i+1<nrofnames && names[i+1]) MESSAGE("%s=",debugstr_w(names[i+1]));
682 /* No need to marshal other data than FOUT I think */
683 if (!(elem->u.paramdesc.wParamFlags & PARAMFLAG_FOUT)) {
684 xargs += _argsize(elem->tdesc.vt);
685 if (relaydeb) MESSAGE("[in]");
688 hres = unmarshall_param(tpinfo->tinfo,elem,&(elem->tdesc),xargs,&buf);
689 xargs += _argsize(elem->tdesc.vt);
691 FIXME("Failed to unmarshall param, hres %lx\n",hres);
696 if (relaydeb) MESSAGE(")\n");
697 HeapFree(GetProcessHeap(),0,buf.base);
701 static HRESULT WINAPI
702 PSFacBuf_CreateProxy(
703 LPPSFACTORYBUFFER iface, IUnknown* pUnkOuter, REFIID riid,
704 IRpcProxyBuffer **ppProxy, LPVOID *ppv
712 TRACE("(...%s...)\n",debugstr_guid(riid));
713 hres = _get_typeinfo_for_iid(riid,&tinfo);
715 FIXME("No typeinfo for %s?\n",debugstr_guid(riid));
718 nroffuncs = _nroffuncs(tinfo);
719 proxy = HeapAlloc(GetProcessHeap(),HEAP_ZERO_MEMORY,sizeof(TMProxyImpl));
720 if (!proxy) return E_OUTOFMEMORY;
721 proxy->asmstubs=HeapAlloc(GetProcessHeap(),0,sizeof(TMAsmProxy)*nroffuncs);
723 assert(sizeof(TMAsmProxy) == 12);
725 proxy->lpvtbl = HeapAlloc(GetProcessHeap(),0,sizeof(LPBYTE)*nroffuncs);
726 for (i=0;i<nroffuncs;i++) {
728 TMAsmProxy *xasm = proxy->asmstubs+i;
730 /* nrofargs without This */
732 case 0: nrofargs = 2;
734 case 1: case 2: nrofargs = 0;
738 hres = _get_funcdesc(tinfo,i,&fdesc,NULL,NULL);
740 FIXME("GetFuncDesc %lx should not fail here.\n",hres);
743 /* some args take more than 4 byte on the stack */
745 for (j=0;j<fdesc->cParams;j++)
746 nrofargs += _argsize(fdesc->lprgelemdescParam[j].tdesc.vt);
748 if (fdesc->callconv != CC_STDCALL) {
749 ERR("calling convention is not stdcall????\n");
755 /* popl %eax - return ptr
762 * arg3 arg2 arg1 <method> <returnptr>
764 xasm->popleax = 0x58;
765 xasm->pushlval = 0x6a;
767 xasm->pushleax = 0x50;
768 xasm->lcall = 0xe8; /* relative jump */
769 xasm->xcall = (DWORD)xCall;
770 xasm->xcall -= (DWORD)&(xasm->lret);
772 xasm->bytestopop= (nrofargs+2)*4; /* pop args, This, iMethod */
773 proxy->lpvtbl[i] = (DWORD)xasm;
775 proxy->lpvtbl2 = &tmproxyvtable;
777 proxy->tinfo = tinfo;
778 memcpy(&proxy->iid,riid,sizeof(*riid));
779 *ppv = (LPVOID)proxy;
780 *ppProxy = (IRpcProxyBuffer *)&(proxy->lpvtbl2);
784 typedef struct _TMStubImpl {
785 ICOM_VTABLE(IRpcStubBuffer) *lpvtbl;
793 static HRESULT WINAPI
794 TMStubImpl_QueryInterface(LPRPCSTUBBUFFER iface, REFIID riid, LPVOID *ppv) {
795 if (IsEqualIID(riid,&IID_IRpcStubBuffer)||IsEqualIID(riid,&IID_IUnknown)){
796 *ppv = (LPVOID)iface;
797 IRpcStubBuffer_AddRef(iface);
800 FIXME("%s, not supported IID.\n",debugstr_guid(riid));
801 return E_NOINTERFACE;
805 TMStubImpl_AddRef(LPRPCSTUBBUFFER iface) {
806 ICOM_THIS(TMStubImpl,iface);
813 TMStubImpl_Release(LPRPCSTUBBUFFER iface) {
814 ICOM_THIS(TMStubImpl,iface);
819 HeapFree(GetProcessHeap(),0,This);
823 static HRESULT WINAPI
824 TMStubImpl_Connect(LPRPCSTUBBUFFER iface, LPUNKNOWN pUnkServer) {
825 ICOM_THIS(TMStubImpl,iface);
827 IUnknown_AddRef(pUnkServer);
828 This->pUnk = pUnkServer;
833 TMStubImpl_Disconnect(LPRPCSTUBBUFFER iface) {
834 ICOM_THIS(TMStubImpl,iface);
836 IUnknown_Release(This->pUnk);
843 ITypeInfo *tinfo, ELEMDESC *elem, TYPEDESC *tdesc, DWORD *arg, marshal_state *buf
851 VARIANT *vt = (VARIANT*)arg;
854 hres = xbuf_get(buf,(LPBYTE)&vttype,sizeof(vttype));
855 if (hres) return hres;
856 memset(&tdesc2,0,sizeof(tdesc));
859 return stuballoc_param(tinfo,elem,&tdesc2,&(V_I4(vt)),buf);
861 case VT_BOOL: case VT_I4:
862 xbuf_get(buf,(LPBYTE)arg,sizeof(DWORD));
868 hres = xbuf_get(buf,(LPBYTE)&len,sizeof(DWORD));
871 str = HeapAlloc(GetProcessHeap(),HEAP_ZERO_MEMORY,len+sizeof(WCHAR));
872 hres = xbuf_get(buf,(LPBYTE)str,len);
873 if (hres) return hres;
874 *arg = (DWORD)SysAllocStringLen(str,len);
875 HeapFree(GetProcessHeap(),0,str);
879 if ((tdesc->u.lptdesc->vt != VT_USERDEFINED) &&
880 (tdesc->u.lptdesc->vt != VT_VOID)
882 *arg=(DWORD)HeapAlloc(GetProcessHeap(),HEAP_ZERO_MEMORY,sizeof(LPVOID));
885 tdesc = tdesc->u.lptdesc;
888 /* FIXME: UNKNOWN is unknown ..., but allocate 4 byte for it */
889 *arg=(DWORD)HeapAlloc(GetProcessHeap(),HEAP_ZERO_MEMORY,sizeof(DWORD));
892 *arg = (DWORD)HeapAlloc(GetProcessHeap(),0,sizeof(LPVOID));
894 if (elem->u.paramdesc.wParamFlags & PARAMFLAG_FIN)
895 hres = _unmarshal_interface(buf,&(buf->iid),(LPUNKNOWN*)arg);
897 case VT_USERDEFINED: {
898 if (elem->u.paramdesc.wParamFlags & PARAMFLAG_FIN) {
902 hres = ITypeInfo_GetRefTypeInfo(tinfo,tdesc->u.hreftype,&tinfo2);
904 FIXME("Could not get typeinfo of hreftype %lx for VT_USERDEFINED.\n",tdesc->u.hreftype);
907 hres = ITypeInfo_GetTypeAttr(tinfo2,&tattr);
909 FIXME("Could not get typeattr in VT_USERDEFINED.\n");
912 switch (tattr->typekind) {
913 case TKIND_INTERFACE:
914 hres = _unmarshal_interface(buf,&(tattr->guid),(LPUNKNOWN*)arg);
917 *arg = (DWORD)HeapAlloc(GetProcessHeap(),0,tattr->cbSizeInstance);
918 hres = xbuf_get(buf,(LPBYTE)*arg,tattr->cbSizeInstance);
920 memcpy(&(buf->iid),(LPBYTE)*arg,sizeof(buf->iid));
923 FIXME("Don't know how to marshal type kind %d\n",tattr->typekind);
927 ITypeInfo_Release(tinfo2);
930 *arg = (DWORD)HeapAlloc(GetProcessHeap(),0,sizeof(LPVOID));
935 ERR("No handler for VT type %d, just allocating 4 bytes.\n",tdesc->vt);
936 *arg=(DWORD)HeapAlloc(GetProcessHeap(),HEAP_ZERO_MEMORY,sizeof(DWORD));
944 ITypeInfo *tinfo, ELEMDESC *elem, TYPEDESC *tdesc, DWORD *arg, marshal_state *buf
948 if (!tdesc) tdesc = &(elem->tdesc);
954 if (elem->u.paramdesc.wParamFlags & PARAMFLAG_FOUT)
955 hres = xbuf_add(buf,(LPBYTE)arg,sizeof(DWORD));
959 VARIANT *vt = (VARIANT*)arg;
960 DWORD vttype = V_VT(vt);
963 if (elem->u.paramdesc.wParamFlags & PARAMFLAG_FOUT) {
964 hres = xbuf_add(buf,(LPBYTE)&vttype,sizeof(vttype));
965 if (hres) return hres;
967 /* need to recurse since we need to free the stuff */
968 hres = stubunalloc_param(tinfo,elem,&tdesc2,&(V_I4(vt)),buf);
972 if (elem->u.paramdesc.wParamFlags & PARAMFLAG_FOUT) {
973 DWORD *bstr = ((DWORD*)(*arg))-1;
975 hres = xbuf_add(buf,(LPBYTE)bstr,bstr[0]+4);
979 SysFreeString((BSTR)*arg);
983 /*FIXME("VT_PTR *arg is %p\n",(LPVOID)*arg);*/
984 if ((tdesc->u.lptdesc->vt != VT_USERDEFINED) &&
985 (tdesc->u.lptdesc->vt != VT_VOID)
987 hres = stubunalloc_param(tinfo,elem,tdesc->u.lptdesc,arg,buf);
989 hres = stubunalloc_param(tinfo,elem,tdesc->u.lptdesc,(DWORD*)*arg,buf);
990 HeapFree(GetProcessHeap(),0,(LPVOID)*arg);
994 if (elem->u.paramdesc.wParamFlags & PARAMFLAG_FOUT) {
995 FIXME("Marshaling back VT_UNKNOWN %lx\n",*arg);
996 hres = xbuf_add(buf,(LPBYTE)*arg,sizeof(DWORD));
998 HeapFree(GetProcessHeap(),0,(LPVOID)*arg);
1002 if (elem->u.paramdesc.wParamFlags & PARAMFLAG_FOUT)
1003 hres = _marshal_interface(buf,&(buf->iid),(LPUNKNOWN)*arg);
1005 case VT_USERDEFINED: {
1009 if (elem->u.paramdesc.wParamFlags & PARAMFLAG_FOUT) {
1010 /*FIXME("VT_USERDEFINED arg is %p, *arg is %p\n",arg,*arg);*/
1011 hres = ITypeInfo_GetRefTypeInfo(tinfo,tdesc->u.hreftype,&tinfo2);
1013 FIXME("Could not get typeinfo of hreftype %lx for VT_USERDEFINED.\n",tdesc->u.hreftype);
1016 ITypeInfo_GetTypeAttr(tinfo2,&tattr);
1017 switch (tattr->typekind) {
1018 case TKIND_INTERFACE:
1019 hres = _marshal_interface(buf,&(tattr->guid),(LPUNKNOWN)*arg);
1022 hres = xbuf_add(buf,(LPBYTE)arg,tattr->cbSizeInstance);
1025 FIXME("Don't know how to marshal type kind %d\n",tattr->typekind);
1029 ITypeInfo_Release(tinfo2);
1034 ERR("Unhandled marshal type %d.\n",tdesc->vt);
1035 HeapFree(GetProcessHeap(),0,(LPVOID)*arg);
1040 static HRESULT WINAPI
1042 LPRPCSTUBBUFFER iface, RPCOLEMESSAGE* xmsg,IRpcChannelBuffer*rpcchanbuf
1046 ICOM_THIS(TMStubImpl,iface);
1048 DWORD *args, res, *xargs, nrofargs;
1053 memset(&buf,0,sizeof(buf));
1054 buf.size = xmsg->cbBuffer;
1055 buf.base = xmsg->Buffer;
1057 buf.iid = IID_IUnknown;
1060 if (xmsg->iMethod == 0) { /* QI */
1062 /* in: IID, out: <iface> */
1064 xbuf_get(&buf,(LPBYTE)&xiid,sizeof(xiid));
1066 hres = _marshal_interface(&buf,&xiid,This->pUnk);
1067 xmsg->Buffer = buf.base; /* Might have been reallocated */
1068 xmsg->cbBuffer = buf.size;
1071 hres = _get_funcdesc(This->tinfo,xmsg->iMethod,&fdesc,NULL,NULL);
1073 FIXME("GetFuncDesc on method %ld failed with %lx\n",xmsg->iMethod,hres);
1076 /* Need them for hack below */
1077 memset(names,0,sizeof(names));
1078 ITypeInfo_GetNames(This->tinfo,fdesc->memid,names,sizeof(names)/sizeof(names[0]),&nrofnames);
1079 if (nrofnames > sizeof(names)/sizeof(names[0])) {
1080 ERR("Need more names!\n");
1083 /*dump_FUNCDESC(fdesc);*/
1085 for (i=0;i<fdesc->cParams;i++)
1086 nrofargs += _argsize(fdesc->lprgelemdescParam[i].tdesc.vt);
1087 args = HeapAlloc(GetProcessHeap(),0,(nrofargs+1)*sizeof(DWORD));
1088 if (!args) return E_OUTOFMEMORY;
1090 /* Allocate all stuff used by call. */
1092 for (i=0;i<fdesc->cParams;i++) {
1093 ELEMDESC *elem = fdesc->lprgelemdescParam+i;
1095 if (((i+1)<nrofnames) && !lstrcmpW(names[i+1],riidW))
1096 buf.thisisiid = TRUE;
1098 buf.thisisiid = FALSE;
1099 hres = stuballoc_param(This->tinfo,elem,&(elem->tdesc),xargs,&buf);
1100 xargs += _argsize(elem->tdesc.vt);
1102 FIXME("Failed to stuballoc param %s, hres %lx\n",debugstr_w(names[i+1]),hres);
1106 hres = IUnknown_QueryInterface(This->pUnk,&(This->iid),(LPVOID*)&(args[0]));
1108 ERR("Does not support iface %s\n",debugstr_guid(&(This->iid)));
1112 (*((LPVOID**)args[0]))[fdesc->oVft/4],
1117 IUnknown_Release((LPUNKNOWN)args[0]);
1120 for (i=0;i<fdesc->cParams;i++) {
1121 ELEMDESC *elem = fdesc->lprgelemdescParam+i;
1122 hres = stubunalloc_param(This->tinfo,elem,NULL,xargs,&buf);
1123 xargs += _argsize(elem->tdesc.vt);
1125 FIXME("Failed to stuballoc param, hres %lx\n",hres);
1129 /* might need to use IRpcChannelBuffer_GetBuffer ? */
1130 xmsg->cbBuffer = buf.curoff;
1131 xmsg->Buffer = buf.base;
1132 HeapFree(GetProcessHeap(),0,args);
1136 static LPRPCSTUBBUFFER WINAPI
1137 TMStubImpl_IsIIDSupported(LPRPCSTUBBUFFER iface, REFIID riid) {
1138 FIXME("Huh (%s)?\n",debugstr_guid(riid));
1143 TMStubImpl_CountRefs(LPRPCSTUBBUFFER iface) {
1144 ICOM_THIS(TMStubImpl,iface);
1146 return This->ref; /*FIXME? */
1149 static HRESULT WINAPI
1150 TMStubImpl_DebugServerQueryInterface(LPRPCSTUBBUFFER iface, LPVOID *ppv) {
1155 TMStubImpl_DebugServerRelease(LPRPCSTUBBUFFER iface, LPVOID ppv) {
1159 ICOM_VTABLE(IRpcStubBuffer) tmstubvtbl = {
1160 ICOM_MSVTABLE_COMPAT_DummyRTTIVALUE
1161 TMStubImpl_QueryInterface,
1165 TMStubImpl_Disconnect,
1167 TMStubImpl_IsIIDSupported,
1168 TMStubImpl_CountRefs,
1169 TMStubImpl_DebugServerQueryInterface,
1170 TMStubImpl_DebugServerRelease
1173 static HRESULT WINAPI
1174 PSFacBuf_CreateStub(
1175 LPPSFACTORYBUFFER iface, REFIID riid,IUnknown *pUnkServer,
1176 IRpcStubBuffer** ppStub
1182 TRACE("(%s,%p,%p)\n",debugstr_guid(riid),pUnkServer,ppStub);
1183 hres = _get_typeinfo_for_iid(riid,&tinfo);
1185 FIXME("No typeinfo for %s?\n",debugstr_guid(riid));
1188 stub = HeapAlloc(GetProcessHeap(),HEAP_ZERO_MEMORY,sizeof(TMStubImpl));
1190 return E_OUTOFMEMORY;
1191 stub->lpvtbl = &tmstubvtbl;
1193 stub->tinfo = tinfo;
1194 memcpy(&(stub->iid),riid,sizeof(*riid));
1195 hres = IRpcStubBuffer_Connect((LPRPCSTUBBUFFER)stub,pUnkServer);
1196 *ppStub = (LPRPCSTUBBUFFER)stub;
1198 FIXME("Connect to pUnkServer failed?\n");
1202 static ICOM_VTABLE(IPSFactoryBuffer) psfacbufvtbl = {
1203 ICOM_MSVTABLE_COMPAT_DummyRTTIVALUE
1204 PSFacBuf_QueryInterface,
1207 PSFacBuf_CreateProxy,
1211 /* This is the whole PSFactoryBuffer object, just the vtableptr */
1212 static ICOM_VTABLE(IPSFactoryBuffer) *lppsfac = &psfacbufvtbl;
1214 /***********************************************************************
1215 * DllGetClassObject [OLE32.63]
1218 TypeLibFac_DllGetClassObject(REFCLSID rclsid, REFIID iid,LPVOID *ppv)
1220 if (IsEqualIID(iid,&IID_IPSFactoryBuffer)) {
1224 return E_NOINTERFACE;