| 1 | /*
|
|---|
| 2 | * Marshaling library
|
|---|
| 3 | *
|
|---|
| 4 | * Copyright 2002 Marcus Meissner
|
|---|
| 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., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
|
|---|
| 19 | */
|
|---|
| 20 |
|
|---|
| 21 | #include "config.h"
|
|---|
| 22 |
|
|---|
| 23 | #include <stdlib.h>
|
|---|
| 24 | #include <stdio.h>
|
|---|
| 25 | #include <string.h>
|
|---|
| 26 | #include <assert.h>
|
|---|
| 27 |
|
|---|
| 28 | #include "windef.h"
|
|---|
| 29 | #include "objbase.h"
|
|---|
| 30 | #include "ole2.h"
|
|---|
| 31 | #include "ole2ver.h"
|
|---|
| 32 | #include "rpc.h"
|
|---|
| 33 | #include "winerror.h"
|
|---|
| 34 | #include "winreg.h"
|
|---|
| 35 | #include "wownt32.h"
|
|---|
| 36 | #include "wtypes.h"
|
|---|
| 37 | #include "wine/unicode.h"
|
|---|
| 38 | #include "wine/obj_base.h"
|
|---|
| 39 | #include "wine/obj_clientserver.h"
|
|---|
| 40 | #include "wine/obj_misc.h"
|
|---|
| 41 | #include "wine/obj_marshal.h"
|
|---|
| 42 | #include "wine/obj_storage.h"
|
|---|
| 43 | #include "wine/obj_channel.h"
|
|---|
| 44 | #include "wine/winbase16.h"
|
|---|
| 45 | #include "compobj_private.h"
|
|---|
| 46 | #include "ifs.h"
|
|---|
| 47 |
|
|---|
| 48 | #include "wine/debug.h"
|
|---|
| 49 |
|
|---|
| 50 | WINE_DEFAULT_DEBUG_CHANNEL(ole);
|
|---|
| 51 |
|
|---|
| 52 | /* Marshaling just passes a unique identifier to the remote client,
|
|---|
| 53 | * that makes it possible to find the passed interface again.
|
|---|
| 54 | *
|
|---|
| 55 | * So basically we need a set of values that make it unique.
|
|---|
| 56 | *
|
|---|
| 57 | * Process Identifier, Object IUnknown ptr, IID
|
|---|
| 58 | *
|
|---|
| 59 | * Note that the IUnknown_QI(ob,xiid,&ppv) always returns the SAME ppv value!
|
|---|
| 60 | */
|
|---|
| 61 |
|
|---|
| 62 | typedef struct _mid2unknown {
|
|---|
| 63 | wine_marshal_id mid;
|
|---|
| 64 | LPUNKNOWN pUnk;
|
|---|
| 65 | } mid2unknown;
|
|---|
| 66 |
|
|---|
| 67 | typedef struct _mid2stub {
|
|---|
| 68 | wine_marshal_id mid;
|
|---|
| 69 | IRpcStubBuffer *stub;
|
|---|
| 70 | LPUNKNOWN pUnkServer;
|
|---|
| 71 | } mid2stub;
|
|---|
| 72 |
|
|---|
| 73 | static mid2stub *stubs = NULL;
|
|---|
| 74 | static int nrofstubs = 0;
|
|---|
| 75 |
|
|---|
| 76 | static mid2unknown *proxies = NULL;
|
|---|
| 77 | static int nrofproxies = 0;
|
|---|
| 78 |
|
|---|
| 79 | HRESULT
|
|---|
| 80 | MARSHAL_Find_Stub_Server(wine_marshal_id *mid,LPUNKNOWN *punk) {
|
|---|
| 81 | int i;
|
|---|
| 82 |
|
|---|
| 83 | for (i=0;i<nrofstubs;i++) {
|
|---|
| 84 | if (MARSHAL_Compare_Mids_NoInterface(mid,&(stubs[i].mid))) {
|
|---|
| 85 | *punk = stubs[i].pUnkServer;
|
|---|
| 86 | IUnknown_AddRef((*punk));
|
|---|
| 87 | return S_OK;
|
|---|
| 88 | }
|
|---|
| 89 | }
|
|---|
| 90 | return E_FAIL;
|
|---|
| 91 | }
|
|---|
| 92 |
|
|---|
| 93 | HRESULT
|
|---|
| 94 | MARSHAL_Find_Stub_Buffer(wine_marshal_id *mid,IRpcStubBuffer **stub) {
|
|---|
| 95 | int i;
|
|---|
| 96 |
|
|---|
| 97 | for (i=0;i<nrofstubs;i++) {
|
|---|
| 98 | if (MARSHAL_Compare_Mids(mid,&(stubs[i].mid))) {
|
|---|
| 99 | *stub = stubs[i].stub;
|
|---|
| 100 | IUnknown_AddRef((*stub));
|
|---|
| 101 | return S_OK;
|
|---|
| 102 | }
|
|---|
| 103 | }
|
|---|
| 104 | return E_FAIL;
|
|---|
| 105 | }
|
|---|
| 106 |
|
|---|
| 107 | HRESULT
|
|---|
| 108 | MARSHAL_Find_Stub(wine_marshal_id *mid,LPUNKNOWN *pUnk) {
|
|---|
| 109 | int i;
|
|---|
| 110 |
|
|---|
| 111 | for (i=0;i<nrofstubs;i++) {
|
|---|
| 112 | if (MARSHAL_Compare_Mids(mid,&(stubs[i].mid))) {
|
|---|
| 113 | *pUnk = stubs[i].pUnkServer;
|
|---|
| 114 | IUnknown_AddRef((*pUnk));
|
|---|
| 115 | return S_OK;
|
|---|
| 116 | }
|
|---|
| 117 | }
|
|---|
| 118 | return E_FAIL;
|
|---|
| 119 | }
|
|---|
| 120 |
|
|---|
| 121 | HRESULT
|
|---|
| 122 | MARSHAL_Register_Stub(wine_marshal_id *mid,LPUNKNOWN pUnk,IRpcStubBuffer *stub) {
|
|---|
| 123 | LPUNKNOWN xPunk;
|
|---|
| 124 | if (!MARSHAL_Find_Stub(mid,&xPunk)) {
|
|---|
| 125 | FIXME("Already have entry for (%lx/%s)!\n",mid->objectid,debugstr_guid(&(mid->iid)));
|
|---|
| 126 | return S_OK;
|
|---|
| 127 | }
|
|---|
| 128 | if (nrofstubs)
|
|---|
| 129 | stubs=HeapReAlloc(GetProcessHeap(),0,stubs,sizeof(stubs[0])*(nrofstubs+1));
|
|---|
| 130 | else
|
|---|
| 131 | stubs=HeapAlloc(GetProcessHeap(),0,sizeof(stubs[0]));
|
|---|
| 132 | if (!stubs) return E_OUTOFMEMORY;
|
|---|
| 133 | stubs[nrofstubs].stub = stub;
|
|---|
| 134 | stubs[nrofstubs].pUnkServer = pUnk;
|
|---|
| 135 | memcpy(&(stubs[nrofstubs].mid),mid,sizeof(*mid));
|
|---|
| 136 | nrofstubs++;
|
|---|
| 137 | return S_OK;
|
|---|
| 138 | }
|
|---|
| 139 |
|
|---|
| 140 | HRESULT
|
|---|
| 141 | MARSHAL_Find_Proxy(wine_marshal_id *mid,LPUNKNOWN *punk) {
|
|---|
| 142 | int i;
|
|---|
| 143 |
|
|---|
| 144 | for (i=0;i<nrofproxies;i++)
|
|---|
| 145 | if (MARSHAL_Compare_Mids(mid,&(proxies[i].mid))) {
|
|---|
| 146 | *punk = proxies[i].pUnk;
|
|---|
| 147 | IUnknown_AddRef((*punk));
|
|---|
| 148 | return S_OK;
|
|---|
| 149 | }
|
|---|
| 150 | return E_FAIL;
|
|---|
| 151 | }
|
|---|
| 152 |
|
|---|
| 153 | HRESULT
|
|---|
| 154 | MARSHAL_Find_Proxy_Object(wine_marshal_id *mid,LPUNKNOWN *punk) {
|
|---|
| 155 | int i;
|
|---|
| 156 |
|
|---|
| 157 | for (i=0;i<nrofproxies;i++)
|
|---|
| 158 | if (MARSHAL_Compare_Mids_NoInterface(mid,&(proxies[i].mid))) {
|
|---|
| 159 | *punk = proxies[i].pUnk;
|
|---|
| 160 | IUnknown_AddRef((*punk));
|
|---|
| 161 | return S_OK;
|
|---|
| 162 | }
|
|---|
| 163 | return E_FAIL;
|
|---|
| 164 | }
|
|---|
| 165 |
|
|---|
| 166 | HRESULT
|
|---|
| 167 | MARSHAL_Register_Proxy(wine_marshal_id *mid,LPUNKNOWN punk) {
|
|---|
| 168 | int i;
|
|---|
| 169 |
|
|---|
| 170 | for (i=0;i<nrofproxies;i++) {
|
|---|
| 171 | if (MARSHAL_Compare_Mids(mid,&(proxies[i].mid))) {
|
|---|
| 172 | ERR("Already have mid?\n");
|
|---|
| 173 | return E_FAIL;
|
|---|
| 174 | }
|
|---|
| 175 | }
|
|---|
| 176 | if (nrofproxies)
|
|---|
| 177 | proxies = HeapReAlloc(GetProcessHeap(),0,proxies,sizeof(proxies[0])*(nrofproxies+1));
|
|---|
| 178 | else
|
|---|
| 179 | proxies = HeapAlloc(GetProcessHeap(),0,sizeof(proxies[0]));
|
|---|
| 180 | memcpy(&(proxies[nrofproxies].mid),mid,sizeof(*mid));
|
|---|
| 181 | proxies[nrofproxies].pUnk = punk;
|
|---|
| 182 | nrofproxies++;
|
|---|
| 183 | IUnknown_AddRef(punk);
|
|---|
| 184 | return S_OK;
|
|---|
| 185 | }
|
|---|
| 186 |
|
|---|
| 187 | /********************** StdMarshal implementation ****************************/
|
|---|
| 188 | typedef struct _StdMarshalImpl {
|
|---|
| 189 | ICOM_VTABLE(IMarshal) *lpvtbl;
|
|---|
| 190 | DWORD ref;
|
|---|
| 191 |
|
|---|
| 192 | IID iid;
|
|---|
| 193 | DWORD dwDestContext;
|
|---|
| 194 | LPVOID pvDestContext;
|
|---|
| 195 | DWORD mshlflags;
|
|---|
| 196 | } StdMarshalImpl;
|
|---|
| 197 |
|
|---|
| 198 | HRESULT WINAPI
|
|---|
| 199 | StdMarshalImpl_QueryInterface(LPMARSHAL iface,REFIID riid,LPVOID *ppv) {
|
|---|
| 200 | *ppv = NULL;
|
|---|
| 201 | if (IsEqualIID(&IID_IUnknown,riid) || IsEqualIID(&IID_IMarshal,riid)) {
|
|---|
| 202 | *ppv = iface;
|
|---|
| 203 | IUnknown_AddRef(iface);
|
|---|
| 204 | return S_OK;
|
|---|
| 205 | }
|
|---|
| 206 | FIXME("No interface for %s.\n",debugstr_guid(riid));
|
|---|
| 207 | return E_NOINTERFACE;
|
|---|
| 208 | }
|
|---|
| 209 |
|
|---|
| 210 | ULONG WINAPI
|
|---|
| 211 | StdMarshalImpl_AddRef(LPMARSHAL iface) {
|
|---|
| 212 | ICOM_THIS(StdMarshalImpl,iface);
|
|---|
| 213 | This->ref++;
|
|---|
| 214 | return This->ref;
|
|---|
| 215 | }
|
|---|
| 216 |
|
|---|
| 217 | ULONG WINAPI
|
|---|
| 218 | StdMarshalImpl_Release(LPMARSHAL iface) {
|
|---|
| 219 | ICOM_THIS(StdMarshalImpl,iface);
|
|---|
| 220 | This->ref--;
|
|---|
| 221 |
|
|---|
| 222 | if (This->ref)
|
|---|
| 223 | return This->ref;
|
|---|
| 224 | HeapFree(GetProcessHeap(),0,This);
|
|---|
| 225 | return 0;
|
|---|
| 226 | }
|
|---|
| 227 |
|
|---|
| 228 | HRESULT WINAPI
|
|---|
| 229 | StdMarshalImpl_GetUnmarshalClass(
|
|---|
| 230 | LPMARSHAL iface, REFIID riid, void* pv, DWORD dwDestContext,
|
|---|
| 231 | void* pvDestContext, DWORD mshlflags, CLSID* pCid
|
|---|
| 232 | ) {
|
|---|
| 233 | memcpy(pCid,&CLSID_DfMarshal,sizeof(CLSID_DfMarshal));
|
|---|
| 234 | return S_OK;
|
|---|
| 235 | }
|
|---|
| 236 |
|
|---|
| 237 | HRESULT WINAPI
|
|---|
| 238 | StdMarshalImpl_GetMarshalSizeMax(
|
|---|
| 239 | LPMARSHAL iface, REFIID riid, void* pv, DWORD dwDestContext,
|
|---|
| 240 | void* pvDestContext, DWORD mshlflags, DWORD* pSize
|
|---|
| 241 | ) {
|
|---|
| 242 | *pSize = sizeof(wine_marshal_id)+sizeof(wine_marshal_data);
|
|---|
| 243 | return S_OK;
|
|---|
| 244 | }
|
|---|
| 245 |
|
|---|
| 246 | HRESULT WINAPI
|
|---|
| 247 | StdMarshalImpl_MarshalInterface(
|
|---|
| 248 | LPMARSHAL iface, IStream *pStm,REFIID riid, void* pv, DWORD dwDestContext,
|
|---|
| 249 | void* pvDestContext, DWORD mshlflags
|
|---|
| 250 | ) {
|
|---|
| 251 | wine_marshal_id mid;
|
|---|
| 252 | wine_marshal_data md;
|
|---|
| 253 | IUnknown *pUnk;
|
|---|
| 254 | ULONG res;
|
|---|
| 255 | HRESULT hres;
|
|---|
| 256 | IRpcStubBuffer *stub;
|
|---|
| 257 | IPSFactoryBuffer *psfacbuf;
|
|---|
| 258 |
|
|---|
| 259 | TRACE("(...,%s,...)\n",debugstr_guid(riid));
|
|---|
| 260 | IUnknown_QueryInterface((LPUNKNOWN)pv,&IID_IUnknown,(LPVOID*)&pUnk);
|
|---|
| 261 | mid.processid = GetCurrentProcessId();
|
|---|
| 262 | mid.objectid = (DWORD)pUnk; /* FIXME */
|
|---|
| 263 | IUnknown_Release(pUnk);
|
|---|
| 264 | memcpy(&mid.iid,riid,sizeof(mid.iid));
|
|---|
| 265 | md.dwDestContext = dwDestContext;
|
|---|
| 266 | md.mshlflags = mshlflags;
|
|---|
| 267 | hres = IStream_Write(pStm,&mid,sizeof(mid),&res);
|
|---|
| 268 | if (hres) return hres;
|
|---|
| 269 | hres = IStream_Write(pStm,&md,sizeof(md),&res);
|
|---|
| 270 | if (hres) return hres;
|
|---|
| 271 |
|
|---|
| 272 | if (SUCCEEDED(MARSHAL_Find_Stub(&mid,&pUnk))) {
|
|---|
| 273 | IUnknown_Release(pUnk);
|
|---|
| 274 | return S_OK;
|
|---|
| 275 | }
|
|---|
| 276 | hres = get_facbuf_for_iid(riid,&psfacbuf);
|
|---|
| 277 | if (hres) return hres;
|
|---|
| 278 | hres = IPSFactoryBuffer_CreateStub(psfacbuf,riid,pv,&stub);
|
|---|
| 279 | IPSFactoryBuffer_Release(psfacbuf);
|
|---|
| 280 | if (hres) {
|
|---|
| 281 | FIXME("Failed to create a stub for %s\n",debugstr_guid(riid));
|
|---|
| 282 | return hres;
|
|---|
| 283 | }
|
|---|
| 284 | IUnknown_QueryInterface((LPUNKNOWN)pv,riid,(LPVOID*)&pUnk);
|
|---|
| 285 | MARSHAL_Register_Stub(&mid,pUnk,stub);
|
|---|
| 286 | IUnknown_Release(pUnk);
|
|---|
| 287 | return S_OK;
|
|---|
| 288 | }
|
|---|
| 289 |
|
|---|
| 290 | HRESULT WINAPI
|
|---|
| 291 | StdMarshalImpl_UnmarshalInterface(
|
|---|
| 292 | LPMARSHAL iface, IStream *pStm, REFIID riid, void **ppv
|
|---|
| 293 | ) {
|
|---|
| 294 | wine_marshal_id mid;
|
|---|
| 295 | wine_marshal_data md;
|
|---|
| 296 | ULONG res;
|
|---|
| 297 | HRESULT hres;
|
|---|
| 298 | IPSFactoryBuffer *psfacbuf;
|
|---|
| 299 | IRpcProxyBuffer *rpcproxy;
|
|---|
| 300 | IRpcChannelBuffer *chanbuf;
|
|---|
| 301 |
|
|---|
| 302 | TRACE("(...,%s,....)\n",debugstr_guid(riid));
|
|---|
| 303 | hres = IStream_Read(pStm,&mid,sizeof(mid),&res);
|
|---|
| 304 | if (hres) return hres;
|
|---|
| 305 | hres = IStream_Read(pStm,&md,sizeof(md),&res);
|
|---|
| 306 | if (hres) return hres;
|
|---|
| 307 | if (SUCCEEDED(MARSHAL_Find_Stub(&mid,(LPUNKNOWN*)ppv))) {
|
|---|
| 308 | FIXME("Calling back to ourselves for %s!\n",debugstr_guid(riid));
|
|---|
| 309 | return S_OK;
|
|---|
| 310 | }
|
|---|
| 311 | hres = get_facbuf_for_iid(riid,&psfacbuf);
|
|---|
| 312 | if (hres) return hres;
|
|---|
| 313 | hres = IPSFactoryBuffer_CreateProxy(psfacbuf,NULL,riid,&rpcproxy,ppv);
|
|---|
| 314 | if (hres) {
|
|---|
| 315 | FIXME("Failed to create a proxy for %s\n",debugstr_guid(riid));
|
|---|
| 316 | return hres;
|
|---|
| 317 | }
|
|---|
| 318 | hres = PIPE_GetNewPipeBuf(&mid,&chanbuf);
|
|---|
| 319 | if (hres)
|
|---|
| 320 | FIXME("Failed to get an rpc channel buffer for %s\n",debugstr_guid(riid));
|
|---|
| 321 | IRpcProxyBuffer_Connect(rpcproxy,chanbuf);
|
|---|
| 322 | IRpcProxyBuffer_Release(rpcproxy); /* no need */
|
|---|
| 323 | IPSFactoryBuffer_Release(psfacbuf);
|
|---|
| 324 | return S_OK;
|
|---|
| 325 | }
|
|---|
| 326 |
|
|---|
| 327 | HRESULT WINAPI
|
|---|
| 328 | StdMarshalImpl_ReleaseMarshalData(LPMARSHAL iface, IStream *pStm) {
|
|---|
| 329 | FIXME("(), stub!\n");
|
|---|
| 330 | return S_OK;
|
|---|
| 331 | }
|
|---|
| 332 |
|
|---|
| 333 | HRESULT WINAPI
|
|---|
| 334 | StdMarshalImpl_DisconnectObject(LPMARSHAL iface, DWORD dwReserved) {
|
|---|
| 335 | FIXME("(), stub!\n");
|
|---|
| 336 | return S_OK;
|
|---|
| 337 | }
|
|---|
| 338 |
|
|---|
| 339 | ICOM_VTABLE(IMarshal) stdmvtbl = {
|
|---|
| 340 | ICOM_MSVTABLE_COMPAT_DummyRTTIVALUE
|
|---|
| 341 | StdMarshalImpl_QueryInterface,
|
|---|
| 342 | StdMarshalImpl_AddRef,
|
|---|
| 343 | StdMarshalImpl_Release,
|
|---|
| 344 | StdMarshalImpl_GetUnmarshalClass,
|
|---|
| 345 | StdMarshalImpl_GetMarshalSizeMax,
|
|---|
| 346 | StdMarshalImpl_MarshalInterface,
|
|---|
| 347 | StdMarshalImpl_UnmarshalInterface,
|
|---|
| 348 | StdMarshalImpl_ReleaseMarshalData,
|
|---|
| 349 | StdMarshalImpl_DisconnectObject
|
|---|
| 350 | };
|
|---|
| 351 |
|
|---|
| 352 | /***********************************************************************
|
|---|
| 353 | * CoGetStandardMarshal [OLE32.23]
|
|---|
| 354 | *
|
|---|
| 355 | * When the COM library in the client process receives a marshaled
|
|---|
| 356 | * interface pointer, it looks for a CLSID to be used in creating a proxy
|
|---|
| 357 | * for the purposes of unmarshaling the packet. If the packet does not
|
|---|
| 358 | * contain a CLSID for the proxy, COM calls CoGetStandardMarshal, passing a
|
|---|
| 359 | * NULL pUnk value.
|
|---|
| 360 | * This function creates a standard proxy in the client process and returns
|
|---|
| 361 | * a pointer to that proxy's implementation of IMarshal.
|
|---|
| 362 | * COM uses this pointer to call CoUnmarshalInterface to retrieve the pointer
|
|---|
| 363 | * to the requested interface.
|
|---|
| 364 | */
|
|---|
| 365 | HRESULT WINAPI
|
|---|
| 366 | CoGetStandardMarshal(
|
|---|
| 367 | REFIID riid,IUnknown *pUnk,DWORD dwDestContext,LPVOID pvDestContext,
|
|---|
| 368 | DWORD mshlflags, LPMARSHAL *pMarshal
|
|---|
| 369 | ) {
|
|---|
| 370 | StdMarshalImpl *dm;
|
|---|
| 371 |
|
|---|
| 372 | if (pUnk == NULL) {
|
|---|
| 373 | FIXME("(%s,NULL,%lx,%p,%lx,%p), unimplemented yet.\n",
|
|---|
| 374 | debugstr_guid(riid),dwDestContext,pvDestContext,mshlflags,pMarshal
|
|---|
| 375 | );
|
|---|
| 376 | return E_FAIL;
|
|---|
| 377 | }
|
|---|
| 378 | TRACE("(%s,%p,%lx,%p,%lx,%p)\n",
|
|---|
| 379 | debugstr_guid(riid),pUnk,dwDestContext,pvDestContext,mshlflags,pMarshal
|
|---|
| 380 | );
|
|---|
| 381 | #ifdef __WIN32OS2__
|
|---|
| 382 | *pMarshal = HeapAlloc(GetProcessHeap(),0,sizeof(StdMarshalImpl));
|
|---|
| 383 | dm = (StdMarshalImpl*) *pMarshal;
|
|---|
| 384 | #else
|
|---|
| 385 | dm = (StdMarshalImpl*) *pMarshal = HeapAlloc(GetProcessHeap(),0,sizeof(StdMarshalImpl));
|
|---|
| 386 | #endif
|
|---|
| 387 | if (!dm) return E_FAIL;
|
|---|
| 388 | dm->lpvtbl = &stdmvtbl;
|
|---|
| 389 | dm->ref = 1;
|
|---|
| 390 |
|
|---|
| 391 | memcpy(&dm->iid,riid,sizeof(dm->iid));
|
|---|
| 392 | dm->dwDestContext = dwDestContext;
|
|---|
| 393 | dm->pvDestContext = pvDestContext;
|
|---|
| 394 | dm->mshlflags = mshlflags;
|
|---|
| 395 | return S_OK;
|
|---|
| 396 | }
|
|---|
| 397 |
|
|---|
| 398 | /* Helper function for getting Marshaler */
|
|---|
| 399 | static HRESULT WINAPI
|
|---|
| 400 | _GetMarshaller(REFIID riid, IUnknown *pUnk,DWORD dwDestContext,
|
|---|
| 401 | void *pvDestContext, DWORD mshlFlags, LPMARSHAL *pMarshal
|
|---|
| 402 | ) {
|
|---|
| 403 | HRESULT hres;
|
|---|
| 404 |
|
|---|
| 405 | if (!pUnk)
|
|---|
| 406 | return E_POINTER;
|
|---|
| 407 | hres = IUnknown_QueryInterface(pUnk,&IID_IMarshal,(LPVOID*)pMarshal);
|
|---|
| 408 | if (hres)
|
|---|
| 409 | hres = CoGetStandardMarshal(riid,pUnk,dwDestContext,pvDestContext,mshlFlags,pMarshal);
|
|---|
| 410 | return hres;
|
|---|
| 411 | }
|
|---|
| 412 |
|
|---|
| 413 | /***********************************************************************
|
|---|
| 414 | * CoGetMarshalSizeMax [OLE32.21]
|
|---|
| 415 | */
|
|---|
| 416 | HRESULT WINAPI
|
|---|
| 417 | CoGetMarshalSizeMax(ULONG *pulSize, REFIID riid, IUnknown *pUnk,
|
|---|
| 418 | DWORD dwDestContext, void *pvDestContext, DWORD mshlFlags
|
|---|
| 419 | ) {
|
|---|
| 420 | HRESULT hres;
|
|---|
| 421 | LPMARSHAL pMarshal;
|
|---|
| 422 |
|
|---|
| 423 | hres = _GetMarshaller(riid,pUnk,dwDestContext,pvDestContext,mshlFlags,&pMarshal);
|
|---|
| 424 | if (hres)
|
|---|
| 425 | return hres;
|
|---|
| 426 | hres = IMarshal_GetMarshalSizeMax(pMarshal,riid,pUnk,dwDestContext,pvDestContext,mshlFlags,pulSize);
|
|---|
| 427 | *pulSize += sizeof(wine_marshal_id)+sizeof(wine_marshal_data)+sizeof(CLSID);
|
|---|
| 428 | IMarshal_Release(pMarshal);
|
|---|
| 429 | return hres;
|
|---|
| 430 | }
|
|---|
| 431 |
|
|---|
| 432 |
|
|---|
| 433 | /***********************************************************************
|
|---|
| 434 | * CoMarshalInterface [OLE32.34]
|
|---|
| 435 | */
|
|---|
| 436 | HRESULT WINAPI
|
|---|
| 437 | CoMarshalInterface( IStream *pStm, REFIID riid, IUnknown *pUnk,
|
|---|
| 438 | DWORD dwDestContext, void *pvDestContext, DWORD mshlflags
|
|---|
| 439 | ) {
|
|---|
| 440 | HRESULT hres;
|
|---|
| 441 | LPMARSHAL pMarshal;
|
|---|
| 442 | CLSID xclsid;
|
|---|
| 443 | ULONG writeres;
|
|---|
| 444 | wine_marshal_id mid;
|
|---|
| 445 | wine_marshal_data md;
|
|---|
| 446 | ULONG res;
|
|---|
| 447 | IUnknown *pUnknown;
|
|---|
| 448 |
|
|---|
| 449 | TRACE("(%p, %s, %p, %lx, %p, %lx)\n",
|
|---|
| 450 | pStm,debugstr_guid(riid),pUnk,dwDestContext,pvDestContext,mshlflags
|
|---|
| 451 | );
|
|---|
| 452 | STUBMGR_Start(); /* Just to be sure we have one running. */
|
|---|
| 453 | mid.processid = GetCurrentProcessId();
|
|---|
| 454 | IUnknown_QueryInterface(pUnk,&IID_IUnknown,(LPVOID*)&pUnknown);
|
|---|
| 455 | mid.objectid = (DWORD)pUnknown;
|
|---|
| 456 | IUnknown_Release(pUnknown);
|
|---|
| 457 | memcpy(&mid.iid,riid,sizeof(mid.iid));
|
|---|
| 458 | md.dwDestContext = dwDestContext;
|
|---|
| 459 | md.mshlflags = mshlflags;
|
|---|
| 460 | hres = IStream_Write(pStm,&mid,sizeof(mid),&res);
|
|---|
| 461 | if (hres) return hres;
|
|---|
| 462 | hres = IStream_Write(pStm,&md,sizeof(md),&res);
|
|---|
| 463 | if (hres) return hres;
|
|---|
| 464 | hres = _GetMarshaller(riid,pUnk,dwDestContext,pvDestContext,mshlflags,&pMarshal);
|
|---|
| 465 | if (hres) {
|
|---|
| 466 | FIXME("Failed to get marshaller, %lx?\n",hres);
|
|---|
| 467 | return hres;
|
|---|
| 468 | }
|
|---|
| 469 | hres = IMarshal_GetUnmarshalClass(pMarshal,riid,pUnk,dwDestContext,pvDestContext,mshlflags,&xclsid);
|
|---|
| 470 | if (hres) {
|
|---|
| 471 | FIXME("IMarshal:GetUnmarshalClass failed, %lx\n",hres);
|
|---|
| 472 | goto release_marshal;
|
|---|
| 473 | }
|
|---|
| 474 | hres = IStream_Write(pStm,&xclsid,sizeof(xclsid),&writeres);
|
|---|
| 475 | if (hres) {
|
|---|
| 476 | FIXME("Stream write failed, %lx\n",hres);
|
|---|
| 477 | goto release_marshal;
|
|---|
| 478 | }
|
|---|
| 479 | hres = IMarshal_MarshalInterface(pMarshal,pStm,riid,pUnk,dwDestContext,pvDestContext,mshlflags);
|
|---|
| 480 | if (hres) {
|
|---|
| 481 | if (IsEqualGUID(riid,&IID_IClassFactory)) {
|
|---|
| 482 | MESSAGE("\nERROR: You need to merge the 'winedefault.reg' file into your\n");
|
|---|
| 483 | MESSAGE(" Wine registry by running: `regedit winedefault.reg'\n\n");
|
|---|
| 484 | } else {
|
|---|
| 485 | FIXME("Failed to Marshal the interface, %lx?\n",hres);
|
|---|
| 486 | }
|
|---|
| 487 | goto release_marshal;
|
|---|
| 488 | }
|
|---|
| 489 | release_marshal:
|
|---|
| 490 | IMarshal_Release(pMarshal);
|
|---|
| 491 | return hres;
|
|---|
| 492 | }
|
|---|
| 493 |
|
|---|
| 494 |
|
|---|
| 495 | /***********************************************************************
|
|---|
| 496 | * CoUnmarshalInterface [OLE32.50]
|
|---|
| 497 | */
|
|---|
| 498 | HRESULT WINAPI
|
|---|
| 499 | CoUnmarshalInterface(IStream *pStm, REFIID riid, LPVOID *ppv) {
|
|---|
| 500 | HRESULT hres;
|
|---|
| 501 | wine_marshal_id mid;
|
|---|
| 502 | wine_marshal_data md;
|
|---|
| 503 | ULONG res;
|
|---|
| 504 | LPMARSHAL pMarshal;
|
|---|
| 505 | LPUNKNOWN pUnk;
|
|---|
| 506 | CLSID xclsid;
|
|---|
| 507 |
|
|---|
| 508 | TRACE("(%p,%s,%p)\n",pStm,debugstr_guid(riid),ppv);
|
|---|
| 509 |
|
|---|
| 510 | hres = IStream_Read(pStm,&mid,sizeof(mid),&res);
|
|---|
| 511 | if (hres) {
|
|---|
| 512 | FIXME("Stream read 1 failed, %lx, (%ld of %d)\n",hres,res,sizeof(mid));
|
|---|
| 513 | return hres;
|
|---|
| 514 | }
|
|---|
| 515 | hres = IStream_Read(pStm,&md,sizeof(md),&res);
|
|---|
| 516 | if (hres) {
|
|---|
| 517 | FIXME("Stream read 2 failed, %lx, (%ld of %d)\n",hres,res,sizeof(md));
|
|---|
| 518 | return hres;
|
|---|
| 519 | }
|
|---|
| 520 | hres = IStream_Read(pStm,&xclsid,sizeof(xclsid),&res);
|
|---|
| 521 | if (hres) {
|
|---|
| 522 | FIXME("Stream read 3 failed, %lx, (%ld of %d)\n",hres,res,sizeof(xclsid));
|
|---|
| 523 | return hres;
|
|---|
| 524 | }
|
|---|
| 525 | hres=CoCreateInstance(&xclsid,NULL,CLSCTX_INPROC_SERVER | CLSCTX_INPROC_HANDLER | CLSCTX_LOCAL_SERVER,&IID_IMarshal,(void**)&pUnk);
|
|---|
| 526 | if (hres) {
|
|---|
| 527 | FIXME("Failed to create instance of unmarshaller %s.\n",debugstr_guid(&xclsid));
|
|---|
| 528 | return hres;
|
|---|
| 529 | }
|
|---|
| 530 | hres = _GetMarshaller(riid,pUnk,md.dwDestContext,NULL,md.mshlflags,&pMarshal);
|
|---|
| 531 | if (hres) {
|
|---|
| 532 | FIXME("Failed to get unmarshaller, %lx?\n",hres);
|
|---|
| 533 | return hres;
|
|---|
| 534 | }
|
|---|
| 535 | hres = IMarshal_UnmarshalInterface(pMarshal,pStm,riid,ppv);
|
|---|
| 536 | if (hres) {
|
|---|
| 537 | FIXME("Failed to Unmarshal the interface, %lx?\n",hres);
|
|---|
| 538 | goto release_marshal;
|
|---|
| 539 | }
|
|---|
| 540 | release_marshal:
|
|---|
| 541 | IMarshal_Release(pMarshal);
|
|---|
| 542 | return hres;
|
|---|
| 543 | }
|
|---|
| 544 |
|
|---|
| 545 | /***********************************************************************
|
|---|
| 546 | * CoMarshalInterThreadInterfaceInStream [OLE32.33]
|
|---|
| 547 | *
|
|---|
| 548 | * Marshal interfaces across threads. We don't have a thread distinction,
|
|---|
| 549 | * meaning most interfaces just work across different threads, the RPC
|
|---|
| 550 | * handles it.
|
|---|
| 551 | */
|
|---|
| 552 | HRESULT WINAPI
|
|---|
| 553 | CoMarshalInterThreadInterfaceInStream(
|
|---|
| 554 | REFIID riid, LPUNKNOWN pUnk, LPSTREAM * ppStm
|
|---|
| 555 | ) {
|
|---|
| 556 | ULONG res;
|
|---|
| 557 | ULARGE_INTEGER xpos;
|
|---|
| 558 | LARGE_INTEGER seekto;
|
|---|
| 559 | HRESULT hres;
|
|---|
| 560 |
|
|---|
| 561 | TRACE("(,%s,)\n",debugstr_guid(riid));
|
|---|
| 562 | hres = CreateStreamOnHGlobal(0, TRUE, ppStm);
|
|---|
| 563 | if (hres) return hres;
|
|---|
| 564 | /* CoMarshalInterface(...); */
|
|---|
| 565 | hres = IStream_Write(*ppStm,&pUnk,sizeof(LPUNKNOWN),&res);
|
|---|
| 566 | if (hres) return hres;
|
|---|
| 567 | memset(&seekto,0,sizeof(seekto));
|
|---|
| 568 | IStream_Seek(*ppStm,seekto,SEEK_SET,&xpos);
|
|---|
| 569 | return S_OK;
|
|---|
| 570 | }
|
|---|
| 571 |
|
|---|
| 572 | /***********************************************************************
|
|---|
| 573 | * CoGetInterfaceAndReleaseStream [OLE32.19]
|
|---|
| 574 | */
|
|---|
| 575 | HRESULT WINAPI
|
|---|
| 576 | CoGetInterfaceAndReleaseStream(LPSTREAM pStm,REFIID riid, LPVOID *ppv) {
|
|---|
| 577 | ULONG res;
|
|---|
| 578 | HRESULT hres;
|
|---|
| 579 | LPUNKNOWN pUnk;
|
|---|
| 580 |
|
|---|
| 581 | TRACE("(,%s,)\n",debugstr_guid(riid));
|
|---|
| 582 | /* CoUnmarshalInterface(...); */
|
|---|
| 583 | hres = IStream_Read(pStm,&pUnk,sizeof(LPUNKNOWN),&res);
|
|---|
| 584 | if (hres) return hres;
|
|---|
| 585 | IStream_Release(pStm);
|
|---|
| 586 | return IUnknown_QueryInterface(pUnk,riid,ppv);
|
|---|
| 587 | }
|
|---|
| 588 |
|
|---|
| 589 | static HRESULT WINAPI
|
|---|
| 590 | SMCF_QueryInterface(LPCLASSFACTORY iface,REFIID riid, LPVOID *ppv) {
|
|---|
| 591 | *ppv = NULL;
|
|---|
| 592 | if (IsEqualIID(riid,&IID_IUnknown) || IsEqualIID(riid,&IID_IClassFactory)) {
|
|---|
| 593 | *ppv = (LPVOID)iface;
|
|---|
| 594 | return S_OK;
|
|---|
| 595 | }
|
|---|
| 596 | return E_NOINTERFACE;
|
|---|
| 597 | }
|
|---|
| 598 | static ULONG WINAPI SMCF_AddRef(LPCLASSFACTORY iface) { return 2; }
|
|---|
| 599 | static ULONG WINAPI SMCF_Release(LPCLASSFACTORY iface) { return 1; }
|
|---|
| 600 |
|
|---|
| 601 | static HRESULT WINAPI
|
|---|
| 602 | SMCF_CreateInstance(
|
|---|
| 603 | LPCLASSFACTORY iface, LPUNKNOWN pUnk, REFIID riid, LPVOID *ppv
|
|---|
| 604 | ) {
|
|---|
| 605 | if (IsEqualIID(riid,&IID_IMarshal)) {
|
|---|
| 606 | StdMarshalImpl *dm;
|
|---|
| 607 | dm=(StdMarshalImpl*)HeapAlloc(GetProcessHeap(),0,sizeof(StdMarshalImpl));
|
|---|
| 608 | if (!dm)
|
|---|
| 609 | return E_FAIL;
|
|---|
| 610 | dm->lpvtbl = &stdmvtbl;
|
|---|
| 611 | dm->ref = 1;
|
|---|
| 612 | *ppv = (LPVOID)dm;
|
|---|
| 613 | return S_OK;
|
|---|
| 614 | }
|
|---|
| 615 | FIXME("(%s), not supported.\n",debugstr_guid(riid));
|
|---|
| 616 | return E_NOINTERFACE;
|
|---|
| 617 | }
|
|---|
| 618 |
|
|---|
| 619 | static HRESULT WINAPI
|
|---|
| 620 | SMCF_LockServer(LPCLASSFACTORY iface, BOOL fLock) {
|
|---|
| 621 | FIXME("(%d), stub!\n",fLock);
|
|---|
| 622 | return S_OK;
|
|---|
| 623 | }
|
|---|
| 624 |
|
|---|
| 625 | static ICOM_VTABLE(IClassFactory) dfmarshalcfvtbl = {
|
|---|
| 626 | ICOM_MSVTABLE_COMPAT_DummyRTTIVALUE
|
|---|
| 627 | SMCF_QueryInterface,
|
|---|
| 628 | SMCF_AddRef,
|
|---|
| 629 | SMCF_Release,
|
|---|
| 630 | SMCF_CreateInstance,
|
|---|
| 631 | SMCF_LockServer
|
|---|
| 632 | };
|
|---|
| 633 | static ICOM_VTABLE(IClassFactory) *pdfmarshalcfvtbl = &dfmarshalcfvtbl;
|
|---|
| 634 |
|
|---|
| 635 | HRESULT
|
|---|
| 636 | MARSHAL_GetStandardMarshalCF(LPVOID *ppv) {
|
|---|
| 637 | *ppv = &pdfmarshalcfvtbl;
|
|---|
| 638 | return S_OK;
|
|---|
| 639 | }
|
|---|