2 * OLE32 callouts, COM interface marshalling
4 * Copyright 2001 Ove Kåven, TransGaming Technologies
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., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
21 * - fix the wire-protocol to match MS/RPC
22 * - finish RpcStream_Vtbl
30 #define NONAMELESSUNION
31 #define NONAMELESSSTRUCT
42 #include "wine/rpcfc.h"
44 #include "wine/debug.h"
46 WINE_DEFAULT_DEBUG_CHANNEL(ole);
50 static HRESULT (WINAPI *COM_GetMarshalSizeMax)(ULONG *,REFIID,LPUNKNOWN,DWORD,LPVOID,DWORD);
51 static HRESULT (WINAPI *COM_MarshalInterface)(LPSTREAM,REFIID,LPUNKNOWN,DWORD,LPVOID,DWORD);
52 static HRESULT (WINAPI *COM_UnmarshalInterface)(LPSTREAM,REFIID,LPVOID*);
53 static HRESULT (WINAPI *COM_ReleaseMarshalData)(LPSTREAM);
54 static HRESULT (WINAPI *COM_GetClassObject)(REFCLSID,DWORD,COSERVERINFO *,REFIID,LPVOID *);
55 static HRESULT (WINAPI *COM_GetPSClsid)(REFIID,CLSID *);
56 static LPVOID (WINAPI *COM_MemAlloc)(ULONG);
57 static void (WINAPI *COM_MemFree)(LPVOID);
59 static HMODULE LoadCOM(void)
61 if (hOLE) return hOLE;
62 hOLE = LoadLibraryA("OLE32.DLL");
64 COM_GetMarshalSizeMax = (LPVOID)GetProcAddress(hOLE, "CoGetMarshalSizeMax");
65 COM_MarshalInterface = (LPVOID)GetProcAddress(hOLE, "CoMarshalInterface");
66 COM_UnmarshalInterface = (LPVOID)GetProcAddress(hOLE, "CoUnmarshalInterface");
67 COM_ReleaseMarshalData = (LPVOID)GetProcAddress(hOLE, "CoReleaseMarshalData");
68 COM_GetClassObject = (LPVOID)GetProcAddress(hOLE, "CoGetClassObject");
69 COM_GetPSClsid = (LPVOID)GetProcAddress(hOLE, "CoGetPSClsid");
70 COM_MemAlloc = (LPVOID)GetProcAddress(hOLE, "CoTaskMemAlloc");
71 COM_MemFree = (LPVOID)GetProcAddress(hOLE, "CoTaskMemFree");
75 /* CoMarshalInterface/CoUnmarshalInterface works on streams,
76 * so implement a simple stream on top of the RPC buffer
77 * (which also implements the MInterfacePointer structure) */
78 typedef struct RpcStreamImpl
80 const IStreamVtbl *lpVtbl;
82 PMIDL_STUB_MESSAGE pMsg;
88 static HRESULT WINAPI RpcStream_QueryInterface(LPSTREAM iface,
92 RpcStreamImpl *This = (RpcStreamImpl *)iface;
93 if (IsEqualGUID(&IID_IUnknown, riid) ||
94 IsEqualGUID(&IID_ISequentialStream, riid) ||
95 IsEqualGUID(&IID_IStream, riid)) {
100 return E_NOINTERFACE;
103 static ULONG WINAPI RpcStream_AddRef(LPSTREAM iface)
105 RpcStreamImpl *This = (RpcStreamImpl *)iface;
106 return ++(This->RefCount);
109 static ULONG WINAPI RpcStream_Release(LPSTREAM iface)
111 RpcStreamImpl *This = (RpcStreamImpl *)iface;
112 if (!--(This->RefCount)) {
113 TRACE("size=%ld\n", *This->size);
114 This->pMsg->Buffer = (unsigned char*)This->data + *This->size;
115 HeapFree(GetProcessHeap(),0,This);
118 return This->RefCount;
121 static HRESULT WINAPI RpcStream_Read(LPSTREAM iface,
126 RpcStreamImpl *This = (RpcStreamImpl *)iface;
128 if (This->pos + cb > *This->size)
130 cb = *This->size - This->pos;
134 memcpy(pv, This->data + This->pos, cb);
137 if (pcbRead) *pcbRead = cb;
141 static HRESULT WINAPI RpcStream_Write(LPSTREAM iface,
146 RpcStreamImpl *This = (RpcStreamImpl *)iface;
147 if (This->data + cb > (char *)This->pMsg->BufferEnd)
148 return STG_E_MEDIUMFULL;
149 memcpy(This->data + This->pos, pv, cb);
151 if (This->pos > *This->size) *This->size = This->pos;
152 if (pcbWritten) *pcbWritten = cb;
156 static HRESULT WINAPI RpcStream_Seek(LPSTREAM iface,
159 ULARGE_INTEGER *newPos)
161 RpcStreamImpl *This = (RpcStreamImpl *)iface;
163 case STREAM_SEEK_SET:
164 This->pos = move.u.LowPart;
166 case STREAM_SEEK_CUR:
167 This->pos = This->pos + move.u.LowPart;
169 case STREAM_SEEK_END:
170 This->pos = *This->size + move.u.LowPart;
173 return STG_E_INVALIDFUNCTION;
176 newPos->u.LowPart = This->pos;
177 newPos->u.HighPart = 0;
182 static HRESULT WINAPI RpcStream_SetSize(LPSTREAM iface,
183 ULARGE_INTEGER newSize)
185 RpcStreamImpl *This = (RpcStreamImpl *)iface;
186 *This->size = newSize.u.LowPart;
190 static const IStreamVtbl RpcStream_Vtbl =
192 RpcStream_QueryInterface,
202 NULL, /* LockRegion */
203 NULL, /* UnlockRegion */
208 static LPSTREAM RpcStream_Create(PMIDL_STUB_MESSAGE pStubMsg, BOOL init)
211 This = HeapAlloc(GetProcessHeap(),HEAP_ZERO_MEMORY,sizeof(RpcStreamImpl));
212 if (!This) return NULL;
213 This->lpVtbl = &RpcStream_Vtbl;
215 This->pMsg = pStubMsg;
216 This->size = (LPDWORD)pStubMsg->Buffer;
217 This->data = (char*)(This->size + 1);
219 if (init) *This->size = 0;
220 TRACE("init size=%ld\n", *This->size);
221 return (LPSTREAM)This;
224 static const IID* get_ip_iid(PMIDL_STUB_MESSAGE pStubMsg, unsigned char *pMemory, PFORMAT_STRING pFormat)
227 if (!pFormat) return &IID_IUnknown;
228 TRACE("format=%02x %02x\n", pFormat[0], pFormat[1]);
229 if (pFormat[0] != RPC_FC_IP) FIXME("format=%d\n", pFormat[0]);
230 if (pFormat[1] == RPC_FC_CONSTANT_IID) {
231 riid = (const IID *)&pFormat[2];
233 ComputeConformance(pStubMsg, pMemory, pFormat+2, 0);
234 riid = (const IID *)pStubMsg->MaxCount;
236 if (!riid) riid = &IID_IUnknown;
237 TRACE("got %s\n", debugstr_guid(riid));
241 /***********************************************************************
242 * NdrInterfacePointerMarshall [RPCRT4.@]
244 unsigned char * WINAPI NdrInterfacePointerMarshall(PMIDL_STUB_MESSAGE pStubMsg,
245 unsigned char *pMemory,
246 PFORMAT_STRING pFormat)
248 const IID *riid = get_ip_iid(pStubMsg, pMemory, pFormat);
252 TRACE("(%p,%p,%p)\n", pStubMsg, pMemory, pFormat);
253 pStubMsg->MaxCount = 0;
254 if (!LoadCOM()) return NULL;
255 if (pStubMsg->Buffer + sizeof(DWORD) <= (unsigned char *)pStubMsg->RpcMsg->Buffer + pStubMsg->BufferLength) {
256 stream = RpcStream_Create(pStubMsg, TRUE);
259 hr = COM_MarshalInterface(stream, riid, (LPUNKNOWN)pMemory,
260 pStubMsg->dwDestContext, pStubMsg->pvDestContext,
265 IStream_Release(stream);
267 RpcRaiseException(hr);
273 /***********************************************************************
274 * NdrInterfacePointerUnmarshall [RPCRT4.@]
276 unsigned char * WINAPI NdrInterfacePointerUnmarshall(PMIDL_STUB_MESSAGE pStubMsg,
277 unsigned char **ppMemory,
278 PFORMAT_STRING pFormat,
279 unsigned char fMustAlloc)
284 TRACE("(%p,%p,%p,%d)\n", pStubMsg, ppMemory, pFormat, fMustAlloc);
285 if (!LoadCOM()) return NULL;
286 *(LPVOID*)ppMemory = NULL;
287 if (pStubMsg->Buffer + sizeof(DWORD) < (unsigned char *)pStubMsg->RpcMsg->Buffer + pStubMsg->BufferLength) {
288 stream = RpcStream_Create(pStubMsg, FALSE);
290 hr = COM_UnmarshalInterface(stream, &IID_NULL, (LPVOID*)ppMemory);
291 IStream_Release(stream);
293 RpcRaiseException(hr);
299 /***********************************************************************
300 * NdrInterfacePointerBufferSize [RPCRT4.@]
302 void WINAPI NdrInterfacePointerBufferSize(PMIDL_STUB_MESSAGE pStubMsg,
303 unsigned char *pMemory,
304 PFORMAT_STRING pFormat)
306 const IID *riid = get_ip_iid(pStubMsg, pMemory, pFormat);
310 TRACE("(%p,%p,%p)\n", pStubMsg, pMemory, pFormat);
311 if (!LoadCOM()) return;
312 hr = COM_GetMarshalSizeMax(&size, riid, (LPUNKNOWN)pMemory,
313 pStubMsg->dwDestContext, pStubMsg->pvDestContext,
315 TRACE("size=%ld\n", size);
316 pStubMsg->BufferLength += sizeof(DWORD) + size;
319 /***********************************************************************
320 * NdrInterfacePointerMemorySize [RPCRT4.@]
322 unsigned long WINAPI NdrInterfacePointerMemorySize(PMIDL_STUB_MESSAGE pStubMsg,
323 PFORMAT_STRING pFormat)
327 TRACE("(%p,%p)\n", pStubMsg, pFormat);
329 size = *(ULONG *)pStubMsg->Buffer;
330 pStubMsg->Buffer += 4;
331 pStubMsg->MemorySize += 4;
333 pStubMsg->Buffer += size;
335 return pStubMsg->MemorySize;
338 /***********************************************************************
339 * NdrInterfacePointerFree [RPCRT4.@]
341 void WINAPI NdrInterfacePointerFree(PMIDL_STUB_MESSAGE pStubMsg,
342 unsigned char *pMemory,
343 PFORMAT_STRING pFormat)
345 LPUNKNOWN pUnk = (LPUNKNOWN)pMemory;
346 TRACE("(%p,%p,%p)\n", pStubMsg, pMemory, pFormat);
347 if (pUnk) IUnknown_Release(pUnk);
350 /***********************************************************************
351 * NdrOleAllocate [RPCRT4.@]
353 void * WINAPI NdrOleAllocate(size_t Size)
355 if (!LoadCOM()) return NULL;
356 return COM_MemAlloc(Size);
359 /***********************************************************************
360 * NdrOleFree [RPCRT4.@]
362 void WINAPI NdrOleFree(void *NodeToFree)
364 if (!LoadCOM()) return;
365 COM_MemFree(NodeToFree);