Sync to Wine-0_9_1:
[reactos.git] / reactos / lib / rpcrt4 / cstub.c
1 /*
2 * COM stub (CStdStubBuffer) implementation
3 *
4 * Copyright 2001 Ove Kåven, TransGaming Technologies
5 *
6 * This library is free software; you can redistribute it and/or
7 * modify it under the terms of the GNU Lesser General Public
8 * License as published by the Free Software Foundation; either
9 * version 2.1 of the License, or (at your option) any later version.
10 *
11 * This library is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
14 * Lesser General Public License for more details.
15 *
16 * You should have received a copy of the GNU Lesser General Public
17 * License along with this library; if not, write to the Free Software
18 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
19 */
20
21 #include <stdarg.h>
22
23 #define COBJMACROS
24
25 #include "windef.h"
26 #include "winbase.h"
27 #include "winerror.h"
28
29 #include "objbase.h"
30
31 #include "rpcproxy.h"
32
33 #include "wine/debug.h"
34
35 #include "cpsf.h"
36
37 WINE_DEFAULT_DEBUG_CHANNEL(ole);
38
39 #define STUB_HEADER(This) (((CInterfaceStubHeader*)((This)->lpVtbl))[-1])
40
41 HRESULT WINAPI CStdStubBuffer_Construct(REFIID riid,
42 LPUNKNOWN pUnkServer,
43 PCInterfaceName name,
44 CInterfaceStubVtbl *vtbl,
45 LPPSFACTORYBUFFER pPSFactory,
46 LPRPCSTUBBUFFER *ppStub)
47 {
48 CStdStubBuffer *This;
49
50 TRACE("(%p,%p,%p,%p) %s\n", pUnkServer, vtbl, pPSFactory, ppStub, name);
51 TRACE("iid=%s\n", debugstr_guid(vtbl->header.piid));
52 TRACE("vtbl=%p\n", &vtbl->Vtbl);
53
54 if (!IsEqualGUID(vtbl->header.piid, riid)) {
55 ERR("IID mismatch during stub creation\n");
56 return RPC_E_UNEXPECTED;
57 }
58
59 This = HeapAlloc(GetProcessHeap(),HEAP_ZERO_MEMORY,sizeof(CStdStubBuffer));
60 if (!This) return E_OUTOFMEMORY;
61
62 This->lpVtbl = &vtbl->Vtbl;
63 This->RefCount = 1;
64 This->pvServerObject = pUnkServer;
65 This->pPSFactory = pPSFactory;
66 *ppStub = (LPRPCSTUBBUFFER)This;
67
68 IUnknown_AddRef(This->pvServerObject);
69 IPSFactoryBuffer_AddRef(pPSFactory);
70 return S_OK;
71 }
72
73 HRESULT WINAPI CStdStubBuffer_QueryInterface(LPRPCSTUBBUFFER iface,
74 REFIID riid,
75 LPVOID *obj)
76 {
77 CStdStubBuffer *This = (CStdStubBuffer *)iface;
78 TRACE("(%p)->QueryInterface(%s,%p)\n",This,debugstr_guid(riid),obj);
79
80 if (IsEqualGUID(&IID_IUnknown,riid) ||
81 IsEqualGUID(&IID_IRpcStubBuffer,riid)) {
82 *obj = This;
83 This->RefCount++;
84 return S_OK;
85 }
86 return E_NOINTERFACE;
87 }
88
89 ULONG WINAPI CStdStubBuffer_AddRef(LPRPCSTUBBUFFER iface)
90 {
91 CStdStubBuffer *This = (CStdStubBuffer *)iface;
92 TRACE("(%p)->AddRef()\n",This);
93 return ++(This->RefCount);
94 }
95
96 ULONG WINAPI NdrCStdStubBuffer_Release(LPRPCSTUBBUFFER iface,
97 LPPSFACTORYBUFFER pPSF)
98 {
99 CStdStubBuffer *This = (CStdStubBuffer *)iface;
100 TRACE("(%p)->Release()\n",This);
101
102 if (!--(This->RefCount)) {
103 IRpcStubBuffer_Disconnect(iface);
104 if(This->pPSFactory)
105 IPSFactoryBuffer_Release(This->pPSFactory);
106 HeapFree(GetProcessHeap(),0,This);
107 return 0;
108 }
109 return This->RefCount;
110 }
111
112 ULONG WINAPI NdrCStdStubBuffer2_Release(LPRPCSTUBBUFFER iface,
113 LPPSFACTORYBUFFER pPSF)
114 {
115 FIXME("Not implemented\n");
116 return 0;
117 }
118
119 HRESULT WINAPI CStdStubBuffer_Connect(LPRPCSTUBBUFFER iface,
120 LPUNKNOWN lpUnkServer)
121 {
122 CStdStubBuffer *This = (CStdStubBuffer *)iface;
123 TRACE("(%p)->Connect(%p)\n",This,lpUnkServer);
124 This->pvServerObject = lpUnkServer;
125 return S_OK;
126 }
127
128 void WINAPI CStdStubBuffer_Disconnect(LPRPCSTUBBUFFER iface)
129 {
130 CStdStubBuffer *This = (CStdStubBuffer *)iface;
131 TRACE("(%p)->Disconnect()\n",This);
132 if (This->pvServerObject)
133 {
134 IUnknown_Release(This->pvServerObject);
135 This->pvServerObject = NULL;
136 }
137 }
138
139 HRESULT WINAPI CStdStubBuffer_Invoke(LPRPCSTUBBUFFER iface,
140 PRPCOLEMESSAGE pMsg,
141 LPRPCCHANNELBUFFER pChannel)
142 {
143 CStdStubBuffer *This = (CStdStubBuffer *)iface;
144 DWORD dwPhase = STUB_UNMARSHAL;
145 TRACE("(%p)->Invoke(%p,%p)\n",This,pMsg,pChannel);
146
147 STUB_HEADER(This).pDispatchTable[pMsg->iMethod](iface, pChannel, (PRPC_MESSAGE)pMsg, &dwPhase);
148 return S_OK;
149 }
150
151 LPRPCSTUBBUFFER WINAPI CStdStubBuffer_IsIIDSupported(LPRPCSTUBBUFFER iface,
152 REFIID riid)
153 {
154 CStdStubBuffer *This = (CStdStubBuffer *)iface;
155 TRACE("(%p)->IsIIDSupported(%s)\n",This,debugstr_guid(riid));
156 return IsEqualGUID(STUB_HEADER(This).piid, riid) ? iface : NULL;
157 }
158
159 ULONG WINAPI CStdStubBuffer_CountRefs(LPRPCSTUBBUFFER iface)
160 {
161 CStdStubBuffer *This = (CStdStubBuffer *)iface;
162 TRACE("(%p)->CountRefs()\n",This);
163 return This->RefCount;
164 }
165
166 HRESULT WINAPI CStdStubBuffer_DebugServerQueryInterface(LPRPCSTUBBUFFER iface,
167 LPVOID *ppv)
168 {
169 CStdStubBuffer *This = (CStdStubBuffer *)iface;
170 TRACE("(%p)->DebugServerQueryInterface(%p)\n",This,ppv);
171 return S_OK;
172 }
173
174 void WINAPI CStdStubBuffer_DebugServerRelease(LPRPCSTUBBUFFER iface,
175 LPVOID pv)
176 {
177 CStdStubBuffer *This = (CStdStubBuffer *)iface;
178 TRACE("(%p)->DebugServerRelease(%p)\n",This,pv);
179 }