Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 1 | /* |
| 2 | * TYPELIB Marshaler |
| 3 | * |
Marcus Meissner | 4107986 | 2005-03-29 13:06:44 +0000 | [diff] [blame] | 4 | * Copyright 2002,2005 Marcus Meissner |
Alexandre Julliard | 0799c1a | 2002-03-09 23:29:33 +0000 | [diff] [blame] | 5 | * |
Mike Hearn | d440ca4 | 2004-06-17 23:03:26 +0000 | [diff] [blame] | 6 | * The olerelay debug channel allows you to see calls marshalled by |
| 7 | * the typelib marshaller. It is not a generic COM relaying system. |
| 8 | * |
Alexandre Julliard | 0799c1a | 2002-03-09 23:29:33 +0000 | [diff] [blame] | 9 | * This library is free software; you can redistribute it and/or |
| 10 | * modify it under the terms of the GNU Lesser General Public |
| 11 | * License as published by the Free Software Foundation; either |
| 12 | * version 2.1 of the License, or (at your option) any later version. |
| 13 | * |
| 14 | * This library is distributed in the hope that it will be useful, |
| 15 | * but WITHOUT ANY WARRANTY; without even the implied warranty of |
| 16 | * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU |
| 17 | * Lesser General Public License for more details. |
| 18 | * |
| 19 | * You should have received a copy of the GNU Lesser General Public |
| 20 | * License along with this library; if not, write to the Free Software |
Jonathan Ernst | 360a3f9 | 2006-05-18 14:49:52 +0200 | [diff] [blame] | 21 | * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 22 | */ |
| 23 | |
| 24 | #include "config.h" |
| 25 | |
| 26 | #include <assert.h> |
| 27 | #include <stdlib.h> |
| 28 | #include <string.h> |
Alexandre Julliard | e37c6e1 | 2003-09-05 23:08:26 +0000 | [diff] [blame] | 29 | #include <stdarg.h> |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 30 | #include <stdio.h> |
| 31 | #include <ctype.h> |
| 32 | |
Francois Gouget | 486d020 | 2004-10-07 03:06:48 +0000 | [diff] [blame] | 33 | #define COBJMACROS |
Dimitrie O. Paun | 297f3d8 | 2003-01-07 20:36:20 +0000 | [diff] [blame] | 34 | #define NONAMELESSUNION |
| 35 | #define NONAMELESSSTRUCT |
Francois Gouget | 486d020 | 2004-10-07 03:06:48 +0000 | [diff] [blame] | 36 | |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 37 | #include "winerror.h" |
Alexandre Julliard | e37c6e1 | 2003-09-05 23:08:26 +0000 | [diff] [blame] | 38 | #include "windef.h" |
Alexandre Julliard | 864ca06 | 2003-08-20 18:22:31 +0000 | [diff] [blame] | 39 | #include "winbase.h" |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 40 | #include "winnls.h" |
| 41 | #include "winreg.h" |
| 42 | #include "winuser.h" |
| 43 | |
| 44 | #include "ole2.h" |
Huw Davies | 86d9457 | 2006-05-10 12:11:22 +0100 | [diff] [blame] | 45 | #include "propidl.h" /* for LPSAFEARRAY_User* functions */ |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 46 | #include "typelib.h" |
Robert Shearman | 799ebfc | 2006-01-27 12:54:03 +0100 | [diff] [blame] | 47 | #include "variant.h" |
Alexandre Julliard | 0799c1a | 2002-03-09 23:29:33 +0000 | [diff] [blame] | 48 | #include "wine/debug.h" |
Robert Shearman | a05672e | 2006-07-17 20:19:27 +0100 | [diff] [blame] | 49 | #include "wine/exception.h" |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 50 | |
Marcus Meissner | e6260a3 | 2005-03-27 17:51:59 +0000 | [diff] [blame] | 51 | static const WCHAR IDispatchW[] = { 'I','D','i','s','p','a','t','c','h',0}; |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 52 | |
Alexandre Julliard | 0799c1a | 2002-03-09 23:29:33 +0000 | [diff] [blame] | 53 | WINE_DEFAULT_DEBUG_CHANNEL(ole); |
| 54 | WINE_DECLARE_DEBUG_CHANNEL(olerelay); |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 55 | |
Alexandre Julliard | 936198c | 2004-08-13 00:44:22 +0000 | [diff] [blame] | 56 | #define ICOM_THIS_MULTI(impl,field,iface) impl* const This=(impl*)((char*)(iface) - offsetof(impl,field)) |
| 57 | |
Rob Shearman | 30f2b44 | 2007-02-06 19:19:56 +0000 | [diff] [blame] | 58 | static HRESULT TMarshalDispatchChannel_Create( |
| 59 | IRpcChannelBuffer *pDelegateChannel, REFIID tmarshal_riid, |
| 60 | IRpcChannelBuffer **ppChannel); |
| 61 | |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 62 | typedef struct _marshal_state { |
| 63 | LPBYTE base; |
| 64 | int size; |
| 65 | int curoff; |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 66 | } marshal_state; |
| 67 | |
Mike Hearn | d440ca4 | 2004-06-17 23:03:26 +0000 | [diff] [blame] | 68 | /* used in the olerelay code to avoid having the L"" stuff added by debugstr_w */ |
| 69 | static char *relaystr(WCHAR *in) { |
| 70 | char *tmp = (char *)debugstr_w(in); |
| 71 | tmp += 2; |
| 72 | tmp[strlen(tmp)-1] = '\0'; |
| 73 | return tmp; |
| 74 | } |
| 75 | |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 76 | static HRESULT |
Huw Davies | d0bdf68 | 2006-05-10 11:55:37 +0100 | [diff] [blame] | 77 | xbuf_resize(marshal_state *buf, DWORD newsize) |
| 78 | { |
| 79 | if(buf->size >= newsize) |
| 80 | return S_FALSE; |
| 81 | |
| 82 | if(buf->base) |
| 83 | { |
| 84 | buf->base = HeapReAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, buf->base, newsize); |
| 85 | if(!buf->base) |
| 86 | return E_OUTOFMEMORY; |
| 87 | } |
| 88 | else |
| 89 | { |
| 90 | buf->base = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, newsize); |
| 91 | if(!buf->base) |
| 92 | return E_OUTOFMEMORY; |
| 93 | } |
Huw Davies | 3dce323 | 2006-05-15 10:02:48 +0100 | [diff] [blame] | 94 | buf->size = newsize; |
Huw Davies | d0bdf68 | 2006-05-10 11:55:37 +0100 | [diff] [blame] | 95 | return S_OK; |
| 96 | } |
| 97 | |
| 98 | static HRESULT |
Andrew Talbot | f73821c | 2007-07-25 21:37:23 +0100 | [diff] [blame] | 99 | xbuf_add(marshal_state *buf, const BYTE *stuff, DWORD size) |
Huw Davies | d0bdf68 | 2006-05-10 11:55:37 +0100 | [diff] [blame] | 100 | { |
| 101 | HRESULT hr; |
| 102 | |
| 103 | if(buf->size - buf->curoff < size) |
| 104 | { |
| 105 | hr = xbuf_resize(buf, buf->size + size + 100); |
| 106 | if(FAILED(hr)) return hr; |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 107 | } |
| 108 | memcpy(buf->base+buf->curoff,stuff,size); |
| 109 | buf->curoff += size; |
| 110 | return S_OK; |
| 111 | } |
| 112 | |
| 113 | static HRESULT |
| 114 | xbuf_get(marshal_state *buf, LPBYTE stuff, DWORD size) { |
| 115 | if (buf->size < buf->curoff+size) return E_FAIL; |
| 116 | memcpy(stuff,buf->base+buf->curoff,size); |
| 117 | buf->curoff += size; |
| 118 | return S_OK; |
| 119 | } |
| 120 | |
| 121 | static HRESULT |
| 122 | xbuf_skip(marshal_state *buf, DWORD size) { |
| 123 | if (buf->size < buf->curoff+size) return E_FAIL; |
| 124 | buf->curoff += size; |
| 125 | return S_OK; |
| 126 | } |
| 127 | |
| 128 | static HRESULT |
| 129 | _unmarshal_interface(marshal_state *buf, REFIID riid, LPUNKNOWN *pUnk) { |
| 130 | IStream *pStm; |
| 131 | ULARGE_INTEGER newpos; |
| 132 | LARGE_INTEGER seekto; |
| 133 | ULONG res; |
| 134 | HRESULT hres; |
| 135 | DWORD xsize; |
| 136 | |
| 137 | TRACE("...%s...\n",debugstr_guid(riid)); |
Mike Hearn | 64eac75 | 2005-01-27 11:11:32 +0000 | [diff] [blame] | 138 | |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 139 | *pUnk = NULL; |
| 140 | hres = xbuf_get(buf,(LPBYTE)&xsize,sizeof(xsize)); |
Mike Hearn | 64eac75 | 2005-01-27 11:11:32 +0000 | [diff] [blame] | 141 | if (hres) { |
| 142 | ERR("xbuf_get failed\n"); |
| 143 | return hres; |
| 144 | } |
| 145 | |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 146 | if (xsize == 0) return S_OK; |
Mike Hearn | 64eac75 | 2005-01-27 11:11:32 +0000 | [diff] [blame] | 147 | |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 148 | hres = CreateStreamOnHGlobal(0,TRUE,&pStm); |
| 149 | if (hres) { |
Hans Leidekker | ce66f5d | 2006-10-12 20:57:10 +0200 | [diff] [blame] | 150 | ERR("Stream create failed %x\n",hres); |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 151 | return hres; |
| 152 | } |
Mike Hearn | 64eac75 | 2005-01-27 11:11:32 +0000 | [diff] [blame] | 153 | |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 154 | hres = IStream_Write(pStm,buf->base+buf->curoff,xsize,&res); |
Mike Hearn | 64eac75 | 2005-01-27 11:11:32 +0000 | [diff] [blame] | 155 | if (hres) { |
Hans Leidekker | ce66f5d | 2006-10-12 20:57:10 +0200 | [diff] [blame] | 156 | ERR("stream write %x\n",hres); |
Mike Hearn | 64eac75 | 2005-01-27 11:11:32 +0000 | [diff] [blame] | 157 | return hres; |
| 158 | } |
| 159 | |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 160 | memset(&seekto,0,sizeof(seekto)); |
| 161 | hres = IStream_Seek(pStm,seekto,SEEK_SET,&newpos); |
Mike Hearn | 64eac75 | 2005-01-27 11:11:32 +0000 | [diff] [blame] | 162 | if (hres) { |
Hans Leidekker | ce66f5d | 2006-10-12 20:57:10 +0200 | [diff] [blame] | 163 | ERR("Failed Seek %x\n",hres); |
Mike Hearn | 64eac75 | 2005-01-27 11:11:32 +0000 | [diff] [blame] | 164 | return hres; |
| 165 | } |
| 166 | |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 167 | hres = CoUnmarshalInterface(pStm,riid,(LPVOID*)pUnk); |
| 168 | if (hres) { |
Hans Leidekker | ce66f5d | 2006-10-12 20:57:10 +0200 | [diff] [blame] | 169 | ERR("Unmarshalling interface %s failed with %x\n",debugstr_guid(riid),hres); |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 170 | return hres; |
| 171 | } |
Mike Hearn | 64eac75 | 2005-01-27 11:11:32 +0000 | [diff] [blame] | 172 | |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 173 | IStream_Release(pStm); |
| 174 | return xbuf_skip(buf,xsize); |
| 175 | } |
| 176 | |
| 177 | static HRESULT |
| 178 | _marshal_interface(marshal_state *buf, REFIID riid, LPUNKNOWN pUnk) { |
Mike Hearn | 64eac75 | 2005-01-27 11:11:32 +0000 | [diff] [blame] | 179 | LPBYTE tempbuf = NULL; |
| 180 | IStream *pStm = NULL; |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 181 | STATSTG ststg; |
| 182 | ULARGE_INTEGER newpos; |
| 183 | LARGE_INTEGER seekto; |
| 184 | ULONG res; |
| 185 | DWORD xsize; |
| 186 | HRESULT hres; |
| 187 | |
Mike Hearn | 64eac75 | 2005-01-27 11:11:32 +0000 | [diff] [blame] | 188 | if (!pUnk) { |
Marcus Meissner | 4107986 | 2005-03-29 13:06:44 +0000 | [diff] [blame] | 189 | /* this is valid, if for instance we serialize |
| 190 | * a VT_DISPATCH with NULL ptr which apparently |
| 191 | * can happen. S_OK to make sure we continue |
| 192 | * serializing. |
| 193 | */ |
Robert Shearman | 8984694 | 2006-01-18 11:29:01 +0100 | [diff] [blame] | 194 | WARN("pUnk is NULL\n"); |
Marcus Meissner | 4107986 | 2005-03-29 13:06:44 +0000 | [diff] [blame] | 195 | xsize = 0; |
| 196 | return xbuf_add(buf,(LPBYTE)&xsize,sizeof(xsize)); |
Mike Hearn | 64eac75 | 2005-01-27 11:11:32 +0000 | [diff] [blame] | 197 | } |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 198 | |
Marcus Meissner | 4107986 | 2005-03-29 13:06:44 +0000 | [diff] [blame] | 199 | hres = E_FAIL; |
| 200 | |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 201 | TRACE("...%s...\n",debugstr_guid(riid)); |
Mike Hearn | 64eac75 | 2005-01-27 11:11:32 +0000 | [diff] [blame] | 202 | |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 203 | hres = CreateStreamOnHGlobal(0,TRUE,&pStm); |
| 204 | if (hres) { |
Hans Leidekker | ce66f5d | 2006-10-12 20:57:10 +0200 | [diff] [blame] | 205 | ERR("Stream create failed %x\n",hres); |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 206 | goto fail; |
| 207 | } |
Mike Hearn | 64eac75 | 2005-01-27 11:11:32 +0000 | [diff] [blame] | 208 | |
Robert Shearman | 4519b1b | 2005-08-24 09:44:41 +0000 | [diff] [blame] | 209 | hres = CoMarshalInterface(pStm,riid,pUnk,0,NULL,0); |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 210 | if (hres) { |
Hans Leidekker | ce66f5d | 2006-10-12 20:57:10 +0200 | [diff] [blame] | 211 | ERR("Marshalling interface %s failed with %x\n", debugstr_guid(riid), hres); |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 212 | goto fail; |
| 213 | } |
Mike Hearn | 64eac75 | 2005-01-27 11:11:32 +0000 | [diff] [blame] | 214 | |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 215 | hres = IStream_Stat(pStm,&ststg,0); |
Mike Hearn | 64eac75 | 2005-01-27 11:11:32 +0000 | [diff] [blame] | 216 | if (hres) { |
| 217 | ERR("Stream stat failed\n"); |
| 218 | goto fail; |
| 219 | } |
| 220 | |
Ge van Geldorp | 399901e | 2004-01-23 01:51:33 +0000 | [diff] [blame] | 221 | tempbuf = HeapAlloc(GetProcessHeap(), 0, ststg.cbSize.u.LowPart); |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 222 | memset(&seekto,0,sizeof(seekto)); |
| 223 | hres = IStream_Seek(pStm,seekto,SEEK_SET,&newpos); |
Mike Hearn | 64eac75 | 2005-01-27 11:11:32 +0000 | [diff] [blame] | 224 | if (hres) { |
Hans Leidekker | ce66f5d | 2006-10-12 20:57:10 +0200 | [diff] [blame] | 225 | ERR("Failed Seek %x\n",hres); |
Mike Hearn | 64eac75 | 2005-01-27 11:11:32 +0000 | [diff] [blame] | 226 | goto fail; |
| 227 | } |
| 228 | |
Ge van Geldorp | 399901e | 2004-01-23 01:51:33 +0000 | [diff] [blame] | 229 | hres = IStream_Read(pStm,tempbuf,ststg.cbSize.u.LowPart,&res); |
Mike Hearn | 64eac75 | 2005-01-27 11:11:32 +0000 | [diff] [blame] | 230 | if (hres) { |
Hans Leidekker | ce66f5d | 2006-10-12 20:57:10 +0200 | [diff] [blame] | 231 | ERR("Failed Read %x\n",hres); |
Mike Hearn | 64eac75 | 2005-01-27 11:11:32 +0000 | [diff] [blame] | 232 | goto fail; |
| 233 | } |
| 234 | |
Ge van Geldorp | 399901e | 2004-01-23 01:51:33 +0000 | [diff] [blame] | 235 | xsize = ststg.cbSize.u.LowPart; |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 236 | xbuf_add(buf,(LPBYTE)&xsize,sizeof(xsize)); |
Ge van Geldorp | 399901e | 2004-01-23 01:51:33 +0000 | [diff] [blame] | 237 | hres = xbuf_add(buf,tempbuf,ststg.cbSize.u.LowPart); |
Mike Hearn | 64eac75 | 2005-01-27 11:11:32 +0000 | [diff] [blame] | 238 | |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 239 | HeapFree(GetProcessHeap(),0,tempbuf); |
Mike Hearn | 64eac75 | 2005-01-27 11:11:32 +0000 | [diff] [blame] | 240 | IStream_Release(pStm); |
| 241 | |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 242 | return hres; |
Mike Hearn | 64eac75 | 2005-01-27 11:11:32 +0000 | [diff] [blame] | 243 | |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 244 | fail: |
| 245 | xsize = 0; |
| 246 | xbuf_add(buf,(LPBYTE)&xsize,sizeof(xsize)); |
Mike Hearn | 64eac75 | 2005-01-27 11:11:32 +0000 | [diff] [blame] | 247 | if (pStm) IUnknown_Release(pStm); |
Mike Hearn | 64eac75 | 2005-01-27 11:11:32 +0000 | [diff] [blame] | 248 | HeapFree(GetProcessHeap(), 0, tempbuf); |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 249 | return hres; |
| 250 | } |
| 251 | |
| 252 | /********************* OLE Proxy/Stub Factory ********************************/ |
| 253 | static HRESULT WINAPI |
| 254 | PSFacBuf_QueryInterface(LPPSFACTORYBUFFER iface, REFIID iid, LPVOID *ppv) { |
| 255 | if (IsEqualIID(iid,&IID_IPSFactoryBuffer)||IsEqualIID(iid,&IID_IUnknown)) { |
Vincent Béron | 9a62491 | 2002-05-31 23:06:46 +0000 | [diff] [blame] | 256 | *ppv = (LPVOID)iface; |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 257 | /* No ref counting, static class */ |
| 258 | return S_OK; |
| 259 | } |
| 260 | FIXME("(%s) unknown IID?\n",debugstr_guid(iid)); |
| 261 | return E_NOINTERFACE; |
| 262 | } |
| 263 | |
| 264 | static ULONG WINAPI PSFacBuf_AddRef(LPPSFACTORYBUFFER iface) { return 2; } |
| 265 | static ULONG WINAPI PSFacBuf_Release(LPPSFACTORYBUFFER iface) { return 1; } |
| 266 | |
| 267 | static HRESULT |
| 268 | _get_typeinfo_for_iid(REFIID riid, ITypeInfo**ti) { |
| 269 | HRESULT hres; |
| 270 | HKEY ikey; |
| 271 | char tlguid[200],typelibkey[300],interfacekey[300],ver[100]; |
| 272 | char tlfn[260]; |
| 273 | OLECHAR tlfnW[260]; |
Mike McCormack | d7b9a23 | 2005-07-06 19:08:05 +0000 | [diff] [blame] | 274 | DWORD tlguidlen, verlen, type; |
| 275 | LONG tlfnlen; |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 276 | ITypeLib *tl; |
| 277 | |
Michael Stefaniuc | 15765dd | 2006-10-15 12:27:34 +0200 | [diff] [blame] | 278 | sprintf( interfacekey, "Interface\\{%08x-%04x-%04x-%02x%02x-%02x%02x%02x%02x%02x%02x}\\Typelib", |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 279 | riid->Data1, riid->Data2, riid->Data3, |
| 280 | riid->Data4[0], riid->Data4[1], riid->Data4[2], riid->Data4[3], |
| 281 | riid->Data4[4], riid->Data4[5], riid->Data4[6], riid->Data4[7] |
| 282 | ); |
| 283 | |
| 284 | if (RegOpenKeyA(HKEY_CLASSES_ROOT,interfacekey,&ikey)) { |
Mike Hearn | 64eac75 | 2005-01-27 11:11:32 +0000 | [diff] [blame] | 285 | ERR("No %s key found.\n",interfacekey); |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 286 | return E_FAIL; |
| 287 | } |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 288 | tlguidlen = sizeof(tlguid); |
Mike McCormack | d7b9a23 | 2005-07-06 19:08:05 +0000 | [diff] [blame] | 289 | if (RegQueryValueExA(ikey,NULL,NULL,&type,(LPBYTE)tlguid,&tlguidlen)) { |
Mike Hearn | 64eac75 | 2005-01-27 11:11:32 +0000 | [diff] [blame] | 290 | ERR("Getting typelib guid failed.\n"); |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 291 | RegCloseKey(ikey); |
| 292 | return E_FAIL; |
| 293 | } |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 294 | verlen = sizeof(ver); |
Mike McCormack | d7b9a23 | 2005-07-06 19:08:05 +0000 | [diff] [blame] | 295 | if (RegQueryValueExA(ikey,"Version",NULL,&type,(LPBYTE)ver,&verlen)) { |
Mike Hearn | 64eac75 | 2005-01-27 11:11:32 +0000 | [diff] [blame] | 296 | ERR("Could not get version value?\n"); |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 297 | RegCloseKey(ikey); |
| 298 | return E_FAIL; |
| 299 | } |
| 300 | RegCloseKey(ikey); |
| 301 | sprintf(typelibkey,"Typelib\\%s\\%s\\0\\win32",tlguid,ver); |
| 302 | tlfnlen = sizeof(tlfn); |
| 303 | if (RegQueryValueA(HKEY_CLASSES_ROOT,typelibkey,tlfn,&tlfnlen)) { |
Mike Hearn | 64eac75 | 2005-01-27 11:11:32 +0000 | [diff] [blame] | 304 | ERR("Could not get typelib fn?\n"); |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 305 | return E_FAIL; |
| 306 | } |
Thomas Weidenmueller | 05d6292 | 2006-05-12 00:07:59 +0200 | [diff] [blame] | 307 | MultiByteToWideChar(CP_ACP, 0, tlfn, -1, tlfnW, sizeof(tlfnW) / sizeof(tlfnW[0])); |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 308 | hres = LoadTypeLib(tlfnW,&tl); |
| 309 | if (hres) { |
| 310 | ERR("Failed to load typelib for %s, but it should be there.\n",debugstr_guid(riid)); |
| 311 | return hres; |
| 312 | } |
| 313 | hres = ITypeLib_GetTypeInfoOfGuid(tl,riid,ti); |
| 314 | if (hres) { |
| 315 | ERR("typelib does not contain info for %s?\n",debugstr_guid(riid)); |
| 316 | ITypeLib_Release(tl); |
| 317 | return hres; |
| 318 | } |
Rob Shearman | d1b15c8 | 2006-10-23 11:38:20 +0100 | [diff] [blame] | 319 | ITypeLib_Release(tl); |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 320 | return hres; |
| 321 | } |
| 322 | |
Huw Davies | c24a53d | 2007-05-17 15:29:41 +0100 | [diff] [blame] | 323 | /* |
| 324 | * Determine the number of functions including all inherited functions. |
| 325 | * Note for non-dual dispinterfaces we simply return the size of IDispatch. |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 326 | */ |
Huw Davies | c24a53d | 2007-05-17 15:29:41 +0100 | [diff] [blame] | 327 | static HRESULT num_of_funcs(ITypeInfo *tinfo, unsigned int *num) |
| 328 | { |
| 329 | HRESULT hres; |
Rob Shearman | ecfe54d | 2007-01-02 15:39:48 +0000 | [diff] [blame] | 330 | TYPEATTR *attr; |
| 331 | ITypeInfo *tinfo2; |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 332 | |
Huw Davies | c24a53d | 2007-05-17 15:29:41 +0100 | [diff] [blame] | 333 | *num = 0; |
Rob Shearman | ecfe54d | 2007-01-02 15:39:48 +0000 | [diff] [blame] | 334 | hres = ITypeInfo_GetTypeAttr(tinfo, &attr); |
| 335 | if (hres) { |
| 336 | ERR("GetTypeAttr failed with %x\n",hres); |
| 337 | return hres; |
| 338 | } |
Huw Davies | c24a53d | 2007-05-17 15:29:41 +0100 | [diff] [blame] | 339 | |
| 340 | if(attr->typekind == TKIND_DISPATCH && (attr->wTypeFlags & TYPEFLAG_FDUAL)) |
| 341 | { |
Rob Shearman | ecfe54d | 2007-01-02 15:39:48 +0000 | [diff] [blame] | 342 | HREFTYPE href; |
Huw Davies | c24a53d | 2007-05-17 15:29:41 +0100 | [diff] [blame] | 343 | hres = ITypeInfo_GetRefTypeOfImplType(tinfo, -1, &href); |
| 344 | if(FAILED(hres)) |
| 345 | { |
| 346 | ERR("Unable to get interface href from dual dispinterface\n"); |
| 347 | goto end; |
Rob Shearman | ecfe54d | 2007-01-02 15:39:48 +0000 | [diff] [blame] | 348 | } |
| 349 | hres = ITypeInfo_GetRefTypeInfo(tinfo, href, &tinfo2); |
Huw Davies | c24a53d | 2007-05-17 15:29:41 +0100 | [diff] [blame] | 350 | if(FAILED(hres)) |
| 351 | { |
| 352 | ERR("Unable to get interface from dual dispinterface\n"); |
| 353 | goto end; |
Rob Shearman | ecfe54d | 2007-01-02 15:39:48 +0000 | [diff] [blame] | 354 | } |
Huw Davies | c24a53d | 2007-05-17 15:29:41 +0100 | [diff] [blame] | 355 | hres = num_of_funcs(tinfo2, num); |
Rob Shearman | ecfe54d | 2007-01-02 15:39:48 +0000 | [diff] [blame] | 356 | ITypeInfo_Release(tinfo2); |
| 357 | } |
Huw Davies | c24a53d | 2007-05-17 15:29:41 +0100 | [diff] [blame] | 358 | else |
| 359 | { |
| 360 | *num = attr->cbSizeVft / 4; |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 361 | } |
Huw Davies | c24a53d | 2007-05-17 15:29:41 +0100 | [diff] [blame] | 362 | |
| 363 | end: |
| 364 | ITypeInfo_ReleaseTypeAttr(tinfo, attr); |
| 365 | return hres; |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 366 | } |
| 367 | |
Robert Shearman | 2a54aac | 2004-09-07 20:42:26 +0000 | [diff] [blame] | 368 | #ifdef __i386__ |
| 369 | |
Alexandre Julliard | e828308 | 2004-01-09 22:18:49 +0000 | [diff] [blame] | 370 | #include "pshpack1.h" |
| 371 | |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 372 | typedef struct _TMAsmProxy { |
| 373 | BYTE popleax; |
| 374 | BYTE pushlval; |
| 375 | BYTE nr; |
| 376 | BYTE pushleax; |
| 377 | BYTE lcall; |
| 378 | DWORD xcall; |
| 379 | BYTE lret; |
| 380 | WORD bytestopop; |
Alexandre Julliard | e828308 | 2004-01-09 22:18:49 +0000 | [diff] [blame] | 381 | } TMAsmProxy; |
| 382 | |
| 383 | #include "poppack.h" |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 384 | |
Robert Shearman | 2a54aac | 2004-09-07 20:42:26 +0000 | [diff] [blame] | 385 | #else /* __i386__ */ |
Pierre d'Herbemont | f26e809 | 2005-10-19 19:06:49 +0000 | [diff] [blame] | 386 | # warning You need to implement stubless proxies for your architecture |
| 387 | typedef struct _TMAsmProxy { |
| 388 | } TMAsmProxy; |
Robert Shearman | 2a54aac | 2004-09-07 20:42:26 +0000 | [diff] [blame] | 389 | #endif |
| 390 | |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 391 | typedef struct _TMProxyImpl { |
Dmitry Timoshkov | eba47f1 | 2005-06-06 19:50:35 +0000 | [diff] [blame] | 392 | LPVOID *lpvtbl; |
| 393 | const IRpcProxyBufferVtbl *lpvtbl2; |
Mike McCormack | fc200d0 | 2005-07-11 10:25:19 +0000 | [diff] [blame] | 394 | LONG ref; |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 395 | |
| 396 | TMAsmProxy *asmstubs; |
| 397 | ITypeInfo* tinfo; |
| 398 | IRpcChannelBuffer* chanbuf; |
| 399 | IID iid; |
Robert Shearman | 2a54aac | 2004-09-07 20:42:26 +0000 | [diff] [blame] | 400 | CRITICAL_SECTION crit; |
Robert Shearman | ac23f2c | 2005-03-11 10:18:31 +0000 | [diff] [blame] | 401 | IUnknown *outerunknown; |
Robert Shearman | 5c2a708 | 2005-10-26 10:12:30 +0000 | [diff] [blame] | 402 | IDispatch *dispatch; |
Robert Shearman | b0218db | 2006-02-07 16:26:23 +0100 | [diff] [blame] | 403 | IRpcProxyBuffer *dispatch_proxy; |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 404 | } TMProxyImpl; |
| 405 | |
| 406 | static HRESULT WINAPI |
Robert Shearman | 2a54aac | 2004-09-07 20:42:26 +0000 | [diff] [blame] | 407 | TMProxyImpl_QueryInterface(LPRPCPROXYBUFFER iface, REFIID riid, LPVOID *ppv) |
| 408 | { |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 409 | TRACE("()\n"); |
| 410 | if (IsEqualIID(riid,&IID_IUnknown)||IsEqualIID(riid,&IID_IRpcProxyBuffer)) { |
Robert Shearman | 2a54aac | 2004-09-07 20:42:26 +0000 | [diff] [blame] | 411 | *ppv = (LPVOID)iface; |
| 412 | IRpcProxyBuffer_AddRef(iface); |
| 413 | return S_OK; |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 414 | } |
| 415 | FIXME("no interface for %s\n",debugstr_guid(riid)); |
| 416 | return E_NOINTERFACE; |
| 417 | } |
| 418 | |
| 419 | static ULONG WINAPI |
Robert Shearman | 2a54aac | 2004-09-07 20:42:26 +0000 | [diff] [blame] | 420 | TMProxyImpl_AddRef(LPRPCPROXYBUFFER iface) |
| 421 | { |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 422 | ICOM_THIS_MULTI(TMProxyImpl,lpvtbl2,iface); |
James Hawkins | ba92d29 | 2005-01-14 16:02:20 +0000 | [diff] [blame] | 423 | ULONG refCount = InterlockedIncrement(&This->ref); |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 424 | |
Hans Leidekker | ce66f5d | 2006-10-12 20:57:10 +0200 | [diff] [blame] | 425 | TRACE("(%p)->(ref before=%u)\n",This, refCount - 1); |
Robert Shearman | 2a54aac | 2004-09-07 20:42:26 +0000 | [diff] [blame] | 426 | |
James Hawkins | ba92d29 | 2005-01-14 16:02:20 +0000 | [diff] [blame] | 427 | return refCount; |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 428 | } |
| 429 | |
| 430 | static ULONG WINAPI |
Robert Shearman | 2a54aac | 2004-09-07 20:42:26 +0000 | [diff] [blame] | 431 | TMProxyImpl_Release(LPRPCPROXYBUFFER iface) |
| 432 | { |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 433 | ICOM_THIS_MULTI(TMProxyImpl,lpvtbl2,iface); |
James Hawkins | ba92d29 | 2005-01-14 16:02:20 +0000 | [diff] [blame] | 434 | ULONG refCount = InterlockedDecrement(&This->ref); |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 435 | |
Hans Leidekker | ce66f5d | 2006-10-12 20:57:10 +0200 | [diff] [blame] | 436 | TRACE("(%p)->(ref before=%u)\n",This, refCount + 1); |
Robert Shearman | 2a54aac | 2004-09-07 20:42:26 +0000 | [diff] [blame] | 437 | |
James Hawkins | ba92d29 | 2005-01-14 16:02:20 +0000 | [diff] [blame] | 438 | if (!refCount) |
Robert Shearman | 2a54aac | 2004-09-07 20:42:26 +0000 | [diff] [blame] | 439 | { |
Robert Shearman | b0218db | 2006-02-07 16:26:23 +0100 | [diff] [blame] | 440 | if (This->dispatch_proxy) IRpcProxyBuffer_Release(This->dispatch_proxy); |
Jan Zerebecki | 1e176b4 | 2007-03-10 22:09:58 +0100 | [diff] [blame] | 441 | This->crit.DebugInfo->Spare[0] = 0; |
Robert Shearman | 2a54aac | 2004-09-07 20:42:26 +0000 | [diff] [blame] | 442 | DeleteCriticalSection(&This->crit); |
| 443 | if (This->chanbuf) IRpcChannelBuffer_Release(This->chanbuf); |
| 444 | VirtualFree(This->asmstubs, 0, MEM_RELEASE); |
Robert Shearman | cc21d23 | 2006-09-27 15:51:26 +0100 | [diff] [blame] | 445 | HeapFree(GetProcessHeap(), 0, This->lpvtbl); |
Rob Shearman | d1b15c8 | 2006-10-23 11:38:20 +0100 | [diff] [blame] | 446 | ITypeInfo_Release(This->tinfo); |
Robert Shearman | 2a54aac | 2004-09-07 20:42:26 +0000 | [diff] [blame] | 447 | CoTaskMemFree(This); |
| 448 | } |
James Hawkins | ba92d29 | 2005-01-14 16:02:20 +0000 | [diff] [blame] | 449 | return refCount; |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 450 | } |
| 451 | |
| 452 | static HRESULT WINAPI |
| 453 | TMProxyImpl_Connect( |
Robert Shearman | 2a54aac | 2004-09-07 20:42:26 +0000 | [diff] [blame] | 454 | LPRPCPROXYBUFFER iface,IRpcChannelBuffer* pRpcChannelBuffer) |
| 455 | { |
| 456 | ICOM_THIS_MULTI(TMProxyImpl, lpvtbl2, iface); |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 457 | |
Robert Shearman | 2a54aac | 2004-09-07 20:42:26 +0000 | [diff] [blame] | 458 | TRACE("(%p)\n", pRpcChannelBuffer); |
| 459 | |
| 460 | EnterCriticalSection(&This->crit); |
| 461 | |
| 462 | IRpcChannelBuffer_AddRef(pRpcChannelBuffer); |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 463 | This->chanbuf = pRpcChannelBuffer; |
Robert Shearman | 2a54aac | 2004-09-07 20:42:26 +0000 | [diff] [blame] | 464 | |
| 465 | LeaveCriticalSection(&This->crit); |
| 466 | |
Robert Shearman | b0218db | 2006-02-07 16:26:23 +0100 | [diff] [blame] | 467 | if (This->dispatch_proxy) |
Rob Shearman | 30f2b44 | 2007-02-06 19:19:56 +0000 | [diff] [blame] | 468 | { |
| 469 | IRpcChannelBuffer *pDelegateChannel; |
| 470 | HRESULT hr = TMarshalDispatchChannel_Create(pRpcChannelBuffer, &This->iid, &pDelegateChannel); |
| 471 | if (FAILED(hr)) |
| 472 | return hr; |
| 473 | return IRpcProxyBuffer_Connect(This->dispatch_proxy, pDelegateChannel); |
| 474 | } |
Robert Shearman | b0218db | 2006-02-07 16:26:23 +0100 | [diff] [blame] | 475 | |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 476 | return S_OK; |
| 477 | } |
| 478 | |
| 479 | static void WINAPI |
Robert Shearman | 2a54aac | 2004-09-07 20:42:26 +0000 | [diff] [blame] | 480 | TMProxyImpl_Disconnect(LPRPCPROXYBUFFER iface) |
| 481 | { |
| 482 | ICOM_THIS_MULTI(TMProxyImpl, lpvtbl2, iface); |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 483 | |
Robert Shearman | 2a54aac | 2004-09-07 20:42:26 +0000 | [diff] [blame] | 484 | TRACE("()\n"); |
| 485 | |
| 486 | EnterCriticalSection(&This->crit); |
| 487 | |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 488 | IRpcChannelBuffer_Release(This->chanbuf); |
| 489 | This->chanbuf = NULL; |
Robert Shearman | 2a54aac | 2004-09-07 20:42:26 +0000 | [diff] [blame] | 490 | |
| 491 | LeaveCriticalSection(&This->crit); |
Robert Shearman | b0218db | 2006-02-07 16:26:23 +0100 | [diff] [blame] | 492 | |
| 493 | if (This->dispatch_proxy) |
| 494 | IRpcProxyBuffer_Disconnect(This->dispatch_proxy); |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 495 | } |
| 496 | |
| 497 | |
Dmitry Timoshkov | eba47f1 | 2005-06-06 19:50:35 +0000 | [diff] [blame] | 498 | static const IRpcProxyBufferVtbl tmproxyvtable = { |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 499 | TMProxyImpl_QueryInterface, |
| 500 | TMProxyImpl_AddRef, |
| 501 | TMProxyImpl_Release, |
| 502 | TMProxyImpl_Connect, |
| 503 | TMProxyImpl_Disconnect |
| 504 | }; |
| 505 | |
Marcus Meissner | e81f10c | 2002-06-10 02:30:20 +0000 | [diff] [blame] | 506 | /* how much space do we use on stack in DWORD steps. */ |
Marcus Meissner | 5539e28 | 2004-09-27 20:33:27 +0000 | [diff] [blame] | 507 | int |
Marcus Meissner | e81f10c | 2002-06-10 02:30:20 +0000 | [diff] [blame] | 508 | _argsize(DWORD vt) { |
| 509 | switch (vt) { |
Marcus Meissner | 2045ed2 | 2005-11-04 11:13:55 +0000 | [diff] [blame] | 510 | case VT_UI8: |
| 511 | return 8/sizeof(DWORD); |
Robert Shearman | 1329f87 | 2005-06-09 09:43:38 +0000 | [diff] [blame] | 512 | case VT_R8: |
| 513 | return sizeof(double)/sizeof(DWORD); |
| 514 | case VT_CY: |
| 515 | return sizeof(CY)/sizeof(DWORD); |
Marcus Meissner | c18a6aa | 2002-06-14 00:36:42 +0000 | [diff] [blame] | 516 | case VT_DATE: |
| 517 | return sizeof(DATE)/sizeof(DWORD); |
Marcus Meissner | e81f10c | 2002-06-10 02:30:20 +0000 | [diff] [blame] | 518 | case VT_VARIANT: |
| 519 | return (sizeof(VARIANT)+3)/sizeof(DWORD); |
| 520 | default: |
| 521 | return 1; |
| 522 | } |
| 523 | } |
| 524 | |
| 525 | static int |
Andrew Talbot | f73821c | 2007-07-25 21:37:23 +0100 | [diff] [blame] | 526 | _xsize(const TYPEDESC *td) { |
Marcus Meissner | e81f10c | 2002-06-10 02:30:20 +0000 | [diff] [blame] | 527 | switch (td->vt) { |
Marcus Meissner | c18a6aa | 2002-06-14 00:36:42 +0000 | [diff] [blame] | 528 | case VT_DATE: |
| 529 | return sizeof(DATE); |
Marcus Meissner | e81f10c | 2002-06-10 02:30:20 +0000 | [diff] [blame] | 530 | case VT_VARIANT: |
| 531 | return sizeof(VARIANT)+3; |
| 532 | case VT_CARRAY: { |
| 533 | int i, arrsize = 1; |
Andrew Talbot | f73821c | 2007-07-25 21:37:23 +0100 | [diff] [blame] | 534 | const ARRAYDESC *adesc = td->u.lpadesc; |
Marcus Meissner | e81f10c | 2002-06-10 02:30:20 +0000 | [diff] [blame] | 535 | |
| 536 | for (i=0;i<adesc->cDims;i++) |
| 537 | arrsize *= adesc->rgbounds[i].cElements; |
| 538 | return arrsize*_xsize(&adesc->tdescElem); |
| 539 | } |
Marcus Meissner | 2045ed2 | 2005-11-04 11:13:55 +0000 | [diff] [blame] | 540 | case VT_UI8: |
| 541 | case VT_I8: |
| 542 | return 8; |
Marcus Meissner | e81f10c | 2002-06-10 02:30:20 +0000 | [diff] [blame] | 543 | case VT_UI2: |
| 544 | case VT_I2: |
| 545 | return 2; |
| 546 | case VT_UI1: |
| 547 | case VT_I1: |
| 548 | return 1; |
| 549 | default: |
| 550 | return 4; |
| 551 | } |
| 552 | } |
| 553 | |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 554 | static HRESULT |
Marcus Meissner | e81f10c | 2002-06-10 02:30:20 +0000 | [diff] [blame] | 555 | serialize_param( |
| 556 | ITypeInfo *tinfo, |
| 557 | BOOL writeit, |
| 558 | BOOL debugout, |
| 559 | BOOL dealloc, |
| 560 | TYPEDESC *tdesc, |
| 561 | DWORD *arg, |
Robert Shearman | 2a54aac | 2004-09-07 20:42:26 +0000 | [diff] [blame] | 562 | marshal_state *buf) |
| 563 | { |
Marcus Meissner | e81f10c | 2002-06-10 02:30:20 +0000 | [diff] [blame] | 564 | HRESULT hres = S_OK; |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 565 | |
Robert Shearman | 799ebfc | 2006-01-27 12:54:03 +0100 | [diff] [blame] | 566 | TRACE("(tdesc.vt %s)\n",debugstr_vt(tdesc->vt)); |
Marcus Meissner | e81f10c | 2002-06-10 02:30:20 +0000 | [diff] [blame] | 567 | |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 568 | switch (tdesc->vt) { |
Marcus Meissner | e81f10c | 2002-06-10 02:30:20 +0000 | [diff] [blame] | 569 | case VT_EMPTY: /* nothing. empty variant for instance */ |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 570 | return S_OK; |
Marcus Meissner | 2045ed2 | 2005-11-04 11:13:55 +0000 | [diff] [blame] | 571 | case VT_I8: |
| 572 | case VT_UI8: |
Robert Shearman | 878af66 | 2006-01-14 17:47:25 +0100 | [diff] [blame] | 573 | case VT_CY: |
Marcus Meissner | 2045ed2 | 2005-11-04 11:13:55 +0000 | [diff] [blame] | 574 | hres = S_OK; |
Francois Gouget | 1f67fa0 | 2006-10-18 21:21:57 +0200 | [diff] [blame] | 575 | if (debugout) TRACE_(olerelay)("%x%x\n",arg[0],arg[1]); |
Marcus Meissner | 2045ed2 | 2005-11-04 11:13:55 +0000 | [diff] [blame] | 576 | if (writeit) |
| 577 | hres = xbuf_add(buf,(LPBYTE)arg,8); |
| 578 | return hres; |
Marcus Meissner | e81f10c | 2002-06-10 02:30:20 +0000 | [diff] [blame] | 579 | case VT_BOOL: |
| 580 | case VT_ERROR: |
Robert Shearman | fd81d9c | 2006-02-07 16:26:02 +0100 | [diff] [blame] | 581 | case VT_INT: |
Marcus Meissner | e81f10c | 2002-06-10 02:30:20 +0000 | [diff] [blame] | 582 | case VT_UINT: |
| 583 | case VT_I4: |
Marcus Meissner | d344849 | 2002-12-05 19:12:50 +0000 | [diff] [blame] | 584 | case VT_R4: |
Alexandre Julliard | 232760d | 2005-03-14 10:09:11 +0000 | [diff] [blame] | 585 | case VT_UI4: |
Marcus Meissner | e81f10c | 2002-06-10 02:30:20 +0000 | [diff] [blame] | 586 | hres = S_OK; |
Francois Gouget | 1f67fa0 | 2006-10-18 21:21:57 +0200 | [diff] [blame] | 587 | if (debugout) TRACE_(olerelay)("%x\n",*arg); |
Marcus Meissner | e81f10c | 2002-06-10 02:30:20 +0000 | [diff] [blame] | 588 | if (writeit) |
| 589 | hres = xbuf_add(buf,(LPBYTE)arg,sizeof(DWORD)); |
| 590 | return hres; |
Marcus Meissner | c91df67 | 2005-03-27 18:17:19 +0000 | [diff] [blame] | 591 | case VT_I2: |
| 592 | case VT_UI2: |
| 593 | hres = S_OK; |
Francois Gouget | 1f67fa0 | 2006-10-18 21:21:57 +0200 | [diff] [blame] | 594 | if (debugout) TRACE_(olerelay)("%04x\n",*arg & 0xffff); |
Marcus Meissner | c91df67 | 2005-03-27 18:17:19 +0000 | [diff] [blame] | 595 | if (writeit) |
| 596 | hres = xbuf_add(buf,(LPBYTE)arg,sizeof(DWORD)); |
| 597 | return hres; |
| 598 | case VT_I1: |
| 599 | case VT_UI1: |
| 600 | hres = S_OK; |
Francois Gouget | 1f67fa0 | 2006-10-18 21:21:57 +0200 | [diff] [blame] | 601 | if (debugout) TRACE_(olerelay)("%02x\n",*arg & 0xff); |
Marcus Meissner | c91df67 | 2005-03-27 18:17:19 +0000 | [diff] [blame] | 602 | if (writeit) |
| 603 | hres = xbuf_add(buf,(LPBYTE)arg,sizeof(DWORD)); |
| 604 | return hres; |
Marcus Meissner | 4b5f3c6 | 2005-03-29 11:29:11 +0000 | [diff] [blame] | 605 | case VT_I4|VT_BYREF: |
| 606 | hres = S_OK; |
Francois Gouget | 1f67fa0 | 2006-10-18 21:21:57 +0200 | [diff] [blame] | 607 | if (debugout) TRACE_(olerelay)("&0x%x\n",*arg); |
Marcus Meissner | 4b5f3c6 | 2005-03-29 11:29:11 +0000 | [diff] [blame] | 608 | if (writeit) |
| 609 | hres = xbuf_add(buf,(LPBYTE)(DWORD*)*arg,sizeof(DWORD)); |
| 610 | /* do not dealloc at this time */ |
| 611 | return hres; |
Marcus Meissner | e81f10c | 2002-06-10 02:30:20 +0000 | [diff] [blame] | 612 | case VT_VARIANT: { |
| 613 | TYPEDESC tdesc2; |
| 614 | VARIANT *vt = (VARIANT*)arg; |
| 615 | DWORD vttype = V_VT(vt); |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 616 | |
Robert Shearman | 799ebfc | 2006-01-27 12:54:03 +0100 | [diff] [blame] | 617 | if (debugout) TRACE_(olerelay)("Vt(%s%s)(",debugstr_vt(vttype),debugstr_vf(vttype)); |
Marcus Meissner | e81f10c | 2002-06-10 02:30:20 +0000 | [diff] [blame] | 618 | tdesc2.vt = vttype; |
| 619 | if (writeit) { |
| 620 | hres = xbuf_add(buf,(LPBYTE)&vttype,sizeof(vttype)); |
| 621 | if (hres) return hres; |
| 622 | } |
| 623 | /* need to recurse since we need to free the stuff */ |
Mike McCormack | a04b77d | 2005-08-08 18:38:29 +0000 | [diff] [blame] | 624 | hres = serialize_param(tinfo,writeit,debugout,dealloc,&tdesc2,(DWORD*)&(V_I4(vt)),buf); |
Mike Hearn | d440ca4 | 2004-06-17 23:03:26 +0000 | [diff] [blame] | 625 | if (debugout) TRACE_(olerelay)(")"); |
Marcus Meissner | e81f10c | 2002-06-10 02:30:20 +0000 | [diff] [blame] | 626 | return hres; |
| 627 | } |
Alexandre Julliard | 232760d | 2005-03-14 10:09:11 +0000 | [diff] [blame] | 628 | case VT_BSTR|VT_BYREF: { |
Marcus Meissner | 4b5f3c6 | 2005-03-29 11:29:11 +0000 | [diff] [blame] | 629 | if (debugout) TRACE_(olerelay)("[byref]'%s'", *(BSTR*)*arg ? relaystr(*((BSTR*)*arg)) : "<bstr NULL>"); |
Alexandre Julliard | 232760d | 2005-03-14 10:09:11 +0000 | [diff] [blame] | 630 | if (writeit) { |
| 631 | /* ptr to ptr to magic widestring, basically */ |
| 632 | BSTR *bstr = (BSTR *) *arg; |
Robert Shearman | d8780c7 | 2005-07-19 19:42:50 +0000 | [diff] [blame] | 633 | DWORD len; |
Marcus Meissner | 4b5f3c6 | 2005-03-29 11:29:11 +0000 | [diff] [blame] | 634 | if (!*bstr) { |
Alexandre Julliard | 232760d | 2005-03-14 10:09:11 +0000 | [diff] [blame] | 635 | /* -1 means "null string" which is equivalent to empty string */ |
Robert Shearman | d8780c7 | 2005-07-19 19:42:50 +0000 | [diff] [blame] | 636 | len = -1; |
| 637 | hres = xbuf_add(buf, (LPBYTE)&len,sizeof(DWORD)); |
| 638 | if (hres) return hres; |
Alexandre Julliard | 232760d | 2005-03-14 10:09:11 +0000 | [diff] [blame] | 639 | } else { |
Robert Shearman | d8780c7 | 2005-07-19 19:42:50 +0000 | [diff] [blame] | 640 | len = *((DWORD*)*bstr-1)/sizeof(WCHAR); |
| 641 | hres = xbuf_add(buf,(LPBYTE)&len,sizeof(DWORD)); |
| 642 | if (hres) return hres; |
| 643 | hres = xbuf_add(buf,(LPBYTE)*bstr,len * sizeof(WCHAR)); |
| 644 | if (hres) return hres; |
Alexandre Julliard | 232760d | 2005-03-14 10:09:11 +0000 | [diff] [blame] | 645 | } |
| 646 | } |
| 647 | |
| 648 | if (dealloc && arg) { |
| 649 | BSTR *str = *((BSTR **)arg); |
| 650 | SysFreeString(*str); |
| 651 | } |
| 652 | return S_OK; |
| 653 | } |
| 654 | |
Marcus Meissner | e81f10c | 2002-06-10 02:30:20 +0000 | [diff] [blame] | 655 | case VT_BSTR: { |
| 656 | if (debugout) { |
Mike Hearn | edcbd13 | 2005-03-27 18:20:10 +0000 | [diff] [blame] | 657 | if (*arg) |
| 658 | TRACE_(olerelay)("%s",relaystr((WCHAR*)*arg)); |
Marcus Meissner | e81f10c | 2002-06-10 02:30:20 +0000 | [diff] [blame] | 659 | else |
Mike Hearn | d440ca4 | 2004-06-17 23:03:26 +0000 | [diff] [blame] | 660 | TRACE_(olerelay)("<bstr NULL>"); |
Marcus Meissner | e81f10c | 2002-06-10 02:30:20 +0000 | [diff] [blame] | 661 | } |
| 662 | if (writeit) { |
Robert Shearman | d8780c7 | 2005-07-19 19:42:50 +0000 | [diff] [blame] | 663 | BSTR bstr = (BSTR)*arg; |
| 664 | DWORD len; |
| 665 | if (!bstr) { |
| 666 | len = -1; |
| 667 | hres = xbuf_add(buf,(LPBYTE)&len,sizeof(DWORD)); |
| 668 | if (hres) return hres; |
Marcus Meissner | e81f10c | 2002-06-10 02:30:20 +0000 | [diff] [blame] | 669 | } else { |
Robert Shearman | d8780c7 | 2005-07-19 19:42:50 +0000 | [diff] [blame] | 670 | len = *((DWORD*)bstr-1)/sizeof(WCHAR); |
| 671 | hres = xbuf_add(buf,(LPBYTE)&len,sizeof(DWORD)); |
| 672 | if (hres) return hres; |
| 673 | hres = xbuf_add(buf,(LPBYTE)bstr,len * sizeof(WCHAR)); |
| 674 | if (hres) return hres; |
Marcus Meissner | 5b63cc2 | 2002-05-06 20:12:11 +0000 | [diff] [blame] | 675 | } |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 676 | } |
Marcus Meissner | 236ad0d | 2003-12-01 22:36:49 +0000 | [diff] [blame] | 677 | |
Marcus Meissner | e81f10c | 2002-06-10 02:30:20 +0000 | [diff] [blame] | 678 | if (dealloc && arg) |
Marcus Meissner | 236ad0d | 2003-12-01 22:36:49 +0000 | [diff] [blame] | 679 | SysFreeString((BSTR)*arg); |
Marcus Meissner | e81f10c | 2002-06-10 02:30:20 +0000 | [diff] [blame] | 680 | return S_OK; |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 681 | } |
Marcus Meissner | e81f10c | 2002-06-10 02:30:20 +0000 | [diff] [blame] | 682 | case VT_PTR: { |
| 683 | DWORD cookie; |
Marcus Meissner | 6f823db | 2005-09-07 09:23:04 +0000 | [diff] [blame] | 684 | BOOL derefhere = TRUE; |
Robert Shearman | 1329f87 | 2005-06-09 09:43:38 +0000 | [diff] [blame] | 685 | |
Marcus Meissner | 6f823db | 2005-09-07 09:23:04 +0000 | [diff] [blame] | 686 | if (tdesc->u.lptdesc->vt == VT_USERDEFINED) { |
| 687 | ITypeInfo *tinfo2; |
| 688 | TYPEATTR *tattr; |
| 689 | |
| 690 | hres = ITypeInfo_GetRefTypeInfo(tinfo,tdesc->u.lptdesc->u.hreftype,&tinfo2); |
| 691 | if (hres) { |
Hans Leidekker | ce66f5d | 2006-10-12 20:57:10 +0200 | [diff] [blame] | 692 | ERR("Could not get typeinfo of hreftype %x for VT_USERDEFINED.\n",tdesc->u.lptdesc->u.hreftype); |
Marcus Meissner | 6f823db | 2005-09-07 09:23:04 +0000 | [diff] [blame] | 693 | return hres; |
| 694 | } |
| 695 | ITypeInfo_GetTypeAttr(tinfo2,&tattr); |
| 696 | switch (tattr->typekind) { |
| 697 | case TKIND_ENUM: /* confirmed */ |
| 698 | case TKIND_RECORD: /* FIXME: mostly untested */ |
| 699 | derefhere=TRUE; |
| 700 | break; |
| 701 | case TKIND_ALIAS: /* FIXME: untested */ |
| 702 | case TKIND_DISPATCH: /* will be done in VT_USERDEFINED case */ |
| 703 | case TKIND_INTERFACE: /* will be done in VT_USERDEFINED case */ |
| 704 | derefhere=FALSE; |
| 705 | break; |
| 706 | default: |
| 707 | FIXME("unhandled switch cases tattr->typekind %d\n", tattr->typekind); |
| 708 | derefhere=FALSE; |
| 709 | break; |
| 710 | } |
Rob Shearman | d1b15c8 | 2006-10-23 11:38:20 +0100 | [diff] [blame] | 711 | ITypeInfo_ReleaseTypeAttr(tinfo, tattr); |
Marcus Meissner | 6f823db | 2005-09-07 09:23:04 +0000 | [diff] [blame] | 712 | ITypeInfo_Release(tinfo2); |
| 713 | } |
Marcus Meissner | e81f10c | 2002-06-10 02:30:20 +0000 | [diff] [blame] | 714 | |
Mike Hearn | d440ca4 | 2004-06-17 23:03:26 +0000 | [diff] [blame] | 715 | if (debugout) TRACE_(olerelay)("*"); |
Marcus Meissner | e6260a3 | 2005-03-27 17:51:59 +0000 | [diff] [blame] | 716 | /* Write always, so the other side knows when it gets a NULL pointer. |
| 717 | */ |
| 718 | cookie = *arg ? 0x42424242 : 0; |
| 719 | hres = xbuf_add(buf,(LPBYTE)&cookie,sizeof(cookie)); |
| 720 | if (hres) |
| 721 | return hres; |
Marcus Meissner | e81f10c | 2002-06-10 02:30:20 +0000 | [diff] [blame] | 722 | if (!*arg) { |
Mike Hearn | d440ca4 | 2004-06-17 23:03:26 +0000 | [diff] [blame] | 723 | if (debugout) TRACE_(olerelay)("NULL"); |
Marcus Meissner | e81f10c | 2002-06-10 02:30:20 +0000 | [diff] [blame] | 724 | return S_OK; |
| 725 | } |
| 726 | hres = serialize_param(tinfo,writeit,debugout,dealloc,tdesc->u.lptdesc,(DWORD*)*arg,buf); |
Robert Shearman | 1329f87 | 2005-06-09 09:43:38 +0000 | [diff] [blame] | 727 | if (derefhere && dealloc) HeapFree(GetProcessHeap(),0,(LPVOID)*arg); |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 728 | return hres; |
Marcus Meissner | e81f10c | 2002-06-10 02:30:20 +0000 | [diff] [blame] | 729 | } |
| 730 | case VT_UNKNOWN: |
Hans Leidekker | ce66f5d | 2006-10-12 20:57:10 +0200 | [diff] [blame] | 731 | if (debugout) TRACE_(olerelay)("unk(0x%x)",*arg); |
Marcus Meissner | e81f10c | 2002-06-10 02:30:20 +0000 | [diff] [blame] | 732 | if (writeit) |
| 733 | hres = _marshal_interface(buf,&IID_IUnknown,(LPUNKNOWN)*arg); |
Robert Shearman | 9030ff2 | 2006-02-14 11:33:03 +0100 | [diff] [blame] | 734 | if (dealloc && *(IUnknown **)arg) |
Robert Shearman | ec5d739 | 2006-02-07 16:25:42 +0100 | [diff] [blame] | 735 | IUnknown_Release((LPUNKNOWN)*arg); |
Marcus Meissner | e81f10c | 2002-06-10 02:30:20 +0000 | [diff] [blame] | 736 | return hres; |
| 737 | case VT_DISPATCH: |
Hans Leidekker | ce66f5d | 2006-10-12 20:57:10 +0200 | [diff] [blame] | 738 | if (debugout) TRACE_(olerelay)("idisp(0x%x)",*arg); |
Marcus Meissner | e81f10c | 2002-06-10 02:30:20 +0000 | [diff] [blame] | 739 | if (writeit) |
| 740 | hres = _marshal_interface(buf,&IID_IDispatch,(LPUNKNOWN)*arg); |
Robert Shearman | 9030ff2 | 2006-02-14 11:33:03 +0100 | [diff] [blame] | 741 | if (dealloc && *(IUnknown **)arg) |
Robert Shearman | ec5d739 | 2006-02-07 16:25:42 +0100 | [diff] [blame] | 742 | IUnknown_Release((LPUNKNOWN)*arg); |
Marcus Meissner | e81f10c | 2002-06-10 02:30:20 +0000 | [diff] [blame] | 743 | return hres; |
| 744 | case VT_VOID: |
Mike Hearn | d440ca4 | 2004-06-17 23:03:26 +0000 | [diff] [blame] | 745 | if (debugout) TRACE_(olerelay)("<void>"); |
Marcus Meissner | e81f10c | 2002-06-10 02:30:20 +0000 | [diff] [blame] | 746 | return S_OK; |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 747 | case VT_USERDEFINED: { |
| 748 | ITypeInfo *tinfo2; |
| 749 | TYPEATTR *tattr; |
| 750 | |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 751 | hres = ITypeInfo_GetRefTypeInfo(tinfo,tdesc->u.hreftype,&tinfo2); |
| 752 | if (hres) { |
Hans Leidekker | ce66f5d | 2006-10-12 20:57:10 +0200 | [diff] [blame] | 753 | ERR("Could not get typeinfo of hreftype %x for VT_USERDEFINED.\n",tdesc->u.hreftype); |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 754 | return hres; |
| 755 | } |
| 756 | ITypeInfo_GetTypeAttr(tinfo2,&tattr); |
| 757 | switch (tattr->typekind) { |
Marcus Meissner | 26ffca4 | 2002-12-12 22:25:07 +0000 | [diff] [blame] | 758 | case TKIND_DISPATCH: |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 759 | case TKIND_INTERFACE: |
Marcus Meissner | e81f10c | 2002-06-10 02:30:20 +0000 | [diff] [blame] | 760 | if (writeit) |
| 761 | hres=_marshal_interface(buf,&(tattr->guid),(LPUNKNOWN)arg); |
Robert Shearman | 4519b1b | 2005-08-24 09:44:41 +0000 | [diff] [blame] | 762 | if (dealloc) |
| 763 | IUnknown_Release((LPUNKNOWN)arg); |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 764 | break; |
Marcus Meissner | e81f10c | 2002-06-10 02:30:20 +0000 | [diff] [blame] | 765 | case TKIND_RECORD: { |
| 766 | int i; |
Mike Hearn | d440ca4 | 2004-06-17 23:03:26 +0000 | [diff] [blame] | 767 | if (debugout) TRACE_(olerelay)("{"); |
Marcus Meissner | e81f10c | 2002-06-10 02:30:20 +0000 | [diff] [blame] | 768 | for (i=0;i<tattr->cVars;i++) { |
| 769 | VARDESC *vdesc; |
| 770 | ELEMDESC *elem2; |
| 771 | TYPEDESC *tdesc2; |
| 772 | |
| 773 | hres = ITypeInfo2_GetVarDesc(tinfo2, i, &vdesc); |
| 774 | if (hres) { |
Mike Hearn | 64eac75 | 2005-01-27 11:11:32 +0000 | [diff] [blame] | 775 | ERR("Could not get vardesc of %d\n",i); |
Marcus Meissner | e81f10c | 2002-06-10 02:30:20 +0000 | [diff] [blame] | 776 | return hres; |
| 777 | } |
Marcus Meissner | e81f10c | 2002-06-10 02:30:20 +0000 | [diff] [blame] | 778 | elem2 = &vdesc->elemdescVar; |
| 779 | tdesc2 = &elem2->tdesc; |
| 780 | hres = serialize_param( |
| 781 | tinfo2, |
| 782 | writeit, |
| 783 | debugout, |
| 784 | dealloc, |
| 785 | tdesc2, |
| 786 | (DWORD*)(((LPBYTE)arg)+vdesc->u.oInst), |
| 787 | buf |
| 788 | ); |
Robert Shearman | c6296c4 | 2005-06-01 11:03:40 +0000 | [diff] [blame] | 789 | ITypeInfo_ReleaseVarDesc(tinfo2, vdesc); |
Marcus Meissner | e81f10c | 2002-06-10 02:30:20 +0000 | [diff] [blame] | 790 | if (hres!=S_OK) |
| 791 | return hres; |
| 792 | if (debugout && (i<(tattr->cVars-1))) |
Mike Hearn | d440ca4 | 2004-06-17 23:03:26 +0000 | [diff] [blame] | 793 | TRACE_(olerelay)(","); |
Marcus Meissner | e81f10c | 2002-06-10 02:30:20 +0000 | [diff] [blame] | 794 | } |
Mike Hearn | d440ca4 | 2004-06-17 23:03:26 +0000 | [diff] [blame] | 795 | if (debugout) TRACE_(olerelay)("}"); |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 796 | break; |
Marcus Meissner | e81f10c | 2002-06-10 02:30:20 +0000 | [diff] [blame] | 797 | } |
Robert Shearman | d2e2e85 | 2005-07-19 19:13:23 +0000 | [diff] [blame] | 798 | case TKIND_ALIAS: |
Rob Shearman | d1b15c8 | 2006-10-23 11:38:20 +0100 | [diff] [blame] | 799 | hres = serialize_param(tinfo2,writeit,debugout,dealloc,&tattr->tdescAlias,arg,buf); |
| 800 | break; |
Robert Shearman | d2e2e85 | 2005-07-19 19:13:23 +0000 | [diff] [blame] | 801 | case TKIND_ENUM: |
| 802 | hres = S_OK; |
Hans Leidekker | ce66f5d | 2006-10-12 20:57:10 +0200 | [diff] [blame] | 803 | if (debugout) TRACE_(olerelay)("%x",*arg); |
Robert Shearman | d2e2e85 | 2005-07-19 19:13:23 +0000 | [diff] [blame] | 804 | if (writeit) |
| 805 | hres = xbuf_add(buf,(LPBYTE)arg,sizeof(DWORD)); |
Rob Shearman | d1b15c8 | 2006-10-23 11:38:20 +0100 | [diff] [blame] | 806 | break; |
Marcus Meissner | 26ffca4 | 2002-12-12 22:25:07 +0000 | [diff] [blame] | 807 | default: |
| 808 | FIXME("Unhandled typekind %d\n",tattr->typekind); |
| 809 | hres = E_FAIL; |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 810 | break; |
| 811 | } |
Rob Shearman | d1b15c8 | 2006-10-23 11:38:20 +0100 | [diff] [blame] | 812 | ITypeInfo_ReleaseTypeAttr(tinfo2, tattr); |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 813 | ITypeInfo_Release(tinfo2); |
| 814 | return hres; |
| 815 | } |
Marcus Meissner | e81f10c | 2002-06-10 02:30:20 +0000 | [diff] [blame] | 816 | case VT_CARRAY: { |
| 817 | ARRAYDESC *adesc = tdesc->u.lpadesc; |
| 818 | int i, arrsize = 1; |
| 819 | |
Mike Hearn | d440ca4 | 2004-06-17 23:03:26 +0000 | [diff] [blame] | 820 | if (debugout) TRACE_(olerelay)("carr"); |
Marcus Meissner | e81f10c | 2002-06-10 02:30:20 +0000 | [diff] [blame] | 821 | for (i=0;i<adesc->cDims;i++) { |
Hans Leidekker | ce66f5d | 2006-10-12 20:57:10 +0200 | [diff] [blame] | 822 | if (debugout) TRACE_(olerelay)("[%d]",adesc->rgbounds[i].cElements); |
Marcus Meissner | e81f10c | 2002-06-10 02:30:20 +0000 | [diff] [blame] | 823 | arrsize *= adesc->rgbounds[i].cElements; |
| 824 | } |
Robert Shearman | 799ebfc | 2006-01-27 12:54:03 +0100 | [diff] [blame] | 825 | if (debugout) TRACE_(olerelay)("(vt %s)",debugstr_vt(adesc->tdescElem.vt)); |
Mike Hearn | d440ca4 | 2004-06-17 23:03:26 +0000 | [diff] [blame] | 826 | if (debugout) TRACE_(olerelay)("["); |
Marcus Meissner | e81f10c | 2002-06-10 02:30:20 +0000 | [diff] [blame] | 827 | for (i=0;i<arrsize;i++) { |
| 828 | hres = serialize_param(tinfo, writeit, debugout, dealloc, &adesc->tdescElem, (DWORD*)((LPBYTE)arg+i*_xsize(&adesc->tdescElem)), buf); |
| 829 | if (hres) |
| 830 | return hres; |
Mike Hearn | d440ca4 | 2004-06-17 23:03:26 +0000 | [diff] [blame] | 831 | if (debugout && (i<arrsize-1)) TRACE_(olerelay)(","); |
Marcus Meissner | e81f10c | 2002-06-10 02:30:20 +0000 | [diff] [blame] | 832 | } |
Mike Hearn | d440ca4 | 2004-06-17 23:03:26 +0000 | [diff] [blame] | 833 | if (debugout) TRACE_(olerelay)("]"); |
Marcus Meissner | e81f10c | 2002-06-10 02:30:20 +0000 | [diff] [blame] | 834 | return S_OK; |
| 835 | } |
Huw Davies | 86d9457 | 2006-05-10 12:11:22 +0100 | [diff] [blame] | 836 | case VT_SAFEARRAY: { |
| 837 | if (writeit) |
| 838 | { |
Michael Stefaniuc | 9d625ad | 2006-11-08 23:46:13 +0100 | [diff] [blame] | 839 | ULONG flags = MAKELONG(MSHCTX_DIFFERENTMACHINE, NDR_LOCAL_DATA_REPRESENTATION); |
| 840 | ULONG size = LPSAFEARRAY_UserSize(&flags, buf->curoff, (LPSAFEARRAY *)arg); |
Huw Davies | 86d9457 | 2006-05-10 12:11:22 +0100 | [diff] [blame] | 841 | xbuf_resize(buf, size); |
| 842 | LPSAFEARRAY_UserMarshal(&flags, buf->base + buf->curoff, (LPSAFEARRAY *)arg); |
| 843 | buf->curoff = size; |
| 844 | } |
| 845 | return S_OK; |
| 846 | } |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 847 | default: |
Marcus Meissner | e81f10c | 2002-06-10 02:30:20 +0000 | [diff] [blame] | 848 | ERR("Unhandled marshal type %d.\n",tdesc->vt); |
| 849 | return S_OK; |
| 850 | } |
| 851 | } |
| 852 | |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 853 | static HRESULT |
Marcus Meissner | e81f10c | 2002-06-10 02:30:20 +0000 | [diff] [blame] | 854 | deserialize_param( |
| 855 | ITypeInfo *tinfo, |
| 856 | BOOL readit, |
| 857 | BOOL debugout, |
| 858 | BOOL alloc, |
| 859 | TYPEDESC *tdesc, |
| 860 | DWORD *arg, |
Robert Shearman | 2a54aac | 2004-09-07 20:42:26 +0000 | [diff] [blame] | 861 | marshal_state *buf) |
| 862 | { |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 863 | HRESULT hres = S_OK; |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 864 | |
Robert Shearman | 799ebfc | 2006-01-27 12:54:03 +0100 | [diff] [blame] | 865 | TRACE("vt %s at %p\n",debugstr_vt(tdesc->vt),arg); |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 866 | |
Marcus Meissner | e81f10c | 2002-06-10 02:30:20 +0000 | [diff] [blame] | 867 | while (1) { |
| 868 | switch (tdesc->vt) { |
| 869 | case VT_EMPTY: |
Francois Gouget | 1f67fa0 | 2006-10-18 21:21:57 +0200 | [diff] [blame] | 870 | if (debugout) TRACE_(olerelay)("<empty>\n"); |
Marcus Meissner | e81f10c | 2002-06-10 02:30:20 +0000 | [diff] [blame] | 871 | return S_OK; |
| 872 | case VT_NULL: |
Francois Gouget | 1f67fa0 | 2006-10-18 21:21:57 +0200 | [diff] [blame] | 873 | if (debugout) TRACE_(olerelay)("<null>\n"); |
Marcus Meissner | e81f10c | 2002-06-10 02:30:20 +0000 | [diff] [blame] | 874 | return S_OK; |
| 875 | case VT_VARIANT: { |
| 876 | VARIANT *vt = (VARIANT*)arg; |
| 877 | |
| 878 | if (readit) { |
| 879 | DWORD vttype; |
| 880 | TYPEDESC tdesc2; |
| 881 | hres = xbuf_get(buf,(LPBYTE)&vttype,sizeof(vttype)); |
| 882 | if (hres) { |
| 883 | FIXME("vt type not read?\n"); |
| 884 | return hres; |
| 885 | } |
| 886 | memset(&tdesc2,0,sizeof(tdesc2)); |
| 887 | tdesc2.vt = vttype; |
| 888 | V_VT(vt) = vttype; |
Robert Shearman | 799ebfc | 2006-01-27 12:54:03 +0100 | [diff] [blame] | 889 | if (debugout) TRACE_(olerelay)("Vt(%s%s)(",debugstr_vt(vttype),debugstr_vf(vttype)); |
Mike McCormack | a04b77d | 2005-08-08 18:38:29 +0000 | [diff] [blame] | 890 | hres = deserialize_param(tinfo, readit, debugout, alloc, &tdesc2, (DWORD*)&(V_I4(vt)), buf); |
Mike Hearn | d440ca4 | 2004-06-17 23:03:26 +0000 | [diff] [blame] | 891 | TRACE_(olerelay)(")"); |
Marcus Meissner | e81f10c | 2002-06-10 02:30:20 +0000 | [diff] [blame] | 892 | return hres; |
| 893 | } else { |
| 894 | VariantInit(vt); |
| 895 | return S_OK; |
| 896 | } |
| 897 | } |
Marcus Meissner | 2045ed2 | 2005-11-04 11:13:55 +0000 | [diff] [blame] | 898 | case VT_I8: |
| 899 | case VT_UI8: |
Robert Shearman | 878af66 | 2006-01-14 17:47:25 +0100 | [diff] [blame] | 900 | case VT_CY: |
Marcus Meissner | 2045ed2 | 2005-11-04 11:13:55 +0000 | [diff] [blame] | 901 | if (readit) { |
| 902 | hres = xbuf_get(buf,(LPBYTE)arg,8); |
| 903 | if (hres) ERR("Failed to read integer 8 byte\n"); |
| 904 | } |
Hans Leidekker | ce66f5d | 2006-10-12 20:57:10 +0200 | [diff] [blame] | 905 | if (debugout) TRACE_(olerelay)("%x%x",arg[0],arg[1]); |
Marcus Meissner | 2045ed2 | 2005-11-04 11:13:55 +0000 | [diff] [blame] | 906 | return hres; |
Marcus Meissner | e81f10c | 2002-06-10 02:30:20 +0000 | [diff] [blame] | 907 | case VT_ERROR: |
Alexandre Julliard | 232760d | 2005-03-14 10:09:11 +0000 | [diff] [blame] | 908 | case VT_BOOL: |
| 909 | case VT_I4: |
Robert Shearman | fd81d9c | 2006-02-07 16:26:02 +0100 | [diff] [blame] | 910 | case VT_INT: |
Alexandre Julliard | 232760d | 2005-03-14 10:09:11 +0000 | [diff] [blame] | 911 | case VT_UINT: |
| 912 | case VT_R4: |
| 913 | case VT_UI4: |
Marcus Meissner | e81f10c | 2002-06-10 02:30:20 +0000 | [diff] [blame] | 914 | if (readit) { |
| 915 | hres = xbuf_get(buf,(LPBYTE)arg,sizeof(DWORD)); |
Mike Hearn | 64eac75 | 2005-01-27 11:11:32 +0000 | [diff] [blame] | 916 | if (hres) ERR("Failed to read integer 4 byte\n"); |
Marcus Meissner | e81f10c | 2002-06-10 02:30:20 +0000 | [diff] [blame] | 917 | } |
Hans Leidekker | ce66f5d | 2006-10-12 20:57:10 +0200 | [diff] [blame] | 918 | if (debugout) TRACE_(olerelay)("%x",*arg); |
Marcus Meissner | e81f10c | 2002-06-10 02:30:20 +0000 | [diff] [blame] | 919 | return hres; |
Marcus Meissner | c91df67 | 2005-03-27 18:17:19 +0000 | [diff] [blame] | 920 | case VT_I2: |
| 921 | case VT_UI2: |
| 922 | if (readit) { |
| 923 | DWORD x; |
| 924 | hres = xbuf_get(buf,(LPBYTE)&x,sizeof(DWORD)); |
| 925 | if (hres) ERR("Failed to read integer 4 byte\n"); |
| 926 | memcpy(arg,&x,2); |
| 927 | } |
Hans Leidekker | ce66f5d | 2006-10-12 20:57:10 +0200 | [diff] [blame] | 928 | if (debugout) TRACE_(olerelay)("%04x",*arg & 0xffff); |
Marcus Meissner | c91df67 | 2005-03-27 18:17:19 +0000 | [diff] [blame] | 929 | return hres; |
| 930 | case VT_I1: |
| 931 | case VT_UI1: |
| 932 | if (readit) { |
| 933 | DWORD x; |
| 934 | hres = xbuf_get(buf,(LPBYTE)&x,sizeof(DWORD)); |
| 935 | if (hres) ERR("Failed to read integer 4 byte\n"); |
| 936 | memcpy(arg,&x,1); |
| 937 | } |
Hans Leidekker | ce66f5d | 2006-10-12 20:57:10 +0200 | [diff] [blame] | 938 | if (debugout) TRACE_(olerelay)("%02x",*arg & 0xff); |
Marcus Meissner | c91df67 | 2005-03-27 18:17:19 +0000 | [diff] [blame] | 939 | return hres; |
Marcus Meissner | 4b5f3c6 | 2005-03-29 11:29:11 +0000 | [diff] [blame] | 940 | case VT_I4|VT_BYREF: |
| 941 | hres = S_OK; |
| 942 | if (alloc) |
| 943 | *arg = (DWORD)HeapAlloc(GetProcessHeap(),HEAP_ZERO_MEMORY,sizeof(DWORD)); |
| 944 | if (readit) { |
| 945 | hres = xbuf_get(buf,(LPBYTE)*arg,sizeof(DWORD)); |
| 946 | if (hres) ERR("Failed to read integer 4 byte\n"); |
| 947 | } |
Hans Leidekker | ce66f5d | 2006-10-12 20:57:10 +0200 | [diff] [blame] | 948 | if (debugout) TRACE_(olerelay)("&0x%x",*(DWORD*)*arg); |
Marcus Meissner | 4b5f3c6 | 2005-03-29 11:29:11 +0000 | [diff] [blame] | 949 | return hres; |
Alexandre Julliard | 232760d | 2005-03-14 10:09:11 +0000 | [diff] [blame] | 950 | case VT_BSTR|VT_BYREF: { |
| 951 | BSTR **bstr = (BSTR **)arg; |
| 952 | WCHAR *str; |
| 953 | DWORD len; |
| 954 | |
| 955 | if (readit) { |
| 956 | hres = xbuf_get(buf,(LPBYTE)&len,sizeof(DWORD)); |
| 957 | if (hres) { |
| 958 | ERR("failed to read bstr klen\n"); |
| 959 | return hres; |
| 960 | } |
| 961 | if (len == -1) { |
Marcus Meissner | 4b5f3c6 | 2005-03-29 11:29:11 +0000 | [diff] [blame] | 962 | *bstr = CoTaskMemAlloc(sizeof(BSTR *)); |
| 963 | **bstr = NULL; |
Alexandre Julliard | 232760d | 2005-03-14 10:09:11 +0000 | [diff] [blame] | 964 | if (debugout) TRACE_(olerelay)("<bstr NULL>"); |
| 965 | } else { |
Robert Shearman | d8780c7 | 2005-07-19 19:42:50 +0000 | [diff] [blame] | 966 | str = HeapAlloc(GetProcessHeap(),HEAP_ZERO_MEMORY,(len+1)*sizeof(WCHAR)); |
| 967 | hres = xbuf_get(buf,(LPBYTE)str,len*sizeof(WCHAR)); |
Alexandre Julliard | 232760d | 2005-03-14 10:09:11 +0000 | [diff] [blame] | 968 | if (hres) { |
| 969 | ERR("Failed to read BSTR.\n"); |
Andrew Talbot | edbcf7c | 2007-10-01 21:33:07 +0100 | [diff] [blame] | 970 | HeapFree(GetProcessHeap(),0,str); |
Alexandre Julliard | 232760d | 2005-03-14 10:09:11 +0000 | [diff] [blame] | 971 | return hres; |
| 972 | } |
| 973 | *bstr = CoTaskMemAlloc(sizeof(BSTR *)); |
| 974 | **bstr = SysAllocStringLen(str,len); |
| 975 | if (debugout) TRACE_(olerelay)("%s",relaystr(str)); |
| 976 | HeapFree(GetProcessHeap(),0,str); |
| 977 | } |
| 978 | } else { |
| 979 | *bstr = NULL; |
| 980 | } |
| 981 | return S_OK; |
| 982 | } |
Marcus Meissner | e81f10c | 2002-06-10 02:30:20 +0000 | [diff] [blame] | 983 | case VT_BSTR: { |
| 984 | WCHAR *str; |
| 985 | DWORD len; |
| 986 | |
| 987 | if (readit) { |
| 988 | hres = xbuf_get(buf,(LPBYTE)&len,sizeof(DWORD)); |
| 989 | if (hres) { |
Mike Hearn | 64eac75 | 2005-01-27 11:11:32 +0000 | [diff] [blame] | 990 | ERR("failed to read bstr klen\n"); |
Marcus Meissner | e81f10c | 2002-06-10 02:30:20 +0000 | [diff] [blame] | 991 | return hres; |
| 992 | } |
| 993 | if (len == -1) { |
| 994 | *arg = 0; |
Mike Hearn | d440ca4 | 2004-06-17 23:03:26 +0000 | [diff] [blame] | 995 | if (debugout) TRACE_(olerelay)("<bstr NULL>"); |
Marcus Meissner | e81f10c | 2002-06-10 02:30:20 +0000 | [diff] [blame] | 996 | } else { |
Robert Shearman | d8780c7 | 2005-07-19 19:42:50 +0000 | [diff] [blame] | 997 | str = HeapAlloc(GetProcessHeap(),HEAP_ZERO_MEMORY,(len+1)*sizeof(WCHAR)); |
| 998 | hres = xbuf_get(buf,(LPBYTE)str,len*sizeof(WCHAR)); |
Marcus Meissner | e81f10c | 2002-06-10 02:30:20 +0000 | [diff] [blame] | 999 | if (hres) { |
Mike Hearn | 64eac75 | 2005-01-27 11:11:32 +0000 | [diff] [blame] | 1000 | ERR("Failed to read BSTR.\n"); |
Andrew Talbot | edbcf7c | 2007-10-01 21:33:07 +0100 | [diff] [blame] | 1001 | HeapFree(GetProcessHeap(),0,str); |
Marcus Meissner | e81f10c | 2002-06-10 02:30:20 +0000 | [diff] [blame] | 1002 | return hres; |
| 1003 | } |
| 1004 | *arg = (DWORD)SysAllocStringLen(str,len); |
Mike Hearn | d440ca4 | 2004-06-17 23:03:26 +0000 | [diff] [blame] | 1005 | if (debugout) TRACE_(olerelay)("%s",relaystr(str)); |
Marcus Meissner | e81f10c | 2002-06-10 02:30:20 +0000 | [diff] [blame] | 1006 | HeapFree(GetProcessHeap(),0,str); |
| 1007 | } |
| 1008 | } else { |
| 1009 | *arg = 0; |
| 1010 | } |
| 1011 | return S_OK; |
| 1012 | } |
| 1013 | case VT_PTR: { |
| 1014 | DWORD cookie; |
Marcus Meissner | 6f823db | 2005-09-07 09:23:04 +0000 | [diff] [blame] | 1015 | BOOL derefhere = TRUE; |
Marcus Meissner | e81f10c | 2002-06-10 02:30:20 +0000 | [diff] [blame] | 1016 | |
Marcus Meissner | 6f823db | 2005-09-07 09:23:04 +0000 | [diff] [blame] | 1017 | if (tdesc->u.lptdesc->vt == VT_USERDEFINED) { |
| 1018 | ITypeInfo *tinfo2; |
| 1019 | TYPEATTR *tattr; |
| 1020 | |
| 1021 | hres = ITypeInfo_GetRefTypeInfo(tinfo,tdesc->u.lptdesc->u.hreftype,&tinfo2); |
| 1022 | if (hres) { |
Hans Leidekker | ce66f5d | 2006-10-12 20:57:10 +0200 | [diff] [blame] | 1023 | ERR("Could not get typeinfo of hreftype %x for VT_USERDEFINED.\n",tdesc->u.lptdesc->u.hreftype); |
Marcus Meissner | 6f823db | 2005-09-07 09:23:04 +0000 | [diff] [blame] | 1024 | return hres; |
| 1025 | } |
| 1026 | ITypeInfo_GetTypeAttr(tinfo2,&tattr); |
| 1027 | switch (tattr->typekind) { |
| 1028 | case TKIND_ENUM: /* confirmed */ |
| 1029 | case TKIND_RECORD: /* FIXME: mostly untested */ |
| 1030 | derefhere=TRUE; |
| 1031 | break; |
| 1032 | case TKIND_ALIAS: /* FIXME: untested */ |
| 1033 | case TKIND_DISPATCH: /* will be done in VT_USERDEFINED case */ |
| 1034 | case TKIND_INTERFACE: /* will be done in VT_USERDEFINED case */ |
| 1035 | derefhere=FALSE; |
| 1036 | break; |
| 1037 | default: |
| 1038 | FIXME("unhandled switch cases tattr->typekind %d\n", tattr->typekind); |
| 1039 | derefhere=FALSE; |
| 1040 | break; |
| 1041 | } |
Rob Shearman | d1b15c8 | 2006-10-23 11:38:20 +0100 | [diff] [blame] | 1042 | ITypeInfo_ReleaseTypeAttr(tinfo2, tattr); |
Marcus Meissner | 6f823db | 2005-09-07 09:23:04 +0000 | [diff] [blame] | 1043 | ITypeInfo_Release(tinfo2); |
| 1044 | } |
Marcus Meissner | e6260a3 | 2005-03-27 17:51:59 +0000 | [diff] [blame] | 1045 | /* read it in all cases, we need to know if we have |
| 1046 | * NULL pointer or not. |
| 1047 | */ |
| 1048 | hres = xbuf_get(buf,(LPBYTE)&cookie,sizeof(cookie)); |
| 1049 | if (hres) { |
| 1050 | ERR("Failed to load pointer cookie.\n"); |
| 1051 | return hres; |
Marcus Meissner | e81f10c | 2002-06-10 02:30:20 +0000 | [diff] [blame] | 1052 | } |
Marcus Meissner | e6260a3 | 2005-03-27 17:51:59 +0000 | [diff] [blame] | 1053 | if (cookie != 0x42424242) { |
| 1054 | /* we read a NULL ptr from the remote side */ |
| 1055 | if (debugout) TRACE_(olerelay)("NULL"); |
| 1056 | *arg = 0; |
| 1057 | return S_OK; |
| 1058 | } |
| 1059 | if (debugout) TRACE_(olerelay)("*"); |
Marcus Meissner | e81f10c | 2002-06-10 02:30:20 +0000 | [diff] [blame] | 1060 | if (alloc) { |
Marcus Meissner | e6260a3 | 2005-03-27 17:51:59 +0000 | [diff] [blame] | 1061 | /* Allocate space for the referenced struct */ |
Marcus Meissner | e81f10c | 2002-06-10 02:30:20 +0000 | [diff] [blame] | 1062 | if (derefhere) |
| 1063 | *arg=(DWORD)HeapAlloc(GetProcessHeap(),HEAP_ZERO_MEMORY,_xsize(tdesc->u.lptdesc)); |
| 1064 | } |
| 1065 | if (derefhere) |
| 1066 | return deserialize_param(tinfo, readit, debugout, alloc, tdesc->u.lptdesc, (LPDWORD)*arg, buf); |
| 1067 | else |
| 1068 | return deserialize_param(tinfo, readit, debugout, alloc, tdesc->u.lptdesc, arg, buf); |
| 1069 | } |
| 1070 | case VT_UNKNOWN: |
| 1071 | /* FIXME: UNKNOWN is unknown ..., but allocate 4 byte for it */ |
| 1072 | if (alloc) |
| 1073 | *arg=(DWORD)HeapAlloc(GetProcessHeap(),HEAP_ZERO_MEMORY,sizeof(DWORD)); |
| 1074 | hres = S_OK; |
| 1075 | if (readit) |
| 1076 | hres = _unmarshal_interface(buf,&IID_IUnknown,(LPUNKNOWN*)arg); |
| 1077 | if (debugout) |
Mike Hearn | d440ca4 | 2004-06-17 23:03:26 +0000 | [diff] [blame] | 1078 | TRACE_(olerelay)("unk(%p)",arg); |
Marcus Meissner | e81f10c | 2002-06-10 02:30:20 +0000 | [diff] [blame] | 1079 | return hres; |
| 1080 | case VT_DISPATCH: |
| 1081 | hres = S_OK; |
| 1082 | if (readit) |
| 1083 | hres = _unmarshal_interface(buf,&IID_IDispatch,(LPUNKNOWN*)arg); |
| 1084 | if (debugout) |
Mike Hearn | d440ca4 | 2004-06-17 23:03:26 +0000 | [diff] [blame] | 1085 | TRACE_(olerelay)("idisp(%p)",arg); |
Marcus Meissner | e81f10c | 2002-06-10 02:30:20 +0000 | [diff] [blame] | 1086 | return hres; |
| 1087 | case VT_VOID: |
Mike Hearn | d440ca4 | 2004-06-17 23:03:26 +0000 | [diff] [blame] | 1088 | if (debugout) TRACE_(olerelay)("<void>"); |
Marcus Meissner | e81f10c | 2002-06-10 02:30:20 +0000 | [diff] [blame] | 1089 | return S_OK; |
| 1090 | case VT_USERDEFINED: { |
| 1091 | ITypeInfo *tinfo2; |
| 1092 | TYPEATTR *tattr; |
| 1093 | |
| 1094 | hres = ITypeInfo_GetRefTypeInfo(tinfo,tdesc->u.hreftype,&tinfo2); |
| 1095 | if (hres) { |
Hans Leidekker | ce66f5d | 2006-10-12 20:57:10 +0200 | [diff] [blame] | 1096 | ERR("Could not get typeinfo of hreftype %x for VT_USERDEFINED.\n",tdesc->u.hreftype); |
Marcus Meissner | e81f10c | 2002-06-10 02:30:20 +0000 | [diff] [blame] | 1097 | return hres; |
| 1098 | } |
| 1099 | hres = ITypeInfo_GetTypeAttr(tinfo2,&tattr); |
| 1100 | if (hres) { |
Mike Hearn | 64eac75 | 2005-01-27 11:11:32 +0000 | [diff] [blame] | 1101 | ERR("Could not get typeattr in VT_USERDEFINED.\n"); |
Marcus Meissner | e81f10c | 2002-06-10 02:30:20 +0000 | [diff] [blame] | 1102 | } else { |
Marcus Meissner | e81f10c | 2002-06-10 02:30:20 +0000 | [diff] [blame] | 1103 | switch (tattr->typekind) { |
Marcus Meissner | 26ffca4 | 2002-12-12 22:25:07 +0000 | [diff] [blame] | 1104 | case TKIND_DISPATCH: |
Marcus Meissner | e81f10c | 2002-06-10 02:30:20 +0000 | [diff] [blame] | 1105 | case TKIND_INTERFACE: |
| 1106 | if (readit) |
| 1107 | hres = _unmarshal_interface(buf,&(tattr->guid),(LPUNKNOWN*)arg); |
| 1108 | break; |
| 1109 | case TKIND_RECORD: { |
| 1110 | int i; |
| 1111 | |
Alexandre Julliard | 232760d | 2005-03-14 10:09:11 +0000 | [diff] [blame] | 1112 | if (alloc) |
| 1113 | *arg = (DWORD)HeapAlloc(GetProcessHeap(),HEAP_ZERO_MEMORY,tattr->cbSizeInstance); |
| 1114 | |
Mike Hearn | d440ca4 | 2004-06-17 23:03:26 +0000 | [diff] [blame] | 1115 | if (debugout) TRACE_(olerelay)("{"); |
Marcus Meissner | e81f10c | 2002-06-10 02:30:20 +0000 | [diff] [blame] | 1116 | for (i=0;i<tattr->cVars;i++) { |
| 1117 | VARDESC *vdesc; |
| 1118 | |
| 1119 | hres = ITypeInfo2_GetVarDesc(tinfo2, i, &vdesc); |
| 1120 | if (hres) { |
Mike Hearn | 64eac75 | 2005-01-27 11:11:32 +0000 | [diff] [blame] | 1121 | ERR("Could not get vardesc of %d\n",i); |
Rob Shearman | d1b15c8 | 2006-10-23 11:38:20 +0100 | [diff] [blame] | 1122 | ITypeInfo_ReleaseTypeAttr(tinfo2, tattr); |
| 1123 | ITypeInfo_Release(tinfo2); |
Marcus Meissner | e81f10c | 2002-06-10 02:30:20 +0000 | [diff] [blame] | 1124 | return hres; |
| 1125 | } |
| 1126 | hres = deserialize_param( |
| 1127 | tinfo2, |
| 1128 | readit, |
| 1129 | debugout, |
| 1130 | alloc, |
| 1131 | &vdesc->elemdescVar.tdesc, |
| 1132 | (DWORD*)(((LPBYTE)*arg)+vdesc->u.oInst), |
| 1133 | buf |
| 1134 | ); |
Robert Shearman | b0d8e06c | 2005-12-06 21:24:19 +0100 | [diff] [blame] | 1135 | ITypeInfo2_ReleaseVarDesc(tinfo2, vdesc); |
Mike Hearn | d440ca4 | 2004-06-17 23:03:26 +0000 | [diff] [blame] | 1136 | if (debugout && (i<tattr->cVars-1)) TRACE_(olerelay)(","); |
Marcus Meissner | e81f10c | 2002-06-10 02:30:20 +0000 | [diff] [blame] | 1137 | } |
Mike Hearn | d440ca4 | 2004-06-17 23:03:26 +0000 | [diff] [blame] | 1138 | if (debugout) TRACE_(olerelay)("}"); |
Marcus Meissner | e81f10c | 2002-06-10 02:30:20 +0000 | [diff] [blame] | 1139 | break; |
| 1140 | } |
Robert Shearman | d2e2e85 | 2005-07-19 19:13:23 +0000 | [diff] [blame] | 1141 | case TKIND_ALIAS: |
Rob Shearman | d1b15c8 | 2006-10-23 11:38:20 +0100 | [diff] [blame] | 1142 | hres = deserialize_param(tinfo2,readit,debugout,alloc,&tattr->tdescAlias,arg,buf); |
| 1143 | break; |
Robert Shearman | d2e2e85 | 2005-07-19 19:13:23 +0000 | [diff] [blame] | 1144 | case TKIND_ENUM: |
| 1145 | if (readit) { |
| 1146 | hres = xbuf_get(buf,(LPBYTE)arg,sizeof(DWORD)); |
| 1147 | if (hres) ERR("Failed to read enum (4 byte)\n"); |
| 1148 | } |
Hans Leidekker | ce66f5d | 2006-10-12 20:57:10 +0200 | [diff] [blame] | 1149 | if (debugout) TRACE_(olerelay)("%x",*arg); |
Rob Shearman | d1b15c8 | 2006-10-23 11:38:20 +0100 | [diff] [blame] | 1150 | break; |
Marcus Meissner | 26ffca4 | 2002-12-12 22:25:07 +0000 | [diff] [blame] | 1151 | default: |
| 1152 | ERR("Unhandled typekind %d\n",tattr->typekind); |
| 1153 | hres = E_FAIL; |
Marcus Meissner | e81f10c | 2002-06-10 02:30:20 +0000 | [diff] [blame] | 1154 | break; |
| 1155 | } |
Rob Shearman | d1b15c8 | 2006-10-23 11:38:20 +0100 | [diff] [blame] | 1156 | ITypeInfo_ReleaseTypeAttr(tinfo2, tattr); |
Marcus Meissner | e81f10c | 2002-06-10 02:30:20 +0000 | [diff] [blame] | 1157 | } |
| 1158 | if (hres) |
Mike Hearn | 64eac75 | 2005-01-27 11:11:32 +0000 | [diff] [blame] | 1159 | ERR("failed to stuballoc in TKIND_RECORD.\n"); |
Marcus Meissner | e81f10c | 2002-06-10 02:30:20 +0000 | [diff] [blame] | 1160 | ITypeInfo_Release(tinfo2); |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 1161 | return hres; |
| 1162 | } |
Marcus Meissner | e81f10c | 2002-06-10 02:30:20 +0000 | [diff] [blame] | 1163 | case VT_CARRAY: { |
| 1164 | /* arg is pointing to the start of the array. */ |
| 1165 | ARRAYDESC *adesc = tdesc->u.lpadesc; |
| 1166 | int arrsize,i; |
| 1167 | arrsize = 1; |
| 1168 | if (adesc->cDims > 1) FIXME("cDims > 1 in VT_CARRAY. Does it work?\n"); |
| 1169 | for (i=0;i<adesc->cDims;i++) |
| 1170 | arrsize *= adesc->rgbounds[i].cElements; |
| 1171 | for (i=0;i<arrsize;i++) |
| 1172 | deserialize_param( |
| 1173 | tinfo, |
| 1174 | readit, |
| 1175 | debugout, |
| 1176 | alloc, |
| 1177 | &adesc->tdescElem, |
| 1178 | (DWORD*)((LPBYTE)(arg)+i*_xsize(&adesc->tdescElem)), |
| 1179 | buf |
| 1180 | ); |
| 1181 | return S_OK; |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 1182 | } |
Huw Davies | 86d9457 | 2006-05-10 12:11:22 +0100 | [diff] [blame] | 1183 | case VT_SAFEARRAY: { |
| 1184 | if (readit) |
| 1185 | { |
Michael Stefaniuc | 9d625ad | 2006-11-08 23:46:13 +0100 | [diff] [blame] | 1186 | ULONG flags = MAKELONG(MSHCTX_DIFFERENTMACHINE, NDR_LOCAL_DATA_REPRESENTATION); |
Huw Davies | 86d9457 | 2006-05-10 12:11:22 +0100 | [diff] [blame] | 1187 | unsigned char *buffer; |
| 1188 | buffer = LPSAFEARRAY_UserUnmarshal(&flags, buf->base + buf->curoff, (LPSAFEARRAY *)arg); |
| 1189 | buf->curoff = buffer - buf->base; |
| 1190 | } |
| 1191 | return S_OK; |
| 1192 | } |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 1193 | default: |
Marcus Meissner | e81f10c | 2002-06-10 02:30:20 +0000 | [diff] [blame] | 1194 | ERR("No handler for VT type %d!\n",tdesc->vt); |
| 1195 | return S_OK; |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 1196 | } |
Marcus Meissner | e81f10c | 2002-06-10 02:30:20 +0000 | [diff] [blame] | 1197 | } |
| 1198 | } |
| 1199 | |
Huw Davies | 294b3eb | 2007-04-26 11:56:44 +0100 | [diff] [blame] | 1200 | /* Retrieves a function's funcdesc, searching back into inherited interfaces. */ |
| 1201 | static HRESULT get_funcdesc(ITypeInfo *tinfo, int iMethod, ITypeInfo **tactual, const FUNCDESC **fdesc, |
| 1202 | BSTR *iname, BSTR *fname, UINT *num) |
Robert Shearman | 2a54aac | 2004-09-07 20:42:26 +0000 | [diff] [blame] | 1203 | { |
Huw Davies | 294b3eb | 2007-04-26 11:56:44 +0100 | [diff] [blame] | 1204 | HRESULT hr; |
| 1205 | UINT i, impl_types; |
| 1206 | UINT inherited_funcs = 0; |
| 1207 | TYPEATTR *attr; |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 1208 | |
| 1209 | if (fname) *fname = NULL; |
| 1210 | if (iname) *iname = NULL; |
Huw Davies | 294b3eb | 2007-04-26 11:56:44 +0100 | [diff] [blame] | 1211 | if (num) *num = 0; |
| 1212 | *tactual = NULL; |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 1213 | |
Huw Davies | 294b3eb | 2007-04-26 11:56:44 +0100 | [diff] [blame] | 1214 | hr = ITypeInfo_GetTypeAttr(tinfo, &attr); |
| 1215 | if (FAILED(hr)) |
| 1216 | { |
| 1217 | ERR("GetTypeAttr failed with %x\n",hr); |
| 1218 | return hr; |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 1219 | } |
Huw Davies | c24a53d | 2007-05-17 15:29:41 +0100 | [diff] [blame] | 1220 | |
| 1221 | if(attr->typekind == TKIND_DISPATCH) |
| 1222 | { |
| 1223 | if(attr->wTypeFlags & TYPEFLAG_FDUAL) |
| 1224 | { |
| 1225 | HREFTYPE href; |
| 1226 | ITypeInfo *tinfo2; |
| 1227 | |
| 1228 | hr = ITypeInfo_GetRefTypeOfImplType(tinfo, -1, &href); |
| 1229 | if(FAILED(hr)) |
| 1230 | { |
| 1231 | ERR("Cannot get interface href from dual dispinterface\n"); |
| 1232 | ITypeInfo_ReleaseTypeAttr(tinfo, attr); |
| 1233 | return hr; |
| 1234 | } |
| 1235 | hr = ITypeInfo_GetRefTypeInfo(tinfo, href, &tinfo2); |
| 1236 | if(FAILED(hr)) |
| 1237 | { |
| 1238 | ERR("Cannot get interface from dual dispinterface\n"); |
| 1239 | ITypeInfo_ReleaseTypeAttr(tinfo, attr); |
| 1240 | return hr; |
| 1241 | } |
| 1242 | hr = get_funcdesc(tinfo2, iMethod, tactual, fdesc, iname, fname, num); |
| 1243 | ITypeInfo_Release(tinfo2); |
| 1244 | ITypeInfo_ReleaseTypeAttr(tinfo, attr); |
| 1245 | return hr; |
| 1246 | } |
| 1247 | ERR("Shouldn't be called with a non-dual dispinterface\n"); |
| 1248 | return E_FAIL; |
| 1249 | } |
| 1250 | |
Huw Davies | 294b3eb | 2007-04-26 11:56:44 +0100 | [diff] [blame] | 1251 | impl_types = attr->cImplTypes; |
| 1252 | ITypeInfo_ReleaseTypeAttr(tinfo, attr); |
| 1253 | |
| 1254 | for (i = 0; i < impl_types; i++) |
| 1255 | { |
| 1256 | HREFTYPE href; |
| 1257 | ITypeInfo *pSubTypeInfo; |
| 1258 | UINT sub_funcs; |
| 1259 | |
| 1260 | hr = ITypeInfo_GetRefTypeOfImplType(tinfo, i, &href); |
| 1261 | if (FAILED(hr)) return hr; |
| 1262 | hr = ITypeInfo_GetRefTypeInfo(tinfo, href, &pSubTypeInfo); |
| 1263 | if (FAILED(hr)) return hr; |
| 1264 | |
| 1265 | hr = get_funcdesc(pSubTypeInfo, iMethod, tactual, fdesc, iname, fname, &sub_funcs); |
| 1266 | inherited_funcs += sub_funcs; |
| 1267 | ITypeInfo_Release(pSubTypeInfo); |
| 1268 | if(SUCCEEDED(hr)) return hr; |
| 1269 | } |
| 1270 | if(iMethod < inherited_funcs) |
| 1271 | { |
| 1272 | ERR("shouldn't be here\n"); |
| 1273 | return E_INVALIDARG; |
| 1274 | } |
| 1275 | |
| 1276 | for(i = inherited_funcs; i <= iMethod; i++) |
| 1277 | { |
| 1278 | hr = ITypeInfoImpl_GetInternalFuncDesc(tinfo, i - inherited_funcs, fdesc); |
| 1279 | if(FAILED(hr)) |
| 1280 | { |
| 1281 | if(num) *num = i; |
| 1282 | return hr; |
| 1283 | } |
| 1284 | } |
| 1285 | |
| 1286 | /* found it. We don't care about num so zero it */ |
| 1287 | if(num) *num = 0; |
| 1288 | *tactual = tinfo; |
| 1289 | ITypeInfo_AddRef(*tactual); |
| 1290 | if (fname) ITypeInfo_GetDocumentation(tinfo,(*fdesc)->memid,fname,NULL,NULL,NULL); |
| 1291 | if (iname) ITypeInfo_GetDocumentation(tinfo,-1,iname,NULL,NULL,NULL); |
| 1292 | return S_OK; |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 1293 | } |
| 1294 | |
Rob Shearman | 0f45b2b | 2007-02-07 19:19:26 +0000 | [diff] [blame] | 1295 | static inline BOOL is_in_elem(const ELEMDESC *elem) |
| 1296 | { |
| 1297 | return (elem->u.paramdesc.wParamFlags & PARAMFLAG_FIN || !elem->u.paramdesc.wParamFlags); |
| 1298 | } |
| 1299 | |
| 1300 | static inline BOOL is_out_elem(const ELEMDESC *elem) |
| 1301 | { |
| 1302 | return (elem->u.paramdesc.wParamFlags & PARAMFLAG_FOUT || !elem->u.paramdesc.wParamFlags); |
| 1303 | } |
| 1304 | |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 1305 | static DWORD |
Robert Shearman | 2a54aac | 2004-09-07 20:42:26 +0000 | [diff] [blame] | 1306 | xCall(LPVOID retptr, int method, TMProxyImpl *tpinfo /*, args */) |
| 1307 | { |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 1308 | DWORD *args = ((DWORD*)&tpinfo)+1, *xargs; |
Robert Shearman | b0d8e06c | 2005-12-06 21:24:19 +0100 | [diff] [blame] | 1309 | const FUNCDESC *fdesc; |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 1310 | HRESULT hres; |
| 1311 | int i, relaydeb = TRACE_ON(olerelay); |
| 1312 | marshal_state buf; |
| 1313 | RPCOLEMESSAGE msg; |
| 1314 | ULONG status; |
| 1315 | BSTR fname,iname; |
| 1316 | BSTR names[10]; |
Mike McCormack | a04b77d | 2005-08-08 18:38:29 +0000 | [diff] [blame] | 1317 | UINT nrofnames; |
Marcus Meissner | 89feaca | 2005-03-28 10:00:49 +0000 | [diff] [blame] | 1318 | DWORD remoteresult = 0; |
Robert Shearman | c6296c4 | 2005-06-01 11:03:40 +0000 | [diff] [blame] | 1319 | ITypeInfo *tinfo; |
Robert Shearman | efe606f | 2005-11-02 20:00:00 +0000 | [diff] [blame] | 1320 | IRpcChannelBuffer *chanbuf; |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 1321 | |
Robert Shearman | 2a54aac | 2004-09-07 20:42:26 +0000 | [diff] [blame] | 1322 | EnterCriticalSection(&tpinfo->crit); |
| 1323 | |
Huw Davies | 294b3eb | 2007-04-26 11:56:44 +0100 | [diff] [blame] | 1324 | hres = get_funcdesc(tpinfo->tinfo,method,&tinfo,&fdesc,&iname,&fname,NULL); |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 1325 | if (hres) { |
Robert Shearman | efe606f | 2005-11-02 20:00:00 +0000 | [diff] [blame] | 1326 | ERR("Did not find typeinfo/funcdesc entry for method %d!\n",method); |
Robert Shearman | 2a54aac | 2004-09-07 20:42:26 +0000 | [diff] [blame] | 1327 | LeaveCriticalSection(&tpinfo->crit); |
Robert Shearman | efe606f | 2005-11-02 20:00:00 +0000 | [diff] [blame] | 1328 | return E_FAIL; |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 1329 | } |
| 1330 | |
Robert Shearman | 6794279 | 2004-12-14 11:44:11 +0000 | [diff] [blame] | 1331 | if (!tpinfo->chanbuf) |
| 1332 | { |
| 1333 | WARN("Tried to use disconnected proxy\n"); |
Robert Shearman | c6296c4 | 2005-06-01 11:03:40 +0000 | [diff] [blame] | 1334 | ITypeInfo_Release(tinfo); |
Robert Shearman | 6794279 | 2004-12-14 11:44:11 +0000 | [diff] [blame] | 1335 | LeaveCriticalSection(&tpinfo->crit); |
| 1336 | return RPC_E_DISCONNECTED; |
| 1337 | } |
Robert Shearman | efe606f | 2005-11-02 20:00:00 +0000 | [diff] [blame] | 1338 | chanbuf = tpinfo->chanbuf; |
| 1339 | IRpcChannelBuffer_AddRef(chanbuf); |
| 1340 | |
| 1341 | LeaveCriticalSection(&tpinfo->crit); |
Robert Shearman | 6794279 | 2004-12-14 11:44:11 +0000 | [diff] [blame] | 1342 | |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 1343 | if (relaydeb) { |
Mike Hearn | 86cc2d5 | 2004-06-23 00:10:29 +0000 | [diff] [blame] | 1344 | TRACE_(olerelay)("->"); |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 1345 | if (iname) |
Mike Hearn | d440ca4 | 2004-06-17 23:03:26 +0000 | [diff] [blame] | 1346 | TRACE_(olerelay)("%s:",relaystr(iname)); |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 1347 | if (fname) |
Mike Hearn | d440ca4 | 2004-06-17 23:03:26 +0000 | [diff] [blame] | 1348 | TRACE_(olerelay)("%s(%d)",relaystr(fname),method); |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 1349 | else |
Mike Hearn | d440ca4 | 2004-06-17 23:03:26 +0000 | [diff] [blame] | 1350 | TRACE_(olerelay)("%d",method); |
| 1351 | TRACE_(olerelay)("("); |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 1352 | } |
Marcus Meissner | e6260a3 | 2005-03-27 17:51:59 +0000 | [diff] [blame] | 1353 | |
| 1354 | if (iname) SysFreeString(iname); |
| 1355 | if (fname) SysFreeString(fname); |
| 1356 | |
| 1357 | memset(&buf,0,sizeof(buf)); |
Marcus Meissner | e6260a3 | 2005-03-27 17:51:59 +0000 | [diff] [blame] | 1358 | |
| 1359 | /* normal typelib driven serializing */ |
| 1360 | |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 1361 | /* Need them for hack below */ |
| 1362 | memset(names,0,sizeof(names)); |
Robert Shearman | c6296c4 | 2005-06-01 11:03:40 +0000 | [diff] [blame] | 1363 | if (ITypeInfo_GetNames(tinfo,fdesc->memid,names,sizeof(names)/sizeof(names[0]),&nrofnames)) |
Marcus Meissner | e81f10c | 2002-06-10 02:30:20 +0000 | [diff] [blame] | 1364 | nrofnames = 0; |
| 1365 | if (nrofnames > sizeof(names)/sizeof(names[0])) |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 1366 | ERR("Need more names!\n"); |
Marcus Meissner | e81f10c | 2002-06-10 02:30:20 +0000 | [diff] [blame] | 1367 | |
Marcus Meissner | e6260a3 | 2005-03-27 17:51:59 +0000 | [diff] [blame] | 1368 | xargs = args; |
| 1369 | for (i=0;i<fdesc->cParams;i++) { |
| 1370 | ELEMDESC *elem = fdesc->lprgelemdescParam+i; |
Marcus Meissner | e6260a3 | 2005-03-27 17:51:59 +0000 | [diff] [blame] | 1371 | if (relaydeb) { |
| 1372 | if (i) TRACE_(olerelay)(","); |
| 1373 | if (i+1<nrofnames && names[i+1]) |
| 1374 | TRACE_(olerelay)("%s=",relaystr(names[i+1])); |
| 1375 | } |
| 1376 | /* No need to marshal other data than FIN and any VT_PTR. */ |
Rob Shearman | 0f45b2b | 2007-02-07 19:19:26 +0000 | [diff] [blame] | 1377 | if (!is_in_elem(elem) && (elem->tdesc.vt != VT_PTR)) { |
Marcus Meissner | e6260a3 | 2005-03-27 17:51:59 +0000 | [diff] [blame] | 1378 | xargs+=_argsize(elem->tdesc.vt); |
| 1379 | if (relaydeb) TRACE_(olerelay)("[out]"); |
| 1380 | continue; |
| 1381 | } |
Robert Shearman | 5c2a708 | 2005-10-26 10:12:30 +0000 | [diff] [blame] | 1382 | hres = serialize_param( |
| 1383 | tinfo, |
Rob Shearman | 0f45b2b | 2007-02-07 19:19:26 +0000 | [diff] [blame] | 1384 | is_in_elem(elem), |
Robert Shearman | 5c2a708 | 2005-10-26 10:12:30 +0000 | [diff] [blame] | 1385 | relaydeb, |
| 1386 | FALSE, |
| 1387 | &elem->tdesc, |
| 1388 | xargs, |
| 1389 | &buf |
| 1390 | ); |
Marcus Meissner | e6260a3 | 2005-03-27 17:51:59 +0000 | [diff] [blame] | 1391 | |
| 1392 | if (hres) { |
Hans Leidekker | ce66f5d | 2006-10-12 20:57:10 +0200 | [diff] [blame] | 1393 | ERR("Failed to serialize param, hres %x\n",hres); |
Marcus Meissner | e6260a3 | 2005-03-27 17:51:59 +0000 | [diff] [blame] | 1394 | break; |
| 1395 | } |
| 1396 | xargs+=_argsize(elem->tdesc.vt); |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 1397 | } |
Mike Hearn | d440ca4 | 2004-06-17 23:03:26 +0000 | [diff] [blame] | 1398 | if (relaydeb) TRACE_(olerelay)(")"); |
Marcus Meissner | e6260a3 | 2005-03-27 17:51:59 +0000 | [diff] [blame] | 1399 | |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 1400 | memset(&msg,0,sizeof(msg)); |
| 1401 | msg.cbBuffer = buf.curoff; |
| 1402 | msg.iMethod = method; |
Robert Shearman | efe606f | 2005-11-02 20:00:00 +0000 | [diff] [blame] | 1403 | hres = IRpcChannelBuffer_GetBuffer(chanbuf,&msg,&(tpinfo->iid)); |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 1404 | if (hres) { |
Hans Leidekker | ce66f5d | 2006-10-12 20:57:10 +0200 | [diff] [blame] | 1405 | ERR("RpcChannelBuffer GetBuffer failed, %x\n",hres); |
Robert Shearman | efe606f | 2005-11-02 20:00:00 +0000 | [diff] [blame] | 1406 | goto exit; |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 1407 | } |
| 1408 | memcpy(msg.Buffer,buf.base,buf.curoff); |
Mike Hearn | d440ca4 | 2004-06-17 23:03:26 +0000 | [diff] [blame] | 1409 | if (relaydeb) TRACE_(olerelay)("\n"); |
Robert Shearman | efe606f | 2005-11-02 20:00:00 +0000 | [diff] [blame] | 1410 | hres = IRpcChannelBuffer_SendReceive(chanbuf,&msg,&status); |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 1411 | if (hres) { |
Hans Leidekker | ce66f5d | 2006-10-12 20:57:10 +0200 | [diff] [blame] | 1412 | ERR("RpcChannelBuffer SendReceive failed, %x\n",hres); |
Robert Shearman | efe606f | 2005-11-02 20:00:00 +0000 | [diff] [blame] | 1413 | goto exit; |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 1414 | } |
Mike Hearn | d440ca4 | 2004-06-17 23:03:26 +0000 | [diff] [blame] | 1415 | |
Hans Leidekker | ce66f5d | 2006-10-12 20:57:10 +0200 | [diff] [blame] | 1416 | if (relaydeb) TRACE_(olerelay)(" status = %08x (",status); |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 1417 | if (buf.base) |
| 1418 | buf.base = HeapReAlloc(GetProcessHeap(),0,buf.base,msg.cbBuffer); |
| 1419 | else |
| 1420 | buf.base = HeapAlloc(GetProcessHeap(),0,msg.cbBuffer); |
| 1421 | buf.size = msg.cbBuffer; |
| 1422 | memcpy(buf.base,msg.Buffer,buf.size); |
| 1423 | buf.curoff = 0; |
Marcus Meissner | e6260a3 | 2005-03-27 17:51:59 +0000 | [diff] [blame] | 1424 | |
Marcus Meissner | e6260a3 | 2005-03-27 17:51:59 +0000 | [diff] [blame] | 1425 | /* generic deserializer using typelib description */ |
| 1426 | xargs = args; |
Marcus Meissner | 89feaca | 2005-03-28 10:00:49 +0000 | [diff] [blame] | 1427 | status = S_OK; |
Marcus Meissner | e6260a3 | 2005-03-27 17:51:59 +0000 | [diff] [blame] | 1428 | for (i=0;i<fdesc->cParams;i++) { |
| 1429 | ELEMDESC *elem = fdesc->lprgelemdescParam+i; |
Marcus Meissner | e81f10c | 2002-06-10 02:30:20 +0000 | [diff] [blame] | 1430 | |
Marcus Meissner | e6260a3 | 2005-03-27 17:51:59 +0000 | [diff] [blame] | 1431 | if (relaydeb) { |
| 1432 | if (i) TRACE_(olerelay)(","); |
| 1433 | if (i+1<nrofnames && names[i+1]) TRACE_(olerelay)("%s=",relaystr(names[i+1])); |
| 1434 | } |
| 1435 | /* No need to marshal other data than FOUT and any VT_PTR */ |
Rob Shearman | 0f45b2b | 2007-02-07 19:19:26 +0000 | [diff] [blame] | 1436 | if (!is_out_elem(elem) && (elem->tdesc.vt != VT_PTR)) { |
Marcus Meissner | e6260a3 | 2005-03-27 17:51:59 +0000 | [diff] [blame] | 1437 | xargs += _argsize(elem->tdesc.vt); |
| 1438 | if (relaydeb) TRACE_(olerelay)("[in]"); |
| 1439 | continue; |
| 1440 | } |
Robert Shearman | 5c2a708 | 2005-10-26 10:12:30 +0000 | [diff] [blame] | 1441 | hres = deserialize_param( |
| 1442 | tinfo, |
Rob Shearman | 0f45b2b | 2007-02-07 19:19:26 +0000 | [diff] [blame] | 1443 | is_out_elem(elem), |
Robert Shearman | 5c2a708 | 2005-10-26 10:12:30 +0000 | [diff] [blame] | 1444 | relaydeb, |
| 1445 | FALSE, |
| 1446 | &(elem->tdesc), |
| 1447 | xargs, |
| 1448 | &buf |
| 1449 | ); |
Marcus Meissner | e6260a3 | 2005-03-27 17:51:59 +0000 | [diff] [blame] | 1450 | if (hres) { |
Hans Leidekker | ce66f5d | 2006-10-12 20:57:10 +0200 | [diff] [blame] | 1451 | ERR("Failed to unmarshall param, hres %x\n",hres); |
Marcus Meissner | e6260a3 | 2005-03-27 17:51:59 +0000 | [diff] [blame] | 1452 | status = hres; |
| 1453 | break; |
| 1454 | } |
| 1455 | xargs += _argsize(elem->tdesc.vt); |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 1456 | } |
Robert Shearman | 5c2a708 | 2005-10-26 10:12:30 +0000 | [diff] [blame] | 1457 | |
Marcus Meissner | 89feaca | 2005-03-28 10:00:49 +0000 | [diff] [blame] | 1458 | hres = xbuf_get(&buf, (LPBYTE)&remoteresult, sizeof(DWORD)); |
| 1459 | if (hres != S_OK) |
Robert Shearman | efe606f | 2005-11-02 20:00:00 +0000 | [diff] [blame] | 1460 | goto exit; |
Hans Leidekker | ce66f5d | 2006-10-12 20:57:10 +0200 | [diff] [blame] | 1461 | if (relaydeb) TRACE_(olerelay)(") = %08x\n", remoteresult); |
Marcus Meissner | 89feaca | 2005-03-28 10:00:49 +0000 | [diff] [blame] | 1462 | |
Robert Shearman | efe606f | 2005-11-02 20:00:00 +0000 | [diff] [blame] | 1463 | hres = remoteresult; |
Marcus Meissner | 89feaca | 2005-03-28 10:00:49 +0000 | [diff] [blame] | 1464 | |
Robert Shearman | efe606f | 2005-11-02 20:00:00 +0000 | [diff] [blame] | 1465 | exit: |
Rob Shearman | 0fa33bb | 2006-10-23 11:29:09 +0100 | [diff] [blame] | 1466 | for (i = 0; i < nrofnames; i++) |
| 1467 | SysFreeString(names[i]); |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 1468 | HeapFree(GetProcessHeap(),0,buf.base); |
Robert Shearman | efe606f | 2005-11-02 20:00:00 +0000 | [diff] [blame] | 1469 | IRpcChannelBuffer_Release(chanbuf); |
Robert Shearman | c6296c4 | 2005-06-01 11:03:40 +0000 | [diff] [blame] | 1470 | ITypeInfo_Release(tinfo); |
Hans Leidekker | ce66f5d | 2006-10-12 20:57:10 +0200 | [diff] [blame] | 1471 | TRACE("-- 0x%08x\n", hres); |
Robert Shearman | efe606f | 2005-11-02 20:00:00 +0000 | [diff] [blame] | 1472 | return hres; |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 1473 | } |
| 1474 | |
Alexandre Julliard | cfc6c48 | 2006-06-10 11:59:33 +0200 | [diff] [blame] | 1475 | static HRESULT WINAPI ProxyIUnknown_QueryInterface(IUnknown *iface, REFIID riid, void **ppv) |
Robert Shearman | ac23f2c | 2005-03-11 10:18:31 +0000 | [diff] [blame] | 1476 | { |
| 1477 | TMProxyImpl *proxy = (TMProxyImpl *)iface; |
| 1478 | |
| 1479 | TRACE("(%s, %p)\n", debugstr_guid(riid), ppv); |
| 1480 | |
| 1481 | if (proxy->outerunknown) |
| 1482 | return IUnknown_QueryInterface(proxy->outerunknown, riid, ppv); |
| 1483 | |
| 1484 | FIXME("No interface\n"); |
| 1485 | return E_NOINTERFACE; |
| 1486 | } |
| 1487 | |
Alexandre Julliard | cfc6c48 | 2006-06-10 11:59:33 +0200 | [diff] [blame] | 1488 | static ULONG WINAPI ProxyIUnknown_AddRef(IUnknown *iface) |
Robert Shearman | ac23f2c | 2005-03-11 10:18:31 +0000 | [diff] [blame] | 1489 | { |
| 1490 | TMProxyImpl *proxy = (TMProxyImpl *)iface; |
| 1491 | |
| 1492 | TRACE("\n"); |
| 1493 | |
| 1494 | if (proxy->outerunknown) |
| 1495 | return IUnknown_AddRef(proxy->outerunknown); |
| 1496 | |
| 1497 | return 2; /* FIXME */ |
| 1498 | } |
| 1499 | |
Alexandre Julliard | cfc6c48 | 2006-06-10 11:59:33 +0200 | [diff] [blame] | 1500 | static ULONG WINAPI ProxyIUnknown_Release(IUnknown *iface) |
Robert Shearman | ac23f2c | 2005-03-11 10:18:31 +0000 | [diff] [blame] | 1501 | { |
| 1502 | TMProxyImpl *proxy = (TMProxyImpl *)iface; |
| 1503 | |
| 1504 | TRACE("\n"); |
| 1505 | |
| 1506 | if (proxy->outerunknown) |
| 1507 | return IUnknown_Release(proxy->outerunknown); |
| 1508 | |
| 1509 | return 1; /* FIXME */ |
| 1510 | } |
| 1511 | |
Robert Shearman | 5c2a708 | 2005-10-26 10:12:30 +0000 | [diff] [blame] | 1512 | static HRESULT WINAPI ProxyIDispatch_GetTypeInfoCount(LPDISPATCH iface, UINT * pctinfo) |
| 1513 | { |
| 1514 | TMProxyImpl *This = (TMProxyImpl *)iface; |
Robert Shearman | 5c2a708 | 2005-10-26 10:12:30 +0000 | [diff] [blame] | 1515 | |
| 1516 | TRACE("(%p)\n", pctinfo); |
| 1517 | |
Robert Shearman | b0218db | 2006-02-07 16:26:23 +0100 | [diff] [blame] | 1518 | return IDispatch_GetTypeInfoCount(This->dispatch, pctinfo); |
Robert Shearman | 5c2a708 | 2005-10-26 10:12:30 +0000 | [diff] [blame] | 1519 | } |
| 1520 | |
| 1521 | static HRESULT WINAPI ProxyIDispatch_GetTypeInfo(LPDISPATCH iface, UINT iTInfo, LCID lcid, ITypeInfo** ppTInfo) |
| 1522 | { |
| 1523 | TMProxyImpl *This = (TMProxyImpl *)iface; |
Robert Shearman | 5c2a708 | 2005-10-26 10:12:30 +0000 | [diff] [blame] | 1524 | |
Hans Leidekker | ce66f5d | 2006-10-12 20:57:10 +0200 | [diff] [blame] | 1525 | TRACE("(%d, %x, %p)\n", iTInfo, lcid, ppTInfo); |
Robert Shearman | 5c2a708 | 2005-10-26 10:12:30 +0000 | [diff] [blame] | 1526 | |
Robert Shearman | b0218db | 2006-02-07 16:26:23 +0100 | [diff] [blame] | 1527 | return IDispatch_GetTypeInfo(This->dispatch, iTInfo, lcid, ppTInfo); |
Robert Shearman | 5c2a708 | 2005-10-26 10:12:30 +0000 | [diff] [blame] | 1528 | } |
| 1529 | |
| 1530 | static HRESULT WINAPI ProxyIDispatch_GetIDsOfNames(LPDISPATCH iface, REFIID riid, LPOLESTR * rgszNames, UINT cNames, LCID lcid, DISPID * rgDispId) |
| 1531 | { |
| 1532 | TMProxyImpl *This = (TMProxyImpl *)iface; |
Robert Shearman | 5c2a708 | 2005-10-26 10:12:30 +0000 | [diff] [blame] | 1533 | |
Hans Leidekker | ce66f5d | 2006-10-12 20:57:10 +0200 | [diff] [blame] | 1534 | TRACE("(%s, %p, %d, 0x%x, %p)\n", debugstr_guid(riid), rgszNames, cNames, lcid, rgDispId); |
Robert Shearman | 5c2a708 | 2005-10-26 10:12:30 +0000 | [diff] [blame] | 1535 | |
Robert Shearman | b0218db | 2006-02-07 16:26:23 +0100 | [diff] [blame] | 1536 | return IDispatch_GetIDsOfNames(This->dispatch, riid, rgszNames, |
| 1537 | cNames, lcid, rgDispId); |
Robert Shearman | 5c2a708 | 2005-10-26 10:12:30 +0000 | [diff] [blame] | 1538 | } |
| 1539 | |
| 1540 | static HRESULT WINAPI ProxyIDispatch_Invoke(LPDISPATCH iface, DISPID dispIdMember, REFIID riid, LCID lcid, |
| 1541 | WORD wFlags, DISPPARAMS * pDispParams, VARIANT * pVarResult, |
| 1542 | EXCEPINFO * pExcepInfo, UINT * puArgErr) |
| 1543 | { |
| 1544 | TMProxyImpl *This = (TMProxyImpl *)iface; |
Robert Shearman | 5c2a708 | 2005-10-26 10:12:30 +0000 | [diff] [blame] | 1545 | |
Hans Leidekker | ce66f5d | 2006-10-12 20:57:10 +0200 | [diff] [blame] | 1546 | TRACE("(%d, %s, 0x%x, 0x%x, %p, %p, %p, %p)\n", dispIdMember, |
Robert Shearman | b0218db | 2006-02-07 16:26:23 +0100 | [diff] [blame] | 1547 | debugstr_guid(riid), lcid, wFlags, pDispParams, pVarResult, |
| 1548 | pExcepInfo, puArgErr); |
Robert Shearman | 5c2a708 | 2005-10-26 10:12:30 +0000 | [diff] [blame] | 1549 | |
Robert Shearman | b0218db | 2006-02-07 16:26:23 +0100 | [diff] [blame] | 1550 | return IDispatch_Invoke(This->dispatch, dispIdMember, riid, lcid, |
| 1551 | wFlags, pDispParams, pVarResult, pExcepInfo, |
| 1552 | puArgErr); |
| 1553 | } |
Robert Shearman | 5c2a708 | 2005-10-26 10:12:30 +0000 | [diff] [blame] | 1554 | |
Rob Shearman | 30f2b44 | 2007-02-06 19:19:56 +0000 | [diff] [blame] | 1555 | typedef struct |
| 1556 | { |
| 1557 | const IRpcChannelBufferVtbl *lpVtbl; |
| 1558 | LONG refs; |
| 1559 | /* the IDispatch-derived interface we are handling */ |
| 1560 | IID tmarshal_iid; |
| 1561 | IRpcChannelBuffer *pDelegateChannel; |
| 1562 | } TMarshalDispatchChannel; |
| 1563 | |
| 1564 | static HRESULT WINAPI TMarshalDispatchChannel_QueryInterface(LPRPCCHANNELBUFFER iface, REFIID riid, LPVOID *ppv) |
| 1565 | { |
| 1566 | *ppv = NULL; |
| 1567 | if (IsEqualIID(riid,&IID_IRpcChannelBuffer) || IsEqualIID(riid,&IID_IUnknown)) |
| 1568 | { |
| 1569 | *ppv = (LPVOID)iface; |
| 1570 | IUnknown_AddRef(iface); |
| 1571 | return S_OK; |
| 1572 | } |
| 1573 | return E_NOINTERFACE; |
| 1574 | } |
| 1575 | |
| 1576 | static ULONG WINAPI TMarshalDispatchChannel_AddRef(LPRPCCHANNELBUFFER iface) |
| 1577 | { |
| 1578 | TMarshalDispatchChannel *This = (TMarshalDispatchChannel *)iface; |
| 1579 | return InterlockedIncrement(&This->refs); |
| 1580 | } |
| 1581 | |
| 1582 | static ULONG WINAPI TMarshalDispatchChannel_Release(LPRPCCHANNELBUFFER iface) |
| 1583 | { |
| 1584 | TMarshalDispatchChannel *This = (TMarshalDispatchChannel *)iface; |
| 1585 | ULONG ref; |
| 1586 | |
| 1587 | ref = InterlockedDecrement(&This->refs); |
| 1588 | if (ref) |
| 1589 | return ref; |
| 1590 | |
| 1591 | IRpcChannelBuffer_Release(This->pDelegateChannel); |
| 1592 | HeapFree(GetProcessHeap(), 0, This); |
| 1593 | return 0; |
| 1594 | } |
| 1595 | |
| 1596 | static HRESULT WINAPI TMarshalDispatchChannel_GetBuffer(LPRPCCHANNELBUFFER iface, RPCOLEMESSAGE* olemsg, REFIID riid) |
| 1597 | { |
| 1598 | TMarshalDispatchChannel *This = (TMarshalDispatchChannel *)iface; |
| 1599 | TRACE("(%p, %s)\n", olemsg, debugstr_guid(riid)); |
| 1600 | /* Note: we are pretending to invoke a method on the interface identified |
| 1601 | * by tmarshal_iid so that we can re-use the IDispatch proxy/stub code |
| 1602 | * without the RPC runtime getting confused by not exporting an IDispatch interface */ |
| 1603 | return IRpcChannelBuffer_GetBuffer(This->pDelegateChannel, olemsg, &This->tmarshal_iid); |
| 1604 | } |
| 1605 | |
| 1606 | static HRESULT WINAPI TMarshalDispatchChannel_SendReceive(LPRPCCHANNELBUFFER iface, RPCOLEMESSAGE *olemsg, ULONG *pstatus) |
| 1607 | { |
| 1608 | TMarshalDispatchChannel *This = (TMarshalDispatchChannel *)iface; |
| 1609 | TRACE("(%p, %p)\n", olemsg, pstatus); |
| 1610 | return IRpcChannelBuffer_SendReceive(This->pDelegateChannel, olemsg, pstatus); |
| 1611 | } |
| 1612 | |
| 1613 | static HRESULT WINAPI TMarshalDispatchChannel_FreeBuffer(LPRPCCHANNELBUFFER iface, RPCOLEMESSAGE* olemsg) |
| 1614 | { |
| 1615 | TMarshalDispatchChannel *This = (TMarshalDispatchChannel *)iface; |
| 1616 | TRACE("(%p)\n", olemsg); |
| 1617 | return IRpcChannelBuffer_FreeBuffer(This->pDelegateChannel, olemsg); |
| 1618 | } |
| 1619 | |
| 1620 | static HRESULT WINAPI TMarshalDispatchChannel_GetDestCtx(LPRPCCHANNELBUFFER iface, DWORD* pdwDestContext, void** ppvDestContext) |
| 1621 | { |
| 1622 | TMarshalDispatchChannel *This = (TMarshalDispatchChannel *)iface; |
| 1623 | TRACE("(%p,%p)\n", pdwDestContext, ppvDestContext); |
| 1624 | return IRpcChannelBuffer_GetDestCtx(This->pDelegateChannel, pdwDestContext, ppvDestContext); |
| 1625 | } |
| 1626 | |
| 1627 | static HRESULT WINAPI TMarshalDispatchChannel_IsConnected(LPRPCCHANNELBUFFER iface) |
| 1628 | { |
| 1629 | TMarshalDispatchChannel *This = (TMarshalDispatchChannel *)iface; |
| 1630 | TRACE("()\n"); |
| 1631 | return IRpcChannelBuffer_IsConnected(This->pDelegateChannel); |
| 1632 | } |
| 1633 | |
| 1634 | static const IRpcChannelBufferVtbl TMarshalDispatchChannelVtbl = |
| 1635 | { |
| 1636 | TMarshalDispatchChannel_QueryInterface, |
| 1637 | TMarshalDispatchChannel_AddRef, |
| 1638 | TMarshalDispatchChannel_Release, |
| 1639 | TMarshalDispatchChannel_GetBuffer, |
| 1640 | TMarshalDispatchChannel_SendReceive, |
| 1641 | TMarshalDispatchChannel_FreeBuffer, |
| 1642 | TMarshalDispatchChannel_GetDestCtx, |
| 1643 | TMarshalDispatchChannel_IsConnected |
| 1644 | }; |
| 1645 | |
| 1646 | static HRESULT TMarshalDispatchChannel_Create( |
| 1647 | IRpcChannelBuffer *pDelegateChannel, REFIID tmarshal_riid, |
| 1648 | IRpcChannelBuffer **ppChannel) |
| 1649 | { |
| 1650 | TMarshalDispatchChannel *This = HeapAlloc(GetProcessHeap(), 0, sizeof(*This)); |
| 1651 | if (!This) |
| 1652 | return E_OUTOFMEMORY; |
| 1653 | |
| 1654 | This->lpVtbl = &TMarshalDispatchChannelVtbl; |
| 1655 | This->refs = 1; |
| 1656 | IRpcChannelBuffer_AddRef(pDelegateChannel); |
| 1657 | This->pDelegateChannel = pDelegateChannel; |
| 1658 | This->tmarshal_iid = *tmarshal_riid; |
| 1659 | |
| 1660 | *ppChannel = (IRpcChannelBuffer *)&This->lpVtbl; |
| 1661 | return S_OK; |
| 1662 | } |
| 1663 | |
| 1664 | |
Robert Shearman | b0218db | 2006-02-07 16:26:23 +0100 | [diff] [blame] | 1665 | static inline HRESULT get_facbuf_for_iid(REFIID riid, IPSFactoryBuffer **facbuf) |
| 1666 | { |
| 1667 | HRESULT hr; |
| 1668 | CLSID clsid; |
| 1669 | |
| 1670 | if ((hr = CoGetPSClsid(riid, &clsid))) |
| 1671 | return hr; |
| 1672 | return CoGetClassObject(&clsid, CLSCTX_INPROC_SERVER, NULL, |
| 1673 | &IID_IPSFactoryBuffer, (LPVOID*)facbuf); |
Robert Shearman | 5c2a708 | 2005-10-26 10:12:30 +0000 | [diff] [blame] | 1674 | } |
| 1675 | |
Huw Davies | f1da672 | 2007-05-17 15:23:51 +0100 | [diff] [blame] | 1676 | static HRESULT init_proxy_entry_point(TMProxyImpl *proxy, unsigned int num) |
| 1677 | { |
| 1678 | int j; |
| 1679 | /* nrofargs without This */ |
| 1680 | int nrofargs; |
| 1681 | ITypeInfo *tinfo2; |
| 1682 | TMAsmProxy *xasm = proxy->asmstubs + num; |
| 1683 | HRESULT hres; |
| 1684 | const FUNCDESC *fdesc; |
| 1685 | |
| 1686 | hres = get_funcdesc(proxy->tinfo, num, &tinfo2, &fdesc, NULL, NULL, NULL); |
| 1687 | if (hres) { |
| 1688 | ERR("GetFuncDesc %x should not fail here.\n",hres); |
| 1689 | return hres; |
| 1690 | } |
| 1691 | ITypeInfo_Release(tinfo2); |
| 1692 | /* some args take more than 4 byte on the stack */ |
| 1693 | nrofargs = 0; |
| 1694 | for (j=0;j<fdesc->cParams;j++) |
| 1695 | nrofargs += _argsize(fdesc->lprgelemdescParam[j].tdesc.vt); |
| 1696 | |
| 1697 | #ifdef __i386__ |
| 1698 | if (fdesc->callconv != CC_STDCALL) { |
| 1699 | ERR("calling convention is not stdcall????\n"); |
| 1700 | return E_FAIL; |
| 1701 | } |
| 1702 | /* popl %eax - return ptr |
| 1703 | * pushl <nr> |
| 1704 | * pushl %eax |
| 1705 | * call xCall |
| 1706 | * lret <nr> (+4) |
| 1707 | * |
| 1708 | * |
| 1709 | * arg3 arg2 arg1 <method> <returnptr> |
| 1710 | */ |
| 1711 | xasm->popleax = 0x58; |
| 1712 | xasm->pushlval = 0x6a; |
| 1713 | xasm->nr = num; |
| 1714 | xasm->pushleax = 0x50; |
| 1715 | xasm->lcall = 0xe8; /* relative jump */ |
| 1716 | xasm->xcall = (DWORD)xCall; |
| 1717 | xasm->xcall -= (DWORD)&(xasm->lret); |
| 1718 | xasm->lret = 0xc2; |
| 1719 | xasm->bytestopop = (nrofargs+2)*4; /* pop args, This, iMethod */ |
| 1720 | proxy->lpvtbl[num] = xasm; |
| 1721 | #else |
| 1722 | FIXME("not implemented on non i386\n"); |
| 1723 | return E_FAIL; |
| 1724 | #endif |
| 1725 | return S_OK; |
| 1726 | } |
| 1727 | |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 1728 | static HRESULT WINAPI |
| 1729 | PSFacBuf_CreateProxy( |
| 1730 | LPPSFACTORYBUFFER iface, IUnknown* pUnkOuter, REFIID riid, |
Robert Shearman | 2a54aac | 2004-09-07 20:42:26 +0000 | [diff] [blame] | 1731 | IRpcProxyBuffer **ppProxy, LPVOID *ppv) |
| 1732 | { |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 1733 | HRESULT hres; |
| 1734 | ITypeInfo *tinfo; |
Huw Davies | c24a53d | 2007-05-17 15:29:41 +0100 | [diff] [blame] | 1735 | unsigned int i, nroffuncs; |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 1736 | TMProxyImpl *proxy; |
Robert Shearman | 5c2a708 | 2005-10-26 10:12:30 +0000 | [diff] [blame] | 1737 | TYPEATTR *typeattr; |
Huw Davies | c24a53d | 2007-05-17 15:29:41 +0100 | [diff] [blame] | 1738 | BOOL defer_to_dispatch = FALSE; |
Vincent Béron | 9a62491 | 2002-05-31 23:06:46 +0000 | [diff] [blame] | 1739 | |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 1740 | TRACE("(...%s...)\n",debugstr_guid(riid)); |
| 1741 | hres = _get_typeinfo_for_iid(riid,&tinfo); |
| 1742 | if (hres) { |
Mike Hearn | 64eac75 | 2005-01-27 11:11:32 +0000 | [diff] [blame] | 1743 | ERR("No typeinfo for %s?\n",debugstr_guid(riid)); |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 1744 | return hres; |
| 1745 | } |
Huw Davies | c24a53d | 2007-05-17 15:29:41 +0100 | [diff] [blame] | 1746 | |
| 1747 | hres = num_of_funcs(tinfo, &nroffuncs); |
| 1748 | if (FAILED(hres)) { |
| 1749 | ERR("Cannot get number of functions for typeinfo %s\n",debugstr_guid(riid)); |
| 1750 | ITypeInfo_Release(tinfo); |
| 1751 | return hres; |
| 1752 | } |
| 1753 | |
Robert Shearman | 2a54aac | 2004-09-07 20:42:26 +0000 | [diff] [blame] | 1754 | proxy = CoTaskMemAlloc(sizeof(TMProxyImpl)); |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 1755 | if (!proxy) return E_OUTOFMEMORY; |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 1756 | |
| 1757 | assert(sizeof(TMAsmProxy) == 12); |
| 1758 | |
Robert Shearman | 5c2a708 | 2005-10-26 10:12:30 +0000 | [diff] [blame] | 1759 | proxy->dispatch = NULL; |
Robert Shearman | b0218db | 2006-02-07 16:26:23 +0100 | [diff] [blame] | 1760 | proxy->dispatch_proxy = NULL; |
Robert Shearman | ac23f2c | 2005-03-11 10:18:31 +0000 | [diff] [blame] | 1761 | proxy->outerunknown = pUnkOuter; |
Mike Hearn | 3186bd2 | 2004-06-28 23:51:24 +0000 | [diff] [blame] | 1762 | proxy->asmstubs = VirtualAlloc(NULL, sizeof(TMAsmProxy) * nroffuncs, MEM_COMMIT, PAGE_EXECUTE_READWRITE); |
| 1763 | if (!proxy->asmstubs) { |
| 1764 | ERR("Could not commit pages for proxy thunks\n"); |
Robert Shearman | 2a54aac | 2004-09-07 20:42:26 +0000 | [diff] [blame] | 1765 | CoTaskMemFree(proxy); |
Mike Hearn | 3186bd2 | 2004-06-28 23:51:24 +0000 | [diff] [blame] | 1766 | return E_OUTOFMEMORY; |
| 1767 | } |
Rob Shearman | ecfe54d | 2007-01-02 15:39:48 +0000 | [diff] [blame] | 1768 | proxy->lpvtbl2 = &tmproxyvtable; |
| 1769 | /* one reference for the proxy */ |
| 1770 | proxy->ref = 1; |
| 1771 | proxy->tinfo = tinfo; |
| 1772 | memcpy(&proxy->iid,riid,sizeof(*riid)); |
| 1773 | proxy->chanbuf = 0; |
Mike Hearn | 3186bd2 | 2004-06-28 23:51:24 +0000 | [diff] [blame] | 1774 | |
Robert Shearman | 2a54aac | 2004-09-07 20:42:26 +0000 | [diff] [blame] | 1775 | InitializeCriticalSection(&proxy->crit); |
Jan Zerebecki | 1e176b4 | 2007-03-10 22:09:58 +0100 | [diff] [blame] | 1776 | proxy->crit.DebugInfo->Spare[0] = (DWORD_PTR)(__FILE__ ": TMProxyImpl.crit"); |
Robert Shearman | 2a54aac | 2004-09-07 20:42:26 +0000 | [diff] [blame] | 1777 | |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 1778 | proxy->lpvtbl = HeapAlloc(GetProcessHeap(),0,sizeof(LPBYTE)*nroffuncs); |
Robert Shearman | 5c2a708 | 2005-10-26 10:12:30 +0000 | [diff] [blame] | 1779 | |
| 1780 | /* if we derive from IDispatch then defer to its proxy for its methods */ |
| 1781 | hres = ITypeInfo_GetTypeAttr(tinfo, &typeattr); |
| 1782 | if (hres == S_OK) |
| 1783 | { |
| 1784 | if (typeattr->wTypeFlags & TYPEFLAG_FDISPATCHABLE) |
| 1785 | { |
Robert Shearman | b0218db | 2006-02-07 16:26:23 +0100 | [diff] [blame] | 1786 | IPSFactoryBuffer *factory_buffer; |
| 1787 | hres = get_facbuf_for_iid(&IID_IDispatch, &factory_buffer); |
| 1788 | if (hres == S_OK) |
| 1789 | { |
| 1790 | hres = IPSFactoryBuffer_CreateProxy(factory_buffer, NULL, |
| 1791 | &IID_IDispatch, &proxy->dispatch_proxy, |
| 1792 | (void **)&proxy->dispatch); |
| 1793 | IPSFactoryBuffer_Release(factory_buffer); |
| 1794 | } |
Rob Shearman | ecfe54d | 2007-01-02 15:39:48 +0000 | [diff] [blame] | 1795 | if ((hres == S_OK) && (nroffuncs < 7)) |
| 1796 | { |
| 1797 | ERR("nroffuncs calculated incorrectly (%d)\n", nroffuncs); |
| 1798 | hres = E_UNEXPECTED; |
| 1799 | } |
Robert Shearman | b0218db | 2006-02-07 16:26:23 +0100 | [diff] [blame] | 1800 | if (hres == S_OK) |
| 1801 | { |
Huw Davies | c24a53d | 2007-05-17 15:29:41 +0100 | [diff] [blame] | 1802 | defer_to_dispatch = TRUE; |
Robert Shearman | b0218db | 2006-02-07 16:26:23 +0100 | [diff] [blame] | 1803 | } |
Robert Shearman | 5c2a708 | 2005-10-26 10:12:30 +0000 | [diff] [blame] | 1804 | } |
| 1805 | ITypeInfo_ReleaseTypeAttr(tinfo, typeattr); |
| 1806 | } |
| 1807 | |
Huw Davies | c24a53d | 2007-05-17 15:29:41 +0100 | [diff] [blame] | 1808 | for (i=0;i<nroffuncs;i++) { |
| 1809 | switch (i) { |
| 1810 | case 0: |
| 1811 | proxy->lpvtbl[i] = ProxyIUnknown_QueryInterface; |
| 1812 | break; |
| 1813 | case 1: |
| 1814 | proxy->lpvtbl[i] = ProxyIUnknown_AddRef; |
| 1815 | break; |
| 1816 | case 2: |
| 1817 | proxy->lpvtbl[i] = ProxyIUnknown_Release; |
| 1818 | break; |
| 1819 | case 3: |
| 1820 | if(!defer_to_dispatch) |
| 1821 | { |
| 1822 | hres = init_proxy_entry_point(proxy, i); |
| 1823 | if(FAILED(hres)) return hres; |
| 1824 | } |
| 1825 | else proxy->lpvtbl[3] = ProxyIDispatch_GetTypeInfoCount; |
| 1826 | break; |
| 1827 | case 4: |
| 1828 | if(!defer_to_dispatch) |
| 1829 | { |
| 1830 | hres = init_proxy_entry_point(proxy, i); |
| 1831 | if(FAILED(hres)) return hres; |
| 1832 | } |
| 1833 | else proxy->lpvtbl[4] = ProxyIDispatch_GetTypeInfo; |
| 1834 | break; |
| 1835 | case 5: |
| 1836 | if(!defer_to_dispatch) |
| 1837 | { |
| 1838 | hres = init_proxy_entry_point(proxy, i); |
| 1839 | if(FAILED(hres)) return hres; |
| 1840 | } |
| 1841 | else proxy->lpvtbl[5] = ProxyIDispatch_GetIDsOfNames; |
| 1842 | break; |
| 1843 | case 6: |
| 1844 | if(!defer_to_dispatch) |
| 1845 | { |
| 1846 | hres = init_proxy_entry_point(proxy, i); |
| 1847 | if(FAILED(hres)) return hres; |
| 1848 | } |
| 1849 | else proxy->lpvtbl[6] = ProxyIDispatch_Invoke; |
| 1850 | break; |
| 1851 | default: |
| 1852 | hres = init_proxy_entry_point(proxy, i); |
| 1853 | if(FAILED(hres)) return hres; |
| 1854 | } |
| 1855 | } |
| 1856 | |
Robert Shearman | b0218db | 2006-02-07 16:26:23 +0100 | [diff] [blame] | 1857 | if (hres == S_OK) |
| 1858 | { |
| 1859 | *ppv = (LPVOID)proxy; |
| 1860 | *ppProxy = (IRpcProxyBuffer *)&(proxy->lpvtbl2); |
| 1861 | IUnknown_AddRef((IUnknown *)*ppv); |
| 1862 | return S_OK; |
| 1863 | } |
| 1864 | else |
| 1865 | TMProxyImpl_Release((IRpcProxyBuffer *)&proxy->lpvtbl2); |
| 1866 | return hres; |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 1867 | } |
| 1868 | |
| 1869 | typedef struct _TMStubImpl { |
Dmitry Timoshkov | eba47f1 | 2005-06-06 19:50:35 +0000 | [diff] [blame] | 1870 | const IRpcStubBufferVtbl *lpvtbl; |
Mike McCormack | fc200d0 | 2005-07-11 10:25:19 +0000 | [diff] [blame] | 1871 | LONG ref; |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 1872 | |
| 1873 | LPUNKNOWN pUnk; |
| 1874 | ITypeInfo *tinfo; |
| 1875 | IID iid; |
Robert Shearman | b0218db | 2006-02-07 16:26:23 +0100 | [diff] [blame] | 1876 | IRpcStubBuffer *dispatch_stub; |
Robert Shearman | 4d0b1e3 | 2006-08-29 21:12:16 +0100 | [diff] [blame] | 1877 | BOOL dispatch_derivative; |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 1878 | } TMStubImpl; |
| 1879 | |
| 1880 | static HRESULT WINAPI |
Robert Shearman | 2a54aac | 2004-09-07 20:42:26 +0000 | [diff] [blame] | 1881 | TMStubImpl_QueryInterface(LPRPCSTUBBUFFER iface, REFIID riid, LPVOID *ppv) |
| 1882 | { |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 1883 | if (IsEqualIID(riid,&IID_IRpcStubBuffer)||IsEqualIID(riid,&IID_IUnknown)){ |
| 1884 | *ppv = (LPVOID)iface; |
| 1885 | IRpcStubBuffer_AddRef(iface); |
| 1886 | return S_OK; |
| 1887 | } |
| 1888 | FIXME("%s, not supported IID.\n",debugstr_guid(riid)); |
| 1889 | return E_NOINTERFACE; |
| 1890 | } |
| 1891 | |
| 1892 | static ULONG WINAPI |
Robert Shearman | 2a54aac | 2004-09-07 20:42:26 +0000 | [diff] [blame] | 1893 | TMStubImpl_AddRef(LPRPCSTUBBUFFER iface) |
| 1894 | { |
Alexandre Julliard | 241a4c3 | 2004-09-09 21:03:58 +0000 | [diff] [blame] | 1895 | TMStubImpl *This = (TMStubImpl *)iface; |
James Hawkins | ba92d29 | 2005-01-14 16:02:20 +0000 | [diff] [blame] | 1896 | ULONG refCount = InterlockedIncrement(&This->ref); |
Mike Hearn | 6545d99 | 2004-07-30 18:43:27 +0000 | [diff] [blame] | 1897 | |
Hans Leidekker | ce66f5d | 2006-10-12 20:57:10 +0200 | [diff] [blame] | 1898 | TRACE("(%p)->(ref before=%u)\n", This, refCount - 1); |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 1899 | |
James Hawkins | ba92d29 | 2005-01-14 16:02:20 +0000 | [diff] [blame] | 1900 | return refCount; |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 1901 | } |
| 1902 | |
| 1903 | static ULONG WINAPI |
Robert Shearman | 2a54aac | 2004-09-07 20:42:26 +0000 | [diff] [blame] | 1904 | TMStubImpl_Release(LPRPCSTUBBUFFER iface) |
| 1905 | { |
Alexandre Julliard | 241a4c3 | 2004-09-09 21:03:58 +0000 | [diff] [blame] | 1906 | TMStubImpl *This = (TMStubImpl *)iface; |
James Hawkins | ba92d29 | 2005-01-14 16:02:20 +0000 | [diff] [blame] | 1907 | ULONG refCount = InterlockedDecrement(&This->ref); |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 1908 | |
Hans Leidekker | ce66f5d | 2006-10-12 20:57:10 +0200 | [diff] [blame] | 1909 | TRACE("(%p)->(ref before=%u)\n", This, refCount + 1); |
Mike Hearn | 6545d99 | 2004-07-30 18:43:27 +0000 | [diff] [blame] | 1910 | |
James Hawkins | ba92d29 | 2005-01-14 16:02:20 +0000 | [diff] [blame] | 1911 | if (!refCount) |
Robert Shearman | 2a54aac | 2004-09-07 20:42:26 +0000 | [diff] [blame] | 1912 | { |
| 1913 | IRpcStubBuffer_Disconnect(iface); |
Robert Shearman | 6bfae1a | 2005-07-19 20:00:07 +0000 | [diff] [blame] | 1914 | ITypeInfo_Release(This->tinfo); |
Robert Shearman | 4d0b1e3 | 2006-08-29 21:12:16 +0100 | [diff] [blame] | 1915 | if (This->dispatch_stub) |
| 1916 | IRpcStubBuffer_Release(This->dispatch_stub); |
Robert Shearman | 2a54aac | 2004-09-07 20:42:26 +0000 | [diff] [blame] | 1917 | CoTaskMemFree(This); |
| 1918 | } |
James Hawkins | ba92d29 | 2005-01-14 16:02:20 +0000 | [diff] [blame] | 1919 | return refCount; |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 1920 | } |
| 1921 | |
| 1922 | static HRESULT WINAPI |
Robert Shearman | 2a54aac | 2004-09-07 20:42:26 +0000 | [diff] [blame] | 1923 | TMStubImpl_Connect(LPRPCSTUBBUFFER iface, LPUNKNOWN pUnkServer) |
| 1924 | { |
Alexandre Julliard | 241a4c3 | 2004-09-09 21:03:58 +0000 | [diff] [blame] | 1925 | TMStubImpl *This = (TMStubImpl *)iface; |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 1926 | |
Mike Hearn | 6545d99 | 2004-07-30 18:43:27 +0000 | [diff] [blame] | 1927 | TRACE("(%p)->(%p)\n", This, pUnkServer); |
| 1928 | |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 1929 | IUnknown_AddRef(pUnkServer); |
| 1930 | This->pUnk = pUnkServer; |
Robert Shearman | b0218db | 2006-02-07 16:26:23 +0100 | [diff] [blame] | 1931 | |
| 1932 | if (This->dispatch_stub) |
| 1933 | IRpcStubBuffer_Connect(This->dispatch_stub, pUnkServer); |
| 1934 | |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 1935 | return S_OK; |
| 1936 | } |
| 1937 | |
| 1938 | static void WINAPI |
Robert Shearman | 2a54aac | 2004-09-07 20:42:26 +0000 | [diff] [blame] | 1939 | TMStubImpl_Disconnect(LPRPCSTUBBUFFER iface) |
| 1940 | { |
Alexandre Julliard | 241a4c3 | 2004-09-09 21:03:58 +0000 | [diff] [blame] | 1941 | TMStubImpl *This = (TMStubImpl *)iface; |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 1942 | |
Mike Hearn | 6545d99 | 2004-07-30 18:43:27 +0000 | [diff] [blame] | 1943 | TRACE("(%p)->()\n", This); |
| 1944 | |
Robert Shearman | 6bfae1a | 2005-07-19 20:00:07 +0000 | [diff] [blame] | 1945 | if (This->pUnk) |
| 1946 | { |
| 1947 | IUnknown_Release(This->pUnk); |
| 1948 | This->pUnk = NULL; |
| 1949 | } |
Robert Shearman | b0218db | 2006-02-07 16:26:23 +0100 | [diff] [blame] | 1950 | |
| 1951 | if (This->dispatch_stub) |
| 1952 | IRpcStubBuffer_Disconnect(This->dispatch_stub); |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 1953 | } |
| 1954 | |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 1955 | static HRESULT WINAPI |
| 1956 | TMStubImpl_Invoke( |
Robert Shearman | 2a54aac | 2004-09-07 20:42:26 +0000 | [diff] [blame] | 1957 | LPRPCSTUBBUFFER iface, RPCOLEMESSAGE* xmsg,IRpcChannelBuffer*rpcchanbuf) |
| 1958 | { |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 1959 | int i; |
Robert Shearman | b0d8e06c | 2005-12-06 21:24:19 +0100 | [diff] [blame] | 1960 | const FUNCDESC *fdesc; |
Alexandre Julliard | 241a4c3 | 2004-09-09 21:03:58 +0000 | [diff] [blame] | 1961 | TMStubImpl *This = (TMStubImpl *)iface; |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 1962 | HRESULT hres; |
Rob Shearman | d1b15c8 | 2006-10-23 11:38:20 +0100 | [diff] [blame] | 1963 | DWORD *args = NULL, res, *xargs, nrofargs; |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 1964 | marshal_state buf; |
James Hawkins | c8eb807 | 2007-01-16 04:15:20 -0600 | [diff] [blame] | 1965 | UINT nrofnames = 0; |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 1966 | BSTR names[10]; |
Robert Shearman | 5c2a708 | 2005-10-26 10:12:30 +0000 | [diff] [blame] | 1967 | BSTR iname = NULL; |
Rob Shearman | d1b15c8 | 2006-10-23 11:38:20 +0100 | [diff] [blame] | 1968 | ITypeInfo *tinfo = NULL; |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 1969 | |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 1970 | TRACE("...\n"); |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 1971 | |
Robert Shearman | 5c2a708 | 2005-10-26 10:12:30 +0000 | [diff] [blame] | 1972 | if (xmsg->iMethod < 3) { |
| 1973 | ERR("IUnknown methods cannot be marshaled by the typelib marshaler\n"); |
| 1974 | return E_UNEXPECTED; |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 1975 | } |
Robert Shearman | 5c2a708 | 2005-10-26 10:12:30 +0000 | [diff] [blame] | 1976 | |
Robert Shearman | 4d0b1e3 | 2006-08-29 21:12:16 +0100 | [diff] [blame] | 1977 | if (This->dispatch_derivative && xmsg->iMethod < sizeof(IDispatchVtbl)/sizeof(void *)) |
| 1978 | { |
| 1979 | IPSFactoryBuffer *factory_buffer; |
| 1980 | hres = get_facbuf_for_iid(&IID_IDispatch, &factory_buffer); |
| 1981 | if (hres == S_OK) |
| 1982 | { |
| 1983 | hres = IPSFactoryBuffer_CreateStub(factory_buffer, &IID_IDispatch, |
| 1984 | This->pUnk, &This->dispatch_stub); |
| 1985 | IPSFactoryBuffer_Release(factory_buffer); |
| 1986 | } |
| 1987 | if (hres != S_OK) |
| 1988 | return hres; |
Robert Shearman | b0218db | 2006-02-07 16:26:23 +0100 | [diff] [blame] | 1989 | return IRpcStubBuffer_Invoke(This->dispatch_stub, xmsg, rpcchanbuf); |
Robert Shearman | 4d0b1e3 | 2006-08-29 21:12:16 +0100 | [diff] [blame] | 1990 | } |
Robert Shearman | b0218db | 2006-02-07 16:26:23 +0100 | [diff] [blame] | 1991 | |
| 1992 | memset(&buf,0,sizeof(buf)); |
| 1993 | buf.size = xmsg->cbBuffer; |
| 1994 | buf.base = HeapAlloc(GetProcessHeap(), 0, xmsg->cbBuffer); |
| 1995 | memcpy(buf.base, xmsg->Buffer, xmsg->cbBuffer); |
| 1996 | buf.curoff = 0; |
| 1997 | |
Huw Davies | 294b3eb | 2007-04-26 11:56:44 +0100 | [diff] [blame] | 1998 | hres = get_funcdesc(This->tinfo,xmsg->iMethod,&tinfo,&fdesc,&iname,NULL,NULL); |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 1999 | if (hres) { |
Hans Leidekker | ce66f5d | 2006-10-12 20:57:10 +0200 | [diff] [blame] | 2000 | ERR("GetFuncDesc on method %d failed with %x\n",xmsg->iMethod,hres); |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 2001 | return hres; |
| 2002 | } |
Marcus Meissner | e6260a3 | 2005-03-27 17:51:59 +0000 | [diff] [blame] | 2003 | |
Robert Shearman | 5c2a708 | 2005-10-26 10:12:30 +0000 | [diff] [blame] | 2004 | if (iname && !lstrcmpW(iname, IDispatchW)) |
| 2005 | { |
| 2006 | ERR("IDispatch cannot be marshaled by the typelib marshaler\n"); |
Rob Shearman | d1b15c8 | 2006-10-23 11:38:20 +0100 | [diff] [blame] | 2007 | hres = E_UNEXPECTED; |
| 2008 | SysFreeString (iname); |
| 2009 | goto exit; |
Robert Shearman | 5c2a708 | 2005-10-26 10:12:30 +0000 | [diff] [blame] | 2010 | } |
Marcus Meissner | e6260a3 | 2005-03-27 17:51:59 +0000 | [diff] [blame] | 2011 | |
| 2012 | if (iname) SysFreeString (iname); |
Marcus Meissner | e6260a3 | 2005-03-27 17:51:59 +0000 | [diff] [blame] | 2013 | |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 2014 | /* Need them for hack below */ |
| 2015 | memset(names,0,sizeof(names)); |
Robert Shearman | c6296c4 | 2005-06-01 11:03:40 +0000 | [diff] [blame] | 2016 | ITypeInfo_GetNames(tinfo,fdesc->memid,names,sizeof(names)/sizeof(names[0]),&nrofnames); |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 2017 | if (nrofnames > sizeof(names)/sizeof(names[0])) { |
| 2018 | ERR("Need more names!\n"); |
| 2019 | } |
| 2020 | |
| 2021 | /*dump_FUNCDESC(fdesc);*/ |
| 2022 | nrofargs = 0; |
| 2023 | for (i=0;i<fdesc->cParams;i++) |
| 2024 | nrofargs += _argsize(fdesc->lprgelemdescParam[i].tdesc.vt); |
| 2025 | args = HeapAlloc(GetProcessHeap(),0,(nrofargs+1)*sizeof(DWORD)); |
Rob Shearman | d1b15c8 | 2006-10-23 11:38:20 +0100 | [diff] [blame] | 2026 | if (!args) |
| 2027 | { |
| 2028 | hres = E_OUTOFMEMORY; |
| 2029 | goto exit; |
| 2030 | } |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 2031 | |
| 2032 | /* Allocate all stuff used by call. */ |
| 2033 | xargs = args+1; |
| 2034 | for (i=0;i<fdesc->cParams;i++) { |
| 2035 | ELEMDESC *elem = fdesc->lprgelemdescParam+i; |
| 2036 | |
Robert Shearman | 5c2a708 | 2005-10-26 10:12:30 +0000 | [diff] [blame] | 2037 | hres = deserialize_param( |
| 2038 | tinfo, |
Rob Shearman | 0f45b2b | 2007-02-07 19:19:26 +0000 | [diff] [blame] | 2039 | is_in_elem(elem), |
Robert Shearman | 5c2a708 | 2005-10-26 10:12:30 +0000 | [diff] [blame] | 2040 | FALSE, |
| 2041 | TRUE, |
| 2042 | &(elem->tdesc), |
| 2043 | xargs, |
| 2044 | &buf |
| 2045 | ); |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 2046 | xargs += _argsize(elem->tdesc.vt); |
| 2047 | if (hres) { |
Hans Leidekker | ce66f5d | 2006-10-12 20:57:10 +0200 | [diff] [blame] | 2048 | ERR("Failed to deserialize param %s, hres %x\n",relaystr(names[i+1]),hres); |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 2049 | break; |
| 2050 | } |
| 2051 | } |
Robert Shearman | 5c2a708 | 2005-10-26 10:12:30 +0000 | [diff] [blame] | 2052 | |
Robert Shearman | 69562b0 | 2005-10-27 12:08:58 +0000 | [diff] [blame] | 2053 | args[0] = (DWORD)This->pUnk; |
Robert Shearman | a05672e | 2006-07-17 20:19:27 +0100 | [diff] [blame] | 2054 | |
| 2055 | __TRY |
| 2056 | { |
| 2057 | res = _invoke( |
| 2058 | (*((FARPROC**)args[0]))[fdesc->oVft/4], |
| 2059 | fdesc->callconv, |
| 2060 | (xargs-args), |
| 2061 | args |
| 2062 | ); |
| 2063 | } |
| 2064 | __EXCEPT(NULL) |
| 2065 | { |
| 2066 | DWORD dwExceptionCode = GetExceptionCode(); |
Hans Leidekker | ce66f5d | 2006-10-12 20:57:10 +0200 | [diff] [blame] | 2067 | ERR("invoke call failed with exception 0x%08x (%d)\n", dwExceptionCode, dwExceptionCode); |
Robert Shearman | a05672e | 2006-07-17 20:19:27 +0100 | [diff] [blame] | 2068 | if (FAILED(dwExceptionCode)) |
| 2069 | hres = dwExceptionCode; |
| 2070 | else |
| 2071 | hres = HRESULT_FROM_WIN32(dwExceptionCode); |
| 2072 | } |
| 2073 | __ENDTRY |
| 2074 | |
| 2075 | if (hres != S_OK) |
Rob Shearman | d1b15c8 | 2006-10-23 11:38:20 +0100 | [diff] [blame] | 2076 | goto exit; |
Robert Shearman | a05672e | 2006-07-17 20:19:27 +0100 | [diff] [blame] | 2077 | |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 2078 | buf.curoff = 0; |
Marcus Meissner | e6260a3 | 2005-03-27 17:51:59 +0000 | [diff] [blame] | 2079 | |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 2080 | xargs = args+1; |
| 2081 | for (i=0;i<fdesc->cParams;i++) { |
| 2082 | ELEMDESC *elem = fdesc->lprgelemdescParam+i; |
Robert Shearman | 5c2a708 | 2005-10-26 10:12:30 +0000 | [diff] [blame] | 2083 | hres = serialize_param( |
| 2084 | tinfo, |
Rob Shearman | 0f45b2b | 2007-02-07 19:19:26 +0000 | [diff] [blame] | 2085 | is_out_elem(elem), |
Robert Shearman | 5c2a708 | 2005-10-26 10:12:30 +0000 | [diff] [blame] | 2086 | FALSE, |
| 2087 | TRUE, |
| 2088 | &elem->tdesc, |
| 2089 | xargs, |
| 2090 | &buf |
| 2091 | ); |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 2092 | xargs += _argsize(elem->tdesc.vt); |
| 2093 | if (hres) { |
Hans Leidekker | ce66f5d | 2006-10-12 20:57:10 +0200 | [diff] [blame] | 2094 | ERR("Failed to stuballoc param, hres %x\n",hres); |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 2095 | break; |
| 2096 | } |
| 2097 | } |
Robert Shearman | 5c2a708 | 2005-10-26 10:12:30 +0000 | [diff] [blame] | 2098 | |
Marcus Meissner | 89feaca | 2005-03-28 10:00:49 +0000 | [diff] [blame] | 2099 | hres = xbuf_add (&buf, (LPBYTE)&res, sizeof(DWORD)); |
Rob Shearman | 0fa33bb | 2006-10-23 11:29:09 +0100 | [diff] [blame] | 2100 | |
Marcus Meissner | 89feaca | 2005-03-28 10:00:49 +0000 | [diff] [blame] | 2101 | if (hres != S_OK) |
Rob Shearman | d1b15c8 | 2006-10-23 11:38:20 +0100 | [diff] [blame] | 2102 | goto exit; |
Robert Shearman | 44d6e39 | 2005-08-29 21:46:51 +0000 | [diff] [blame] | 2103 | |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 2104 | xmsg->cbBuffer = buf.curoff; |
Robert Shearman | 70c34cc | 2006-06-09 17:25:50 +0100 | [diff] [blame] | 2105 | hres = IRpcChannelBuffer_GetBuffer(rpcchanbuf, xmsg, &This->iid); |
| 2106 | if (hres != S_OK) |
Hans Leidekker | ce66f5d | 2006-10-12 20:57:10 +0200 | [diff] [blame] | 2107 | ERR("IRpcChannelBuffer_GetBuffer failed with error 0x%08x\n", hres); |
Robert Shearman | 44d6e39 | 2005-08-29 21:46:51 +0000 | [diff] [blame] | 2108 | |
| 2109 | if (hres == S_OK) |
| 2110 | memcpy(xmsg->Buffer, buf.base, buf.curoff); |
| 2111 | |
Rob Shearman | d1b15c8 | 2006-10-23 11:38:20 +0100 | [diff] [blame] | 2112 | exit: |
| 2113 | for (i = 0; i < nrofnames; i++) |
| 2114 | SysFreeString(names[i]); |
| 2115 | |
| 2116 | ITypeInfo_Release(tinfo); |
| 2117 | HeapFree(GetProcessHeap(), 0, args); |
| 2118 | |
Robert Shearman | 44d6e39 | 2005-08-29 21:46:51 +0000 | [diff] [blame] | 2119 | HeapFree(GetProcessHeap(), 0, buf.base); |
| 2120 | |
| 2121 | TRACE("returning\n"); |
| 2122 | return hres; |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 2123 | } |
| 2124 | |
| 2125 | static LPRPCSTUBBUFFER WINAPI |
| 2126 | TMStubImpl_IsIIDSupported(LPRPCSTUBBUFFER iface, REFIID riid) { |
| 2127 | FIXME("Huh (%s)?\n",debugstr_guid(riid)); |
| 2128 | return NULL; |
| 2129 | } |
| 2130 | |
| 2131 | static ULONG WINAPI |
| 2132 | TMStubImpl_CountRefs(LPRPCSTUBBUFFER iface) { |
Alexandre Julliard | 241a4c3 | 2004-09-09 21:03:58 +0000 | [diff] [blame] | 2133 | TMStubImpl *This = (TMStubImpl *)iface; |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 2134 | |
Robert Shearman | 44d6e39 | 2005-08-29 21:46:51 +0000 | [diff] [blame] | 2135 | FIXME("()\n"); |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 2136 | return This->ref; /*FIXME? */ |
| 2137 | } |
| 2138 | |
| 2139 | static HRESULT WINAPI |
| 2140 | TMStubImpl_DebugServerQueryInterface(LPRPCSTUBBUFFER iface, LPVOID *ppv) { |
| 2141 | return E_NOTIMPL; |
| 2142 | } |
| 2143 | |
| 2144 | static void WINAPI |
| 2145 | TMStubImpl_DebugServerRelease(LPRPCSTUBBUFFER iface, LPVOID ppv) { |
| 2146 | return; |
| 2147 | } |
| 2148 | |
Dmitry Timoshkov | eba47f1 | 2005-06-06 19:50:35 +0000 | [diff] [blame] | 2149 | static const IRpcStubBufferVtbl tmstubvtbl = { |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 2150 | TMStubImpl_QueryInterface, |
| 2151 | TMStubImpl_AddRef, |
| 2152 | TMStubImpl_Release, |
| 2153 | TMStubImpl_Connect, |
| 2154 | TMStubImpl_Disconnect, |
| 2155 | TMStubImpl_Invoke, |
| 2156 | TMStubImpl_IsIIDSupported, |
| 2157 | TMStubImpl_CountRefs, |
| 2158 | TMStubImpl_DebugServerQueryInterface, |
| 2159 | TMStubImpl_DebugServerRelease |
| 2160 | }; |
| 2161 | |
| 2162 | static HRESULT WINAPI |
| 2163 | PSFacBuf_CreateStub( |
| 2164 | LPPSFACTORYBUFFER iface, REFIID riid,IUnknown *pUnkServer, |
| 2165 | IRpcStubBuffer** ppStub |
| 2166 | ) { |
| 2167 | HRESULT hres; |
| 2168 | ITypeInfo *tinfo; |
| 2169 | TMStubImpl *stub; |
Robert Shearman | b0218db | 2006-02-07 16:26:23 +0100 | [diff] [blame] | 2170 | TYPEATTR *typeattr; |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 2171 | |
| 2172 | TRACE("(%s,%p,%p)\n",debugstr_guid(riid),pUnkServer,ppStub); |
Robert Shearman | b0218db | 2006-02-07 16:26:23 +0100 | [diff] [blame] | 2173 | |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 2174 | hres = _get_typeinfo_for_iid(riid,&tinfo); |
| 2175 | if (hres) { |
Mike Hearn | 64eac75 | 2005-01-27 11:11:32 +0000 | [diff] [blame] | 2176 | ERR("No typeinfo for %s?\n",debugstr_guid(riid)); |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 2177 | return hres; |
| 2178 | } |
Robert Shearman | b0218db | 2006-02-07 16:26:23 +0100 | [diff] [blame] | 2179 | |
Robert Shearman | 2a54aac | 2004-09-07 20:42:26 +0000 | [diff] [blame] | 2180 | stub = CoTaskMemAlloc(sizeof(TMStubImpl)); |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 2181 | if (!stub) |
| 2182 | return E_OUTOFMEMORY; |
| 2183 | stub->lpvtbl = &tmstubvtbl; |
| 2184 | stub->ref = 1; |
| 2185 | stub->tinfo = tinfo; |
Robert Shearman | b0218db | 2006-02-07 16:26:23 +0100 | [diff] [blame] | 2186 | stub->dispatch_stub = NULL; |
Robert Shearman | 4d0b1e3 | 2006-08-29 21:12:16 +0100 | [diff] [blame] | 2187 | stub->dispatch_derivative = FALSE; |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 2188 | memcpy(&(stub->iid),riid,sizeof(*riid)); |
| 2189 | hres = IRpcStubBuffer_Connect((LPRPCSTUBBUFFER)stub,pUnkServer); |
| 2190 | *ppStub = (LPRPCSTUBBUFFER)stub; |
Mike Hearn | 6545d99 | 2004-07-30 18:43:27 +0000 | [diff] [blame] | 2191 | TRACE("IRpcStubBuffer: %p\n", stub); |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 2192 | if (hres) |
Mike Hearn | 64eac75 | 2005-01-27 11:11:32 +0000 | [diff] [blame] | 2193 | ERR("Connect to pUnkServer failed?\n"); |
Robert Shearman | b0218db | 2006-02-07 16:26:23 +0100 | [diff] [blame] | 2194 | |
| 2195 | /* if we derive from IDispatch then defer to its stub for some of its methods */ |
| 2196 | hres = ITypeInfo_GetTypeAttr(tinfo, &typeattr); |
| 2197 | if (hres == S_OK) |
| 2198 | { |
| 2199 | if (typeattr->wTypeFlags & TYPEFLAG_FDISPATCHABLE) |
Robert Shearman | 4d0b1e3 | 2006-08-29 21:12:16 +0100 | [diff] [blame] | 2200 | stub->dispatch_derivative = TRUE; |
Robert Shearman | b0218db | 2006-02-07 16:26:23 +0100 | [diff] [blame] | 2201 | ITypeInfo_ReleaseTypeAttr(tinfo, typeattr); |
| 2202 | } |
| 2203 | |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 2204 | return hres; |
| 2205 | } |
| 2206 | |
Dmitry Timoshkov | eba47f1 | 2005-06-06 19:50:35 +0000 | [diff] [blame] | 2207 | static const IPSFactoryBufferVtbl psfacbufvtbl = { |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 2208 | PSFacBuf_QueryInterface, |
| 2209 | PSFacBuf_AddRef, |
| 2210 | PSFacBuf_Release, |
| 2211 | PSFacBuf_CreateProxy, |
| 2212 | PSFacBuf_CreateStub |
| 2213 | }; |
| 2214 | |
| 2215 | /* This is the whole PSFactoryBuffer object, just the vtableptr */ |
Dmitry Timoshkov | eba47f1 | 2005-06-06 19:50:35 +0000 | [diff] [blame] | 2216 | static const IPSFactoryBufferVtbl *lppsfac = &psfacbufvtbl; |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 2217 | |
| 2218 | /*********************************************************************** |
Alexandre Julliard | b8d0f4c | 2005-08-09 10:38:54 +0000 | [diff] [blame] | 2219 | * TMARSHAL_DllGetClassObject |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 2220 | */ |
Alexandre Julliard | b8d0f4c | 2005-08-09 10:38:54 +0000 | [diff] [blame] | 2221 | HRESULT TMARSHAL_DllGetClassObject(REFCLSID rclsid, REFIID iid,LPVOID *ppv) |
Marcus Meissner | 0749fc2 | 2002-02-05 18:11:17 +0000 | [diff] [blame] | 2222 | { |
| 2223 | if (IsEqualIID(iid,&IID_IPSFactoryBuffer)) { |
| 2224 | *ppv = &lppsfac; |
| 2225 | return S_OK; |
| 2226 | } |
| 2227 | return E_NOINTERFACE; |
| 2228 | } |