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
40 #include "wine/unicode.h"
41 #include "wine/winbase16.h"
42 #include "compobj_private.h"
45 #include "wine/debug.h"
47 WINE_DEFAULT_DEBUG_CHANNEL(ole);
49 extern const CLSID CLSID_DfMarshal;
51 /* Marshalling just passes a unique identifier to the remote client,
52 * that makes it possible to find the passed interface again.
54 * So basically we need a set of values that make it unique.
56 * Process Identifier, Object IUnknown ptr, IID
58 * Note that the IUnknown_QI(ob,xiid,&ppv) always returns the SAME ppv value!
62 get_facbuf_for_iid(REFIID riid,IPSFactoryBuffer **facbuf) {
66 if ((hres = CoGetPSClsid(riid,&pxclsid)))
68 return CoGetClassObject(&pxclsid,CLSCTX_INPROC_SERVER,NULL,&IID_IPSFactoryBuffer,(LPVOID*)facbuf);
71 typedef struct _wine_marshal_data {
76 typedef struct _mid2unknown {
81 typedef struct _mid2stub {
88 static mid2stub *stubs = NULL;
89 static int nrofstubs = 0;
91 static mid2unknown *proxies = NULL;
92 static int nrofproxies = 0;
94 void MARSHAL_Invalidate_Stub_From_MID(wine_marshal_id *mid) {
97 for (i=0;i<nrofstubs;i++) {
98 if (!stubs[i].valid) continue;
100 if (MARSHAL_Compare_Mids(mid,&(stubs[i].mid))) {
101 stubs[i].valid = FALSE;
110 MARSHAL_Find_Stub_Buffer(wine_marshal_id *mid,IRpcStubBuffer **stub) {
113 for (i=0;i<nrofstubs;i++) {
114 if (!stubs[i].valid) continue;
116 if (MARSHAL_Compare_Mids(mid,&(stubs[i].mid))) {
117 *stub = stubs[i].stub;
118 IUnknown_AddRef((*stub));
126 MARSHAL_Find_Stub(wine_marshal_id *mid,LPUNKNOWN *pUnk) {
129 for (i=0;i<nrofstubs;i++) {
130 if (!stubs[i].valid) continue;
132 if (MARSHAL_Compare_Mids(mid,&(stubs[i].mid))) {
133 *pUnk = stubs[i].pUnkServer;
134 IUnknown_AddRef((*pUnk));
142 MARSHAL_Register_Stub(wine_marshal_id *mid,LPUNKNOWN pUnk,IRpcStubBuffer *stub) {
144 if (!MARSHAL_Find_Stub(mid,&xPunk)) {
145 FIXME("Already have entry for (%lx/%s)!\n",mid->objectid,debugstr_guid(&(mid->iid)));
149 stubs=HeapReAlloc(GetProcessHeap(),0,stubs,sizeof(stubs[0])*(nrofstubs+1));
151 stubs=HeapAlloc(GetProcessHeap(),0,sizeof(stubs[0]));
152 if (!stubs) return E_OUTOFMEMORY;
153 stubs[nrofstubs].stub = stub;
154 stubs[nrofstubs].pUnkServer = pUnk;
155 memcpy(&(stubs[nrofstubs].mid),mid,sizeof(*mid));
156 stubs[nrofstubs].valid = TRUE; /* set to false when released by ReleaseMarshalData */
162 MARSHAL_Disconnect_Proxies() {
165 TRACE("Disconnecting %d proxies\n", nrofproxies);
167 for (i = 0; i < nrofproxies; i++)
168 IRpcProxyBuffer_Disconnect((IRpcProxyBuffer*)proxies[i].pUnk);
174 MARSHAL_Register_Proxy(wine_marshal_id *mid,LPUNKNOWN punk) {
177 for (i=0;i<nrofproxies;i++) {
178 if (MARSHAL_Compare_Mids(mid,&(proxies[i].mid))) {
179 ERR("Already have mid?\n");
184 proxies = HeapReAlloc(GetProcessHeap(),0,proxies,sizeof(proxies[0])*(nrofproxies+1));
186 proxies = HeapAlloc(GetProcessHeap(),0,sizeof(proxies[0]));
187 memcpy(&(proxies[nrofproxies].mid),mid,sizeof(*mid));
188 proxies[nrofproxies].pUnk = punk;
190 IUnknown_AddRef(punk);
194 /********************** StdMarshal implementation ****************************/
195 typedef struct _StdMarshalImpl {
196 ICOM_VTABLE(IMarshal) *lpvtbl;
201 LPVOID pvDestContext;
205 static HRESULT WINAPI
206 StdMarshalImpl_QueryInterface(LPMARSHAL iface,REFIID riid,LPVOID *ppv) {
208 if (IsEqualIID(&IID_IUnknown,riid) || IsEqualIID(&IID_IMarshal,riid)) {
210 IUnknown_AddRef(iface);
213 FIXME("No interface for %s.\n",debugstr_guid(riid));
214 return E_NOINTERFACE;
218 StdMarshalImpl_AddRef(LPMARSHAL iface) {
219 ICOM_THIS(StdMarshalImpl,iface);
225 StdMarshalImpl_Release(LPMARSHAL iface) {
226 ICOM_THIS(StdMarshalImpl,iface);
231 HeapFree(GetProcessHeap(),0,This);
235 static HRESULT WINAPI
236 StdMarshalImpl_GetUnmarshalClass(
237 LPMARSHAL iface, REFIID riid, void* pv, DWORD dwDestContext,
238 void* pvDestContext, DWORD mshlflags, CLSID* pCid
240 memcpy(pCid,&CLSID_DfMarshal,sizeof(CLSID_DfMarshal));
244 static HRESULT WINAPI
245 StdMarshalImpl_GetMarshalSizeMax(
246 LPMARSHAL iface, REFIID riid, void* pv, DWORD dwDestContext,
247 void* pvDestContext, DWORD mshlflags, DWORD* pSize
249 *pSize = sizeof(wine_marshal_id)+sizeof(wine_marshal_data);
253 static HRESULT WINAPI
254 StdMarshalImpl_MarshalInterface(
255 LPMARSHAL iface, IStream *pStm,REFIID riid, void* pv, DWORD dwDestContext,
256 void* pvDestContext, DWORD mshlflags
259 wine_marshal_data md;
263 IRpcStubBuffer *stub;
264 IPSFactoryBuffer *psfacbuf;
266 TRACE("(...,%s,...)\n",debugstr_guid(riid));
268 IUnknown_QueryInterface((LPUNKNOWN)pv,&IID_IUnknown,(LPVOID*)&pUnk);
269 mid.processid = GetCurrentProcessId();
270 mid.objectid = (DWORD)pUnk; /* FIXME */
271 IUnknown_Release(pUnk);
272 memcpy(&mid.iid,riid,sizeof(mid.iid));
273 md.dwDestContext = dwDestContext;
274 md.mshlflags = mshlflags;
275 hres = IStream_Write(pStm,&mid,sizeof(mid),&res);
276 if (hres) return hres;
277 hres = IStream_Write(pStm,&md,sizeof(md),&res);
278 if (hres) return hres;
280 if (SUCCEEDED(MARSHAL_Find_Stub_Buffer(&mid,&stub))) {
281 /* Find_Stub_Buffer gives us a ref but we want to keep it, as if we'd created a new one */
282 TRACE("Found RpcStubBuffer %p\n", stub);
285 hres = get_facbuf_for_iid(riid,&psfacbuf);
286 if (hres) return hres;
288 hres = IPSFactoryBuffer_CreateStub(psfacbuf,riid,pv,&stub);
289 IPSFactoryBuffer_Release(psfacbuf);
291 FIXME("Failed to create a stub for %s\n",debugstr_guid(riid));
294 IUnknown_QueryInterface((LPUNKNOWN)pv,riid,(LPVOID*)&pUnk);
295 MARSHAL_Register_Stub(&mid,pUnk,stub);
296 IUnknown_Release(pUnk);
300 static HRESULT WINAPI
301 StdMarshalImpl_UnmarshalInterface(
302 LPMARSHAL iface, IStream *pStm, REFIID riid, void **ppv
305 wine_marshal_data md;
308 IPSFactoryBuffer *psfacbuf;
309 IRpcProxyBuffer *rpcproxy;
310 IRpcChannelBuffer *chanbuf;
312 TRACE("(...,%s,....)\n",debugstr_guid(riid));
313 hres = IStream_Read(pStm,&mid,sizeof(mid),&res);
314 if (hres) return hres;
315 hres = IStream_Read(pStm,&md,sizeof(md),&res);
316 if (hres) return hres;
317 if (SUCCEEDED(MARSHAL_Find_Stub(&mid,(LPUNKNOWN*)ppv))) {
318 FIXME("Calling back to ourselves for %s!\n",debugstr_guid(riid));
321 if (IsEqualIID(riid, &IID_IUnknown) || IsEqualIID(riid, &IID_NULL)) {
322 /* should return proxy manager IUnknown object */
323 FIXME("Special treatment required for IID of %s\n", debugstr_guid(riid));
325 hres = get_facbuf_for_iid(riid,&psfacbuf);
326 if (hres) return hres;
327 hres = IPSFactoryBuffer_CreateProxy(psfacbuf,NULL,riid,&rpcproxy,ppv);
329 FIXME("Failed to create a proxy for %s\n",debugstr_guid(riid));
333 MARSHAL_Register_Proxy(&mid, (LPUNKNOWN) rpcproxy);
335 hres = PIPE_GetNewPipeBuf(&mid,&chanbuf);
336 IPSFactoryBuffer_Release(psfacbuf);
338 ERR("Failed to get an rpc channel buffer for %s\n",debugstr_guid(riid));
340 /* Connect the channel buffer to the proxy and release the no longer
342 * NOTE: The proxy should have taken an extra reference because it also
343 * aggregates the object, so we can safely release our reference to it. */
344 IRpcProxyBuffer_Connect(rpcproxy,chanbuf);
345 IRpcProxyBuffer_Release(rpcproxy);
346 /* IRpcProxyBuffer takes a reference on the channel buffer and
347 * we no longer need it, so release it */
348 IRpcChannelBuffer_Release(chanbuf);
353 static HRESULT WINAPI
354 StdMarshalImpl_ReleaseMarshalData(LPMARSHAL iface, IStream *pStm) {
358 IRpcStubBuffer *stub = NULL;
361 hres = IStream_Read(pStm,&mid,sizeof(mid),&res);
362 if (hres) return hres;
364 for (i=0; i < nrofstubs; i++)
366 if (!stubs[i].valid) continue;
368 if (MARSHAL_Compare_Mids(&mid, &(stubs[i].mid)))
370 stub = stubs[i].stub;
377 FIXME("Could not map MID to stub??\n");
381 res = IRpcStubBuffer_Release(stub);
382 stubs[i].valid = FALSE;
383 TRACE("stub refcount of %p is %ld\n", stub, res);
388 static HRESULT WINAPI
389 StdMarshalImpl_DisconnectObject(LPMARSHAL iface, DWORD dwReserved) {
390 FIXME("(), stub!\n");
394 ICOM_VTABLE(IMarshal) stdmvtbl = {
395 ICOM_MSVTABLE_COMPAT_DummyRTTIVALUE
396 StdMarshalImpl_QueryInterface,
397 StdMarshalImpl_AddRef,
398 StdMarshalImpl_Release,
399 StdMarshalImpl_GetUnmarshalClass,
400 StdMarshalImpl_GetMarshalSizeMax,
401 StdMarshalImpl_MarshalInterface,
402 StdMarshalImpl_UnmarshalInterface,
403 StdMarshalImpl_ReleaseMarshalData,
404 StdMarshalImpl_DisconnectObject
407 /***********************************************************************
408 * CoGetStandardMarshal [OLE32.@]
410 * When the COM library in the client process receives a marshalled
411 * interface pointer, it looks for a CLSID to be used in creating a proxy
412 * for the purposes of unmarshalling the packet. If the packet does not
413 * contain a CLSID for the proxy, COM calls CoGetStandardMarshal, passing a
415 * This function creates a standard proxy in the client process and returns
416 * a pointer to that proxy's implementation of IMarshal.
417 * COM uses this pointer to call CoUnmarshalInterface to retrieve the pointer
418 * to the requested interface.
421 CoGetStandardMarshal(
422 REFIID riid,IUnknown *pUnk,DWORD dwDestContext,LPVOID pvDestContext,
423 DWORD mshlflags, LPMARSHAL *pMarshal
428 FIXME("(%s,NULL,%lx,%p,%lx,%p), unimplemented yet.\n",
429 debugstr_guid(riid),dwDestContext,pvDestContext,mshlflags,pMarshal
433 TRACE("(%s,%p,%lx,%p,%lx,%p)\n",
434 debugstr_guid(riid),pUnk,dwDestContext,pvDestContext,mshlflags,pMarshal
436 *pMarshal = HeapAlloc(GetProcessHeap(),0,sizeof(StdMarshalImpl));
437 dm = (StdMarshalImpl*) *pMarshal;
438 if (!dm) return E_FAIL;
439 dm->lpvtbl = &stdmvtbl;
442 memcpy(&dm->iid,riid,sizeof(dm->iid));
443 dm->dwDestContext = dwDestContext;
444 dm->pvDestContext = pvDestContext;
445 dm->mshlflags = mshlflags;
449 /* Helper function for getting Marshaler */
450 static HRESULT WINAPI
451 _GetMarshaller(REFIID riid, IUnknown *pUnk,DWORD dwDestContext,
452 void *pvDestContext, DWORD mshlFlags, LPMARSHAL *pMarshal
458 hres = IUnknown_QueryInterface(pUnk,&IID_IMarshal,(LPVOID*)pMarshal);
460 hres = CoGetStandardMarshal(riid,pUnk,dwDestContext,pvDestContext,mshlFlags,pMarshal);
464 /***********************************************************************
465 * CoGetMarshalSizeMax [OLE32.@]
468 CoGetMarshalSizeMax(ULONG *pulSize, REFIID riid, IUnknown *pUnk,
469 DWORD dwDestContext, void *pvDestContext, DWORD mshlFlags
474 hres = _GetMarshaller(riid,pUnk,dwDestContext,pvDestContext,mshlFlags,&pMarshal);
477 hres = IMarshal_GetMarshalSizeMax(pMarshal,riid,pUnk,dwDestContext,pvDestContext,mshlFlags,pulSize);
478 *pulSize += sizeof(wine_marshal_id)+sizeof(wine_marshal_data)+sizeof(CLSID);
479 IMarshal_Release(pMarshal);
484 /***********************************************************************
485 * CoMarshalInterface [OLE32.@]
488 CoMarshalInterface( IStream *pStm, REFIID riid, IUnknown *pUnk,
489 DWORD dwDestContext, void *pvDestContext, DWORD mshlflags
496 wine_marshal_data md;
500 TRACE("(%p, %s, %p, %lx, %p, %lx)\n",
501 pStm,debugstr_guid(riid),pUnk,dwDestContext,pvDestContext,mshlflags
507 STUBMGR_Start(); /* Just to be sure we have one running. */
508 mid.processid = GetCurrentProcessId();
509 IUnknown_QueryInterface(pUnk,&IID_IUnknown,(LPVOID*)&pUnknown);
510 mid.objectid = (DWORD)pUnknown;
511 IUnknown_Release(pUnknown);
512 memcpy(&mid.iid,riid,sizeof(mid.iid));
513 md.dwDestContext = dwDestContext;
514 md.mshlflags = mshlflags;
515 hres = IStream_Write(pStm,&mid,sizeof(mid),&res);
516 if (hres) return hres;
517 hres = IStream_Write(pStm,&md,sizeof(md),&res);
518 if (hres) return hres;
519 hres = _GetMarshaller(riid,pUnk,dwDestContext,pvDestContext,mshlflags,&pMarshal);
521 FIXME("Failed to get marshaller, %lx?\n",hres);
524 hres = IMarshal_GetUnmarshalClass(pMarshal,riid,pUnk,dwDestContext,pvDestContext,mshlflags,&xclsid);
526 FIXME("IMarshal:GetUnmarshalClass failed, %lx\n",hres);
527 goto release_marshal;
529 hres = IStream_Write(pStm,&xclsid,sizeof(xclsid),&writeres);
531 FIXME("Stream write failed, %lx\n",hres);
532 goto release_marshal;
535 TRACE("Calling IMarshal::MarshalInterace\n");
536 hres = IMarshal_MarshalInterface(pMarshal,pStm,riid,pUnk,dwDestContext,pvDestContext,mshlflags);
539 if (IsEqualGUID(riid,&IID_IOleObject)) {
540 ERR("WINE currently cannot marshal IOleObject interfaces. This means you cannot embed/link OLE objects between applications.\n");
542 FIXME("Failed to marshal the interface %s, %lx?\n",debugstr_guid(riid),hres);
546 IMarshal_Release(pMarshal);
551 /***********************************************************************
552 * CoUnmarshalInterface [OLE32.@]
555 CoUnmarshalInterface(IStream *pStm, REFIID riid, LPVOID *ppv) {
558 wine_marshal_data md;
564 TRACE("(%p,%s,%p)\n",pStm,debugstr_guid(riid),ppv);
566 hres = IStream_Read(pStm,&mid,sizeof(mid),&res);
568 FIXME("Stream read 1 failed, %lx, (%ld of %d)\n",hres,res,sizeof(mid));
571 hres = IStream_Read(pStm,&md,sizeof(md),&res);
573 FIXME("Stream read 2 failed, %lx, (%ld of %d)\n",hres,res,sizeof(md));
576 hres = IStream_Read(pStm,&xclsid,sizeof(xclsid),&res);
578 FIXME("Stream read 3 failed, %lx, (%ld of %d)\n",hres,res,sizeof(xclsid));
581 hres=CoCreateInstance(&xclsid,NULL,CLSCTX_INPROC_SERVER | CLSCTX_INPROC_HANDLER | CLSCTX_LOCAL_SERVER,&IID_IMarshal,(void**)&pUnk);
583 FIXME("Failed to create instance of unmarshaller %s.\n",debugstr_guid(&xclsid));
586 hres = _GetMarshaller(riid,pUnk,md.dwDestContext,NULL,md.mshlflags,&pMarshal);
588 FIXME("Failed to get unmarshaller, %lx?\n",hres);
591 hres = IMarshal_UnmarshalInterface(pMarshal,pStm,riid,ppv);
593 FIXME("Failed to Unmarshal the interface, %lx?\n",hres);
594 goto release_marshal;
597 IMarshal_Release(pMarshal);
601 /***********************************************************************
602 * CoReleaseMarshalData [OLE32.@]
605 CoReleaseMarshalData(IStream *pStm) {
608 wine_marshal_data md;
614 TRACE("(%p)\n",pStm);
616 hres = IStream_Read(pStm,&mid,sizeof(mid),&res);
618 FIXME("Stream read 1 failed, %lx, (%ld of %d)\n",hres,res,sizeof(mid));
621 hres = IStream_Read(pStm,&md,sizeof(md),&res);
623 FIXME("Stream read 2 failed, %lx, (%ld of %d)\n",hres,res,sizeof(md));
626 hres = IStream_Read(pStm,&xclsid,sizeof(xclsid),&res);
628 FIXME("Stream read 3 failed, %lx, (%ld of %d)\n",hres,res,sizeof(xclsid));
631 hres=CoCreateInstance(&xclsid,NULL,CLSCTX_INPROC_SERVER | CLSCTX_INPROC_HANDLER | CLSCTX_LOCAL_SERVER,&IID_IMarshal,(void**)(char*)&pUnk);
633 FIXME("Failed to create instance of unmarshaller %s.\n",debugstr_guid(&xclsid));
636 hres = IUnknown_QueryInterface(pUnk,&IID_IMarshal,(LPVOID*)(char*)&pMarshal);
638 FIXME("Failed to get IMarshal iface, %lx?\n",hres);
641 hres = IMarshal_ReleaseMarshalData(pMarshal,pStm);
643 FIXME("Failed to releasemarshaldata the interface, %lx?\n",hres);
645 IMarshal_Release(pMarshal);
646 IUnknown_Release(pUnk);
651 /***********************************************************************
652 * CoMarshalInterThreadInterfaceInStream [OLE32.@]
654 * Marshal an interface across threads in the same process.
657 * riid [I] Identifier of the interface to be marshalled.
658 * pUnk [I] Pointer to IUnknown-derived interface that will be marshalled.
659 * ppStm [O] Pointer to IStream object that is created and then used to store the marshalled inteface.
663 * Failure: E_OUTOFMEMORY and other COM error codes
666 * CoMarshalInterface(), CoUnmarshalInterface() and CoGetInterfaceAndReleaseStream()
669 CoMarshalInterThreadInterfaceInStream(
670 REFIID riid, LPUNKNOWN pUnk, LPSTREAM * ppStm)
673 LARGE_INTEGER seekto;
676 TRACE("(%s, %p, %p)\n",debugstr_guid(riid), pUnk, ppStm);
678 hres = CreateStreamOnHGlobal(0, TRUE, ppStm);
679 if (FAILED(hres)) return hres;
680 hres = CoMarshalInterface(*ppStm, riid, pUnk, MSHCTX_INPROC, NULL, MSHLFLAGS_NORMAL);
682 /* FIXME: is this needed? */
683 memset(&seekto,0,sizeof(seekto));
684 IStream_Seek(*ppStm,seekto,SEEK_SET,&xpos);
689 /***********************************************************************
690 * CoGetInterfaceAndReleaseStream [OLE32.@]
692 * Unmarshalls an inteface from a stream and then releases the stream.
695 * pStm [I] Stream that contains the marshalled inteface.
696 * riid [I] Interface identifier of the object to unmarshall.
697 * ppv [O] Address of pointer where the requested interface object will be stored.
701 * Failure: A COM error code
704 * CoMarshalInterThreadInterfaceInStream() and CoUnmarshalInteface()
707 CoGetInterfaceAndReleaseStream(LPSTREAM pStm,REFIID riid, LPVOID *ppv)
711 TRACE("(%p, %s, %p)\n", pStm, debugstr_guid(riid), ppv);
713 hres = CoUnmarshalInterface(pStm, riid, ppv);
714 IStream_Release(pStm);
718 static HRESULT WINAPI
719 SMCF_QueryInterface(LPCLASSFACTORY iface,REFIID riid, LPVOID *ppv) {
721 if (IsEqualIID(riid,&IID_IUnknown) || IsEqualIID(riid,&IID_IClassFactory)) {
722 *ppv = (LPVOID)iface;
725 return E_NOINTERFACE;
727 static ULONG WINAPI SMCF_AddRef(LPCLASSFACTORY iface) { return 2; }
728 static ULONG WINAPI SMCF_Release(LPCLASSFACTORY iface) { return 1; }
730 static HRESULT WINAPI
732 LPCLASSFACTORY iface, LPUNKNOWN pUnk, REFIID riid, LPVOID *ppv
734 if (IsEqualIID(riid,&IID_IMarshal)) {
736 dm=(StdMarshalImpl*)HeapAlloc(GetProcessHeap(),0,sizeof(StdMarshalImpl));
739 dm->lpvtbl = &stdmvtbl;
744 FIXME("(%s), not supported.\n",debugstr_guid(riid));
745 return E_NOINTERFACE;
748 static HRESULT WINAPI
749 SMCF_LockServer(LPCLASSFACTORY iface, BOOL fLock) {
750 FIXME("(%d), stub!\n",fLock);
754 static ICOM_VTABLE(IClassFactory) dfmarshalcfvtbl = {
755 ICOM_MSVTABLE_COMPAT_DummyRTTIVALUE
762 static ICOM_VTABLE(IClassFactory) *pdfmarshalcfvtbl = &dfmarshalcfvtbl;
765 MARSHAL_GetStandardMarshalCF(LPVOID *ppv) {
766 *ppv = &pdfmarshalcfvtbl;