atl: Add support for the registrar parameter of AtlModuleUpdateRegistryFromResourceD.
[wine] / dlls / rpcrt4 / ndr_ole.c
1 /*
2  * OLE32 callouts, COM interface marshalling
3  *
4  * Copyright 2001 Ove Kåven, TransGaming Technologies
5  *
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.
10  *
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.
15  *
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
19  *
20  * TODO:
21  *  - fix the wire-protocol to match MS/RPC
22  *  - finish RpcStream_Vtbl
23  */
24
25 #include <stdarg.h>
26 #include <stdio.h>
27 #include <string.h>
28
29 #define COBJMACROS
30 #define NONAMELESSUNION
31 #define NONAMELESSSTRUCT
32
33 #include "windef.h"
34 #include "winbase.h"
35 #include "winerror.h"
36 #include "winreg.h"
37
38 #include "objbase.h"
39
40 #include "ndr_misc.h"
41 #include "rpcndr.h"
42 #include "rpcproxy.h"
43 #include "wine/rpcfc.h"
44 #include "cpsf.h"
45
46 #include "wine/debug.h"
47
48 WINE_DEFAULT_DEBUG_CHANNEL(ole);
49
50 static HMODULE hOLE;
51
52 static HRESULT (WINAPI *COM_GetMarshalSizeMax)(ULONG *,REFIID,LPUNKNOWN,DWORD,LPVOID,DWORD);
53 static HRESULT (WINAPI *COM_MarshalInterface)(LPSTREAM,REFIID,LPUNKNOWN,DWORD,LPVOID,DWORD);
54 static HRESULT (WINAPI *COM_UnmarshalInterface)(LPSTREAM,REFIID,LPVOID*);
55 static HRESULT (WINAPI *COM_ReleaseMarshalData)(LPSTREAM);
56 static HRESULT (WINAPI *COM_GetClassObject)(REFCLSID,DWORD,COSERVERINFO *,REFIID,LPVOID *);
57 static HRESULT (WINAPI *COM_GetPSClsid)(REFIID,CLSID *);
58 static LPVOID (WINAPI *COM_MemAlloc)(ULONG);
59 static void (WINAPI *COM_MemFree)(LPVOID);
60
61 static HMODULE LoadCOM(void)
62 {
63   if (hOLE) return hOLE;
64   hOLE = LoadLibraryA("OLE32.DLL");
65   if (!hOLE) return 0;
66   COM_GetMarshalSizeMax  = (LPVOID)GetProcAddress(hOLE, "CoGetMarshalSizeMax");
67   COM_MarshalInterface   = (LPVOID)GetProcAddress(hOLE, "CoMarshalInterface");
68   COM_UnmarshalInterface = (LPVOID)GetProcAddress(hOLE, "CoUnmarshalInterface");
69   COM_ReleaseMarshalData = (LPVOID)GetProcAddress(hOLE, "CoReleaseMarshalData");
70   COM_GetClassObject     = (LPVOID)GetProcAddress(hOLE, "CoGetClassObject");
71   COM_GetPSClsid         = (LPVOID)GetProcAddress(hOLE, "CoGetPSClsid");
72   COM_MemAlloc = (LPVOID)GetProcAddress(hOLE, "CoTaskMemAlloc");
73   COM_MemFree  = (LPVOID)GetProcAddress(hOLE, "CoTaskMemFree");
74   return hOLE;
75 }
76
77 /* CoMarshalInterface/CoUnmarshalInterface works on streams,
78  * so implement a simple stream on top of the RPC buffer
79  * (which also implements the MInterfacePointer structure) */
80 typedef struct RpcStreamImpl
81 {
82   const IStreamVtbl *lpVtbl;
83   DWORD RefCount;
84   PMIDL_STUB_MESSAGE pMsg;
85   LPDWORD size;
86   char *data;
87   DWORD pos;
88 } RpcStreamImpl;
89
90 static HRESULT WINAPI RpcStream_QueryInterface(LPSTREAM iface,
91                                               REFIID riid,
92                                               LPVOID *obj)
93 {
94   RpcStreamImpl *This = (RpcStreamImpl *)iface;
95   if (IsEqualGUID(&IID_IUnknown, riid) ||
96       IsEqualGUID(&IID_ISequentialStream, riid) ||
97       IsEqualGUID(&IID_IStream, riid)) {
98     *obj = This;
99     This->RefCount++;
100     return S_OK;
101   }
102   return E_NOINTERFACE;
103 }
104
105 static ULONG WINAPI RpcStream_AddRef(LPSTREAM iface)
106 {
107   RpcStreamImpl *This = (RpcStreamImpl *)iface;
108   return ++(This->RefCount);
109 }
110
111 static ULONG WINAPI RpcStream_Release(LPSTREAM iface)
112 {
113   RpcStreamImpl *This = (RpcStreamImpl *)iface;
114   if (!--(This->RefCount)) {
115     TRACE("size=%d\n", *This->size);
116     This->pMsg->Buffer = (unsigned char*)This->data + *This->size;
117     HeapFree(GetProcessHeap(),0,This);
118     return 0;
119   }
120   return This->RefCount;
121 }
122
123 static HRESULT WINAPI RpcStream_Read(LPSTREAM iface,
124                                     void *pv,
125                                     ULONG cb,
126                                     ULONG *pcbRead)
127 {
128   RpcStreamImpl *This = (RpcStreamImpl *)iface;
129   HRESULT hr = S_OK;
130   if (This->pos + cb > *This->size)
131   {
132     cb = *This->size - This->pos;
133     hr = S_FALSE;
134   }
135   if (cb) {
136     memcpy(pv, This->data + This->pos, cb);
137     This->pos += cb;
138   }
139   if (pcbRead) *pcbRead = cb;
140   return hr;
141 }
142
143 static HRESULT WINAPI RpcStream_Write(LPSTREAM iface,
144                                      const void *pv,
145                                      ULONG cb,
146                                      ULONG *pcbWritten)
147 {
148   RpcStreamImpl *This = (RpcStreamImpl *)iface;
149   if (This->data + cb > (char *)This->pMsg->BufferEnd)
150     return STG_E_MEDIUMFULL;
151   memcpy(This->data + This->pos, pv, cb);
152   This->pos += cb;
153   if (This->pos > *This->size) *This->size = This->pos;
154   if (pcbWritten) *pcbWritten = cb;
155   return S_OK;
156 }
157
158 static HRESULT WINAPI RpcStream_Seek(LPSTREAM iface,
159                                     LARGE_INTEGER move,
160                                     DWORD origin,
161                                     ULARGE_INTEGER *newPos)
162 {
163   RpcStreamImpl *This = (RpcStreamImpl *)iface;
164   switch (origin) {
165   case STREAM_SEEK_SET:
166     This->pos = move.u.LowPart;
167     break;
168   case STREAM_SEEK_CUR:
169     This->pos = This->pos + move.u.LowPart;
170     break;
171   case STREAM_SEEK_END:
172     This->pos = *This->size + move.u.LowPart;
173     break;
174   default:
175     return STG_E_INVALIDFUNCTION;
176   }
177   if (newPos) {
178     newPos->u.LowPart = This->pos;
179     newPos->u.HighPart = 0;
180   }
181   return S_OK;
182 }
183
184 static HRESULT WINAPI RpcStream_SetSize(LPSTREAM iface,
185                                        ULARGE_INTEGER newSize)
186 {
187   RpcStreamImpl *This = (RpcStreamImpl *)iface;
188   *This->size = newSize.u.LowPart;
189   return S_OK;
190 }
191
192 static const IStreamVtbl RpcStream_Vtbl =
193 {
194   RpcStream_QueryInterface,
195   RpcStream_AddRef,
196   RpcStream_Release,
197   RpcStream_Read,
198   RpcStream_Write,
199   RpcStream_Seek,
200   RpcStream_SetSize,
201   NULL, /* CopyTo */
202   NULL, /* Commit */
203   NULL, /* Revert */
204   NULL, /* LockRegion */
205   NULL, /* UnlockRegion */
206   NULL, /* Stat */
207   NULL  /* Clone */
208 };
209
210 static LPSTREAM RpcStream_Create(PMIDL_STUB_MESSAGE pStubMsg, BOOL init)
211 {
212   RpcStreamImpl *This;
213   This = HeapAlloc(GetProcessHeap(),HEAP_ZERO_MEMORY,sizeof(RpcStreamImpl));
214   if (!This) return NULL;
215   This->lpVtbl = &RpcStream_Vtbl;
216   This->RefCount = 1;
217   This->pMsg = pStubMsg;
218   This->size = (LPDWORD)pStubMsg->Buffer;
219   This->data = (char*)(This->size + 1);
220   This->pos = 0;
221   if (init) *This->size = 0;
222   TRACE("init size=%d\n", *This->size);
223   return (LPSTREAM)This;
224 }
225
226 static const IID* get_ip_iid(PMIDL_STUB_MESSAGE pStubMsg, unsigned char *pMemory, PFORMAT_STRING pFormat)
227 {
228   const IID *riid;
229   if (!pFormat) return &IID_IUnknown;
230   TRACE("format=%02x %02x\n", pFormat[0], pFormat[1]);
231   if (pFormat[0] != RPC_FC_IP) FIXME("format=%d\n", pFormat[0]);
232   if (pFormat[1] == RPC_FC_CONSTANT_IID) {
233     riid = (const IID *)&pFormat[2];
234   } else {
235     ComputeConformance(pStubMsg, pMemory, pFormat+2, 0);
236     riid = (const IID *)pStubMsg->MaxCount;
237   }
238   if (!riid) riid = &IID_IUnknown;
239   TRACE("got %s\n", debugstr_guid(riid));
240   return riid;
241 }
242
243 /***********************************************************************
244  *           NdrInterfacePointerMarshall [RPCRT4.@]
245  */
246 unsigned char * WINAPI NdrInterfacePointerMarshall(PMIDL_STUB_MESSAGE pStubMsg,
247                                                   unsigned char *pMemory,
248                                                   PFORMAT_STRING pFormat)
249 {
250   const IID *riid = get_ip_iid(pStubMsg, pMemory, pFormat);
251   LPSTREAM stream;
252   HRESULT hr;
253
254   TRACE("(%p,%p,%p)\n", pStubMsg, pMemory, pFormat);
255   pStubMsg->MaxCount = 0;
256   if (!LoadCOM()) return NULL;
257   if (pStubMsg->Buffer + sizeof(DWORD) <= (unsigned char *)pStubMsg->RpcMsg->Buffer + pStubMsg->BufferLength) {
258     stream = RpcStream_Create(pStubMsg, TRUE);
259     if (stream) {
260       if (pMemory)
261         hr = COM_MarshalInterface(stream, riid, (LPUNKNOWN)pMemory,
262                                   pStubMsg->dwDestContext, pStubMsg->pvDestContext,
263                                   MSHLFLAGS_NORMAL);
264       else
265         hr = S_OK;
266
267       IStream_Release(stream);
268       if (FAILED(hr))
269         RpcRaiseException(hr);
270     }
271   }
272   return NULL;
273 }
274
275 /***********************************************************************
276  *           NdrInterfacePointerUnmarshall [RPCRT4.@]
277  */
278 unsigned char * WINAPI NdrInterfacePointerUnmarshall(PMIDL_STUB_MESSAGE pStubMsg,
279                                                     unsigned char **ppMemory,
280                                                     PFORMAT_STRING pFormat,
281                                                     unsigned char fMustAlloc)
282 {
283   LPSTREAM stream;
284   HRESULT hr;
285
286   TRACE("(%p,%p,%p,%d)\n", pStubMsg, ppMemory, pFormat, fMustAlloc);
287   if (!LoadCOM()) return NULL;
288   *(LPVOID*)ppMemory = NULL;
289   if (pStubMsg->Buffer + sizeof(DWORD) < (unsigned char *)pStubMsg->RpcMsg->Buffer + pStubMsg->BufferLength) {
290     stream = RpcStream_Create(pStubMsg, FALSE);
291     if (!stream) RpcRaiseException(E_OUTOFMEMORY);
292     if (*((RpcStreamImpl *)stream)->size != 0)
293       hr = COM_UnmarshalInterface(stream, &IID_NULL, (LPVOID*)ppMemory);
294     else
295       hr = S_OK;
296     IStream_Release(stream);
297     if (FAILED(hr))
298         RpcRaiseException(hr);
299   }
300   return NULL;
301 }
302
303 /***********************************************************************
304  *           NdrInterfacePointerBufferSize [RPCRT4.@]
305  */
306 void WINAPI NdrInterfacePointerBufferSize(PMIDL_STUB_MESSAGE pStubMsg,
307                                          unsigned char *pMemory,
308                                          PFORMAT_STRING pFormat)
309 {
310   const IID *riid = get_ip_iid(pStubMsg, pMemory, pFormat);
311   ULONG size = 0;
312   HRESULT hr;
313
314   TRACE("(%p,%p,%p)\n", pStubMsg, pMemory, pFormat);
315   if (!LoadCOM()) return;
316   hr = COM_GetMarshalSizeMax(&size, riid, (LPUNKNOWN)pMemory,
317                             pStubMsg->dwDestContext, pStubMsg->pvDestContext,
318                             MSHLFLAGS_NORMAL);
319   TRACE("size=%d\n", size);
320   pStubMsg->BufferLength += sizeof(DWORD) + size;
321 }
322
323 /***********************************************************************
324  *           NdrInterfacePointerMemorySize [RPCRT4.@]
325  */
326 ULONG WINAPI NdrInterfacePointerMemorySize(PMIDL_STUB_MESSAGE pStubMsg,
327                                            PFORMAT_STRING pFormat)
328 {
329   ULONG size;
330
331   TRACE("(%p,%p)\n", pStubMsg, pFormat);
332
333   size = *(ULONG *)pStubMsg->Buffer;
334   pStubMsg->Buffer += 4;
335   pStubMsg->MemorySize += 4;
336
337   pStubMsg->Buffer += size;
338
339   return pStubMsg->MemorySize;
340 }
341
342 /***********************************************************************
343  *           NdrInterfacePointerFree [RPCRT4.@]
344  */
345 void WINAPI NdrInterfacePointerFree(PMIDL_STUB_MESSAGE pStubMsg,
346                                    unsigned char *pMemory,
347                                    PFORMAT_STRING pFormat)
348 {
349   LPUNKNOWN pUnk = (LPUNKNOWN)pMemory;
350   TRACE("(%p,%p,%p)\n", pStubMsg, pMemory, pFormat);
351   if (pUnk) IUnknown_Release(pUnk);
352 }
353
354 /***********************************************************************
355  *           NdrOleAllocate [RPCRT4.@]
356  */
357 void * WINAPI NdrOleAllocate(size_t Size)
358 {
359   if (!LoadCOM()) return NULL;
360   return COM_MemAlloc(Size);
361 }
362
363 /***********************************************************************
364  *           NdrOleFree [RPCRT4.@]
365  */
366 void WINAPI NdrOleFree(void *NodeToFree)
367 {
368   if (!LoadCOM()) return;
369   COM_MemFree(NodeToFree);
370 }
371
372 /***********************************************************************
373  * Helper function to create a stub.
374  * This probably looks very much like NdrpCreateStub.
375  */
376 HRESULT create_stub(REFIID iid, IUnknown *pUnk, IRpcStubBuffer **ppstub)
377 {
378     CLSID clsid;
379     IPSFactoryBuffer *psfac;
380     HRESULT r;
381
382     if(!LoadCOM()) return E_FAIL;
383
384     r = COM_GetPSClsid( iid, &clsid );
385     if(FAILED(r)) return r;
386
387     r = COM_GetClassObject( &clsid, CLSCTX_INPROC_SERVER, NULL, &IID_IPSFactoryBuffer, (void**)&psfac );
388     if(FAILED(r)) return r;
389
390     r = IPSFactoryBuffer_CreateStub(psfac, iid, pUnk, ppstub);
391
392     IPSFactoryBuffer_Release(psfac);
393     return r;
394 }