2 * MIDL proxy/stub stuff
4 * Copyright 2002 Ove Kåven, TransGaming Technologies
6 * This library is free software; you can redistribute it and/or
7 * modify it under the terms of the GNU Lesser General Public
8 * License as published by the Free Software Foundation; either
9 * version 2.1 of the License, or (at your option) any later version.
11 * This library is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
14 * Lesser General Public License for more details.
16 * You should have received a copy of the GNU Lesser General Public
17 * License along with this library; if not, write to the Free Software
18 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
21 * - figure out whether we *really* got this right
22 * - check for errors and throw exceptions
41 #include "wine/debug.h"
47 WINE_DEFAULT_DEBUG_CHANNEL(ole
);
49 /***********************************************************************
50 * NdrProxyInitialize [RPCRT4.@]
52 void WINAPI
NdrProxyInitialize(void *This
,
54 PMIDL_STUB_MESSAGE pStubMsg
,
55 PMIDL_STUB_DESC pStubDescriptor
,
60 TRACE("(%p,%p,%p,%p,%d)\n", This
, pRpcMsg
, pStubMsg
, pStubDescriptor
, ProcNum
);
61 NdrClientInitializeNew(pRpcMsg
, pStubMsg
, pStubDescriptor
, ProcNum
);
62 if (This
) StdProxy_GetChannel(This
, &pStubMsg
->pRpcChannelBuffer
);
63 if (pStubMsg
->pRpcChannelBuffer
) {
64 hr
= IRpcChannelBuffer_GetDestCtx(pStubMsg
->pRpcChannelBuffer
,
65 &pStubMsg
->dwDestContext
,
66 &pStubMsg
->pvDestContext
);
68 TRACE("channel=%p\n", pStubMsg
->pRpcChannelBuffer
);
71 /***********************************************************************
72 * NdrProxyGetBuffer [RPCRT4.@]
74 void WINAPI
NdrProxyGetBuffer(void *This
,
75 PMIDL_STUB_MESSAGE pStubMsg
)
78 const IID
*riid
= NULL
;
80 TRACE("(%p,%p)\n", This
, pStubMsg
);
81 pStubMsg
->RpcMsg
->BufferLength
= pStubMsg
->BufferLength
;
82 pStubMsg
->dwStubPhase
= PROXY_GETBUFFER
;
83 hr
= StdProxy_GetIID(This
, &riid
);
84 hr
= IRpcChannelBuffer_GetBuffer(pStubMsg
->pRpcChannelBuffer
,
85 (RPCOLEMESSAGE
*)pStubMsg
->RpcMsg
,
87 pStubMsg
->BufferStart
= pStubMsg
->RpcMsg
->Buffer
;
88 pStubMsg
->BufferEnd
= pStubMsg
->BufferStart
+ pStubMsg
->BufferLength
;
89 pStubMsg
->Buffer
= pStubMsg
->BufferStart
;
90 pStubMsg
->dwStubPhase
= PROXY_MARSHAL
;
93 /***********************************************************************
94 * NdrProxySendReceive [RPCRT4.@]
96 void WINAPI
NdrProxySendReceive(void *This
,
97 PMIDL_STUB_MESSAGE pStubMsg
)
102 TRACE("(%p,%p)\n", This
, pStubMsg
);
104 if (!pStubMsg
->pRpcChannelBuffer
)
106 WARN("Trying to use disconnected proxy %p\n", This
);
107 RpcRaiseException(RPC_E_DISCONNECTED
);
110 pStubMsg
->dwStubPhase
= PROXY_SENDRECEIVE
;
111 hr
= IRpcChannelBuffer_SendReceive(pStubMsg
->pRpcChannelBuffer
,
112 (RPCOLEMESSAGE
*)pStubMsg
->RpcMsg
,
114 pStubMsg
->dwStubPhase
= PROXY_UNMARSHAL
;
115 pStubMsg
->BufferLength
= pStubMsg
->RpcMsg
->BufferLength
;
116 pStubMsg
->BufferStart
= pStubMsg
->RpcMsg
->Buffer
;
117 pStubMsg
->BufferEnd
= pStubMsg
->BufferStart
+ pStubMsg
->BufferLength
;
118 pStubMsg
->Buffer
= pStubMsg
->BufferStart
;
120 /* raise exception if call failed */
121 if (hr
== RPC_S_CALL_FAILED
) RpcRaiseException(*(DWORD
*)pStubMsg
->Buffer
);
122 else if (FAILED(hr
)) RpcRaiseException(hr
);
125 /***********************************************************************
126 * NdrProxyFreeBuffer [RPCRT4.@]
128 void WINAPI
NdrProxyFreeBuffer(void *This
,
129 PMIDL_STUB_MESSAGE pStubMsg
)
133 TRACE("(%p,%p)\n", This
, pStubMsg
);
134 hr
= IRpcChannelBuffer_FreeBuffer(pStubMsg
->pRpcChannelBuffer
,
135 (RPCOLEMESSAGE
*)pStubMsg
->RpcMsg
);
138 /***********************************************************************
139 * NdrProxyErrorHandler [RPCRT4.@]
141 HRESULT WINAPI
NdrProxyErrorHandler(DWORD dwExceptionCode
)
143 FIXME("(0x%08lx): semi-stub\n", dwExceptionCode
);
144 return MAKE_HRESULT(SEVERITY_ERROR
, FACILITY_RPC
, RPC_S_CALL_FAILED
);
147 /***********************************************************************
148 * NdrStubInitialize [RPCRT4.@]
150 void WINAPI
NdrStubInitialize(PRPC_MESSAGE pRpcMsg
,
151 PMIDL_STUB_MESSAGE pStubMsg
,
152 PMIDL_STUB_DESC pStubDescriptor
,
153 LPRPCCHANNELBUFFER pRpcChannelBuffer
)
155 TRACE("(%p,%p,%p,%p)\n", pRpcMsg
, pStubMsg
, pStubDescriptor
, pRpcChannelBuffer
);
156 NdrServerInitializeNew(pRpcMsg
, pStubMsg
, pStubDescriptor
);
157 pStubMsg
->pRpcChannelBuffer
= pRpcChannelBuffer
;
160 /***********************************************************************
161 * NdrStubGetBuffer [RPCRT4.@]
163 void WINAPI
NdrStubGetBuffer(LPRPCSTUBBUFFER This
,
164 LPRPCCHANNELBUFFER pRpcChannelBuffer
,
165 PMIDL_STUB_MESSAGE pStubMsg
)
167 TRACE("(%p,%p)\n", This
, pStubMsg
);
168 pStubMsg
->pRpcChannelBuffer
= pRpcChannelBuffer
;
169 pStubMsg
->RpcMsg
->BufferLength
= pStubMsg
->BufferLength
;
170 I_RpcGetBuffer(pStubMsg
->RpcMsg
); /* ? */
171 pStubMsg
->BufferStart
= pStubMsg
->RpcMsg
->Buffer
;
172 pStubMsg
->BufferEnd
= pStubMsg
->BufferStart
+ pStubMsg
->BufferLength
;
173 pStubMsg
->Buffer
= pStubMsg
->BufferStart
;
176 /************************************************************************
177 * NdrClientInitializeNew [RPCRT4.@]
179 void WINAPI
NdrClientInitializeNew( PRPC_MESSAGE pRpcMessage
, PMIDL_STUB_MESSAGE pStubMsg
,
180 PMIDL_STUB_DESC pStubDesc
, unsigned int ProcNum
)
182 TRACE("(pRpcMessage == ^%p, pStubMsg == ^%p, pStubDesc == ^%p, ProcNum == %d)\n",
183 pRpcMessage
, pStubMsg
, pStubDesc
, ProcNum
);
185 assert( pRpcMessage
&& pStubMsg
&& pStubDesc
);
187 memset(pRpcMessage
, 0, sizeof(RPC_MESSAGE
));
189 /* not everyone allocates stack space for w2kReserved */
190 memset(pStubMsg
, 0, FIELD_OFFSET(MIDL_STUB_MESSAGE
,pCSInfo
));
192 pStubMsg
->ReuseBuffer
= FALSE
;
193 pStubMsg
->IsClient
= TRUE
;
194 pStubMsg
->StubDesc
= pStubDesc
;
195 pStubMsg
->pfnAllocate
= pStubDesc
->pfnAllocate
;
196 pStubMsg
->pfnFree
= pStubDesc
->pfnFree
;
197 pStubMsg
->RpcMsg
= pRpcMessage
;
199 pRpcMessage
->ProcNum
= ProcNum
;
200 pRpcMessage
->RpcInterfaceInformation
= pStubDesc
->RpcInterfaceInformation
;
203 /***********************************************************************
204 * NdrServerInitializeNew [RPCRT4.@]
206 unsigned char* WINAPI
NdrServerInitializeNew( PRPC_MESSAGE pRpcMsg
, PMIDL_STUB_MESSAGE pStubMsg
,
207 PMIDL_STUB_DESC pStubDesc
)
209 TRACE("(pRpcMsg == ^%p, pStubMsg == ^%p, pStubDesc == ^%p)\n", pRpcMsg
, pStubMsg
, pStubDesc
);
211 assert( pRpcMsg
&& pStubMsg
&& pStubDesc
);
213 /* not everyone allocates stack space for w2kReserved */
214 memset(pStubMsg
, 0, FIELD_OFFSET(MIDL_STUB_MESSAGE
,pCSInfo
));
216 pStubMsg
->ReuseBuffer
= TRUE
;
217 pStubMsg
->IsClient
= FALSE
;
218 pStubMsg
->StubDesc
= pStubDesc
;
219 pStubMsg
->pfnAllocate
= pStubDesc
->pfnAllocate
;
220 pStubMsg
->pfnFree
= pStubDesc
->pfnFree
;
221 pStubMsg
->RpcMsg
= pRpcMsg
;
222 pStubMsg
->Buffer
= pStubMsg
->BufferStart
= pRpcMsg
->Buffer
;
223 pStubMsg
->BufferLength
= pRpcMsg
->BufferLength
;
224 pStubMsg
->BufferEnd
= pStubMsg
->Buffer
+ pStubMsg
->BufferLength
;
226 /* FIXME: determine the proper return value */
230 /***********************************************************************
231 * NdrGetBuffer [RPCRT4.@]
233 unsigned char *WINAPI
NdrGetBuffer(MIDL_STUB_MESSAGE
*stubmsg
, unsigned long buflen
, RPC_BINDING_HANDLE handle
)
235 TRACE("(stubmsg == ^%p, buflen == %lu, handle == %p): wild guess.\n", stubmsg
, buflen
, handle
);
237 assert( stubmsg
&& stubmsg
->RpcMsg
);
239 /* I guess this is our chance to put the binding handle into the RPC_MESSAGE */
240 stubmsg
->RpcMsg
->Handle
= handle
;
242 stubmsg
->RpcMsg
->BufferLength
= buflen
;
243 if (I_RpcGetBuffer(stubmsg
->RpcMsg
) != S_OK
)
246 stubmsg
->Buffer
= stubmsg
->BufferStart
= stubmsg
->RpcMsg
->Buffer
;
247 stubmsg
->BufferLength
= stubmsg
->RpcMsg
->BufferLength
;
248 stubmsg
->BufferEnd
= stubmsg
->Buffer
+ stubmsg
->BufferLength
;
249 return (stubmsg
->Buffer
= (unsigned char *)stubmsg
->RpcMsg
->Buffer
);
251 /***********************************************************************
252 * NdrFreeBuffer [RPCRT4.@]
254 void WINAPI
NdrFreeBuffer(MIDL_STUB_MESSAGE
*pStubMsg
)
256 TRACE("(pStubMsg == ^%p): wild guess.\n", pStubMsg
);
257 I_RpcFreeBuffer(pStubMsg
->RpcMsg
);
258 pStubMsg
->BufferLength
= 0;
259 pStubMsg
->Buffer
= pStubMsg
->BufferEnd
= (unsigned char *)(pStubMsg
->RpcMsg
->Buffer
= NULL
);
262 /************************************************************************
263 * NdrSendReceive [RPCRT4.@]
265 unsigned char *WINAPI
NdrSendReceive( MIDL_STUB_MESSAGE
*stubmsg
, unsigned char *buffer
)
267 TRACE("(stubmsg == ^%p, buffer == ^%p)\n", stubmsg
, buffer
);
269 /* FIXME: how to handle errors? (raise exception?) */
271 ERR("NULL stub message. No action taken.\n");
274 if (!stubmsg
->RpcMsg
) {
275 ERR("RPC Message not present in stub message. No action taken.\n");
279 /* FIXME: Seems wrong. Where should this really come from, and when? */
280 stubmsg
->RpcMsg
->DataRepresentation
= NDR_LOCAL_DATA_REPRESENTATION
;
282 if (I_RpcSendReceive(stubmsg
->RpcMsg
) != RPC_S_OK
) {
283 WARN("I_RpcSendReceive did not return success.\n");
284 /* FIXME: raise exception? */
287 /* FIXME: is this the right return value? */