4 * Copyright 2001-2002 Ove Kåven, TransGaming Technologies
5 * Copyright 2004 Filip Navara
7 * This library is free software; you can redistribute it and/or
8 * modify it under the terms of the GNU Lesser General Public
9 * License as published by the Free Software Foundation; either
10 * version 2.1 of the License, or (at your option) any later version.
12 * This library is distributed in the hope that it will be useful,
13 * but WITHOUT ANY WARRANTY; without even the implied warranty of
14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
15 * Lesser General Public License for more details.
17 * You should have received a copy of the GNU Lesser General Public
18 * License along with this library; if not, write to the Free Software
19 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
22 * - figure out whether we *really* got this right
23 * - check for errors and throw exceptions
39 #include "wine/debug.h"
41 #include "rpc_binding.h"
45 WINE_DEFAULT_DEBUG_CHANNEL(ole
);
47 DWORD
RPCRT4_GetHeaderSize(RpcPktHdr
*Header
)
49 static const DWORD header_sizes
[] = {
50 sizeof(Header
->request
), 0, sizeof(Header
->response
),
51 sizeof(Header
->fault
), 0, 0, 0, 0, 0, 0, 0, sizeof(Header
->bind
),
52 sizeof(Header
->bind_ack
), sizeof(Header
->bind_nack
),
57 if (Header
->common
.ptype
< sizeof(header_sizes
) / sizeof(header_sizes
[0])) {
58 ret
= header_sizes
[Header
->common
.ptype
];
60 FIXME("unhandled packet type\n");
61 if (Header
->common
.flags
& RPC_FLG_OBJECT_UUID
)
64 TRACE("invalid packet type\n");
70 VOID
RPCRT4_BuildCommonHeader(RpcPktHdr
*Header
, unsigned char PacketType
,
71 unsigned long DataRepresentation
)
73 Header
->common
.rpc_ver
= RPC_VER_MAJOR
;
74 Header
->common
.rpc_ver_minor
= RPC_VER_MINOR
;
75 Header
->common
.ptype
= PacketType
;
76 Header
->common
.drep
[0] = LOBYTE(LOWORD(DataRepresentation
));
77 Header
->common
.drep
[1] = HIBYTE(LOWORD(DataRepresentation
));
78 Header
->common
.drep
[2] = LOBYTE(HIWORD(DataRepresentation
));
79 Header
->common
.drep
[3] = HIBYTE(HIWORD(DataRepresentation
));
80 Header
->common
.auth_len
= 0;
81 Header
->common
.call_id
= 1;
82 Header
->common
.flags
= 0;
83 /* Flags and fragment length are computed in RPCRT4_Send. */
86 RpcPktHdr
*RPCRT4_BuildRequestHeader(unsigned long DataRepresentation
,
87 unsigned long BufferLength
,
88 unsigned short ProcNum
,
95 has_object
= (ObjectUuid
!= NULL
&& !UuidIsNil(ObjectUuid
, &status
));
96 header
= HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY
,
97 sizeof(header
->request
) + (has_object
? sizeof(UUID
) : 0));
102 RPCRT4_BuildCommonHeader(header
, PKT_REQUEST
, DataRepresentation
);
103 header
->common
.frag_len
= sizeof(header
->request
);
104 header
->request
.alloc_hint
= BufferLength
;
105 header
->request
.context_id
= 0;
106 header
->request
.opnum
= ProcNum
;
108 header
->common
.flags
|= RPC_FLG_OBJECT_UUID
;
109 header
->common
.frag_len
+= sizeof(UUID
);
110 memcpy(&header
->request
+ 1, ObjectUuid
, sizeof(UUID
));
116 RpcPktHdr
*RPCRT4_BuildResponseHeader(unsigned long DataRepresentation
,
117 unsigned long BufferLength
)
121 header
= HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY
, sizeof(header
->response
));
122 if (header
== NULL
) {
126 RPCRT4_BuildCommonHeader(header
, PKT_RESPONSE
, DataRepresentation
);
127 header
->common
.frag_len
= sizeof(header
->response
);
128 header
->response
.alloc_hint
= BufferLength
;
133 RpcPktHdr
*RPCRT4_BuildFaultHeader(unsigned long DataRepresentation
,
138 header
= HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY
, sizeof(header
->fault
));
139 if (header
== NULL
) {
143 RPCRT4_BuildCommonHeader(header
, PKT_FAULT
, DataRepresentation
);
144 header
->common
.frag_len
= sizeof(header
->fault
);
145 header
->fault
.status
= Status
;
150 RpcPktHdr
*RPCRT4_BuildBindHeader(unsigned long DataRepresentation
,
151 unsigned short MaxTransmissionSize
,
152 unsigned short MaxReceiveSize
,
153 RPC_SYNTAX_IDENTIFIER
*AbstractId
,
154 RPC_SYNTAX_IDENTIFIER
*TransferId
)
158 header
= HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY
, sizeof(header
->bind
));
159 if (header
== NULL
) {
163 RPCRT4_BuildCommonHeader(header
, PKT_BIND
, DataRepresentation
);
164 header
->common
.frag_len
= sizeof(header
->bind
);
165 header
->bind
.max_tsize
= MaxTransmissionSize
;
166 header
->bind
.max_rsize
= MaxReceiveSize
;
167 header
->bind
.num_elements
= 1;
168 header
->bind
.num_syntaxes
= 1;
169 memcpy(&header
->bind
.abstract
, AbstractId
, sizeof(RPC_SYNTAX_IDENTIFIER
));
170 memcpy(&header
->bind
.transfer
, TransferId
, sizeof(RPC_SYNTAX_IDENTIFIER
));
175 RpcPktHdr
*RPCRT4_BuildBindNackHeader(unsigned long DataRepresentation
,
176 unsigned char RpcVersion
,
177 unsigned char RpcVersionMinor
)
181 header
= HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY
, sizeof(header
->bind_nack
));
182 if (header
== NULL
) {
186 RPCRT4_BuildCommonHeader(header
, PKT_BIND_NACK
, DataRepresentation
);
187 header
->common
.frag_len
= sizeof(header
->bind_nack
);
188 header
->bind_nack
.protocols_count
= 1;
189 header
->bind_nack
.protocols
[0].rpc_ver
= RpcVersion
;
190 header
->bind_nack
.protocols
[0].rpc_ver_minor
= RpcVersionMinor
;
195 RpcPktHdr
*RPCRT4_BuildBindAckHeader(unsigned long DataRepresentation
,
196 unsigned short MaxTransmissionSize
,
197 unsigned short MaxReceiveSize
,
199 unsigned long Result
,
200 unsigned long Reason
,
201 RPC_SYNTAX_IDENTIFIER
*TransferId
)
204 unsigned long header_size
;
205 RpcAddressString
*server_address
;
207 RPC_SYNTAX_IDENTIFIER
*transfer_id
;
209 header_size
= sizeof(header
->bind_ack
) + sizeof(RpcResults
) +
210 sizeof(RPC_SYNTAX_IDENTIFIER
) + sizeof(RpcAddressString
) +
211 strlen(ServerAddress
);
213 header
= HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY
, header_size
);
214 if (header
== NULL
) {
218 RPCRT4_BuildCommonHeader(header
, PKT_BIND_ACK
, DataRepresentation
);
219 header
->common
.frag_len
= header_size
;
220 header
->bind_ack
.max_tsize
= MaxTransmissionSize
;
221 header
->bind_ack
.max_rsize
= MaxReceiveSize
;
222 server_address
= (RpcAddressString
*)(&header
->bind_ack
+ 1);
223 server_address
->length
= strlen(ServerAddress
) + 1;
224 strcpy(server_address
->string
, ServerAddress
);
225 results
= (RpcResults
*)((ULONG_PTR
)server_address
+ sizeof(RpcAddressString
) + server_address
->length
- 1);
226 results
->num_results
= 1;
227 results
->results
[0].result
= Result
;
228 results
->results
[0].reason
= Reason
;
229 transfer_id
= (RPC_SYNTAX_IDENTIFIER
*)(results
+ 1);
230 memcpy(transfer_id
, TransferId
, sizeof(RPC_SYNTAX_IDENTIFIER
));
235 VOID
RPCRT4_FreeHeader(RpcPktHdr
*Header
)
237 HeapFree(GetProcessHeap(), 0, Header
);
240 /***********************************************************************
241 * RPCRT4_Send (internal)
243 * Transmit a packet over connection in acceptable fragments.
245 RPC_STATUS
RPCRT4_Send(RpcConnection
*Connection
, RpcPktHdr
*Header
,
246 void *Buffer
, unsigned int BufferLength
)
249 DWORD hdr_size
, count
;
252 /* The packet building functions save the packet header size, so we can use it. */
253 hdr_size
= Header
->common
.frag_len
;
254 Header
->common
.flags
|= RPC_FLG_FIRST
;
255 Header
->common
.flags
&= ~RPC_FLG_LAST
;
256 while (!(Header
->common
.flags
& RPC_FLG_LAST
)) {
257 /* decide if we need to split the packet into fragments */
258 if ((BufferLength
+ hdr_size
) <= Connection
->MaxTransmissionSize
) {
259 Header
->common
.flags
|= RPC_FLG_LAST
;
260 Header
->common
.frag_len
= BufferLength
+ hdr_size
;
262 Header
->common
.frag_len
= Connection
->MaxTransmissionSize
;
263 buffer_pos
+= Header
->common
.frag_len
- hdr_size
;
264 BufferLength
-= Header
->common
.frag_len
- hdr_size
;
267 /* transmit packet header */
268 if (!WriteFile(Connection
->conn
, Header
, hdr_size
, &count
, NULL
)) {
269 WARN("WriteFile failed with error %ld\n", GetLastError());
270 return GetLastError();
273 /* fragment consisted of header only and is the last one */
274 if (hdr_size
== Header
->common
.frag_len
&&
275 Header
->common
.flags
& RPC_FLG_LAST
) {
279 /* send the fragment data */
280 if (!WriteFile(Connection
->conn
, buffer_pos
, Header
->common
.frag_len
- hdr_size
, &count
, NULL
)) {
281 WARN("WriteFile failed with error %ld\n", GetLastError());
282 return GetLastError();
285 Header
->common
.flags
&= ~RPC_FLG_FIRST
;
291 /***********************************************************************
292 * RPCRT4_Receive (internal)
294 * Receive a packet from connection and merge the fragments.
296 RPC_STATUS
RPCRT4_Receive(RpcConnection
*Connection
, RpcPktHdr
**Header
,
300 DWORD dwRead
, hdr_length
;
301 unsigned short first_flag
;
302 unsigned long data_length
;
303 unsigned long buffer_length
;
304 unsigned char *buffer_ptr
;
305 RpcPktCommonHdr common_hdr
;
309 TRACE("(%p, %p, %p)\n", Connection
, Header
, pMsg
);
311 /* read packet common header */
312 if (!ReadFile(Connection
->conn
, &common_hdr
, sizeof(common_hdr
), &dwRead
, NULL
)) {
313 if (GetLastError() != ERROR_MORE_DATA
) {
314 WARN("ReadFile failed with error %ld\n", GetLastError());
315 status
= RPC_S_PROTOCOL_ERROR
;
319 if (dwRead
!= sizeof(common_hdr
)) {
320 status
= RPC_S_PROTOCOL_ERROR
;
324 /* verify if the header really makes sense */
325 if (common_hdr
.rpc_ver
!= RPC_VER_MAJOR
||
326 common_hdr
.rpc_ver_minor
!= RPC_VER_MINOR
) {
327 WARN("unhandled packet version\n");
328 status
= RPC_S_PROTOCOL_ERROR
;
332 hdr_length
= RPCRT4_GetHeaderSize((RpcPktHdr
*)&common_hdr
);
333 if (hdr_length
== 0) {
334 status
= RPC_S_PROTOCOL_ERROR
;
338 *Header
= HeapAlloc(GetProcessHeap(), 0, hdr_length
);
339 memcpy(*Header
, &common_hdr
, sizeof(common_hdr
));
341 /* read the rest of packet header */
342 if (!ReadFile(Connection
->conn
, &(*Header
)->common
+ 1,
343 hdr_length
- sizeof(common_hdr
), &dwRead
, NULL
)) {
344 if (GetLastError() != ERROR_MORE_DATA
) {
345 WARN("ReadFile failed with error %ld\n", GetLastError());
346 status
= RPC_S_PROTOCOL_ERROR
;
350 if (dwRead
!= hdr_length
- sizeof(common_hdr
)) {
351 status
= RPC_S_PROTOCOL_ERROR
;
355 /* read packet body */
356 switch (common_hdr
.ptype
) {
358 pMsg
->BufferLength
= (*Header
)->response
.alloc_hint
;
361 pMsg
->BufferLength
= (*Header
)->request
.alloc_hint
;
364 pMsg
->BufferLength
= common_hdr
.frag_len
- hdr_length
;
366 status
= I_RpcGetBuffer(pMsg
);
367 if (status
!= RPC_S_OK
) goto fail
;
369 first_flag
= RPC_FLG_FIRST
;
371 buffer_ptr
= pMsg
->Buffer
;
372 while (buffer_length
< pMsg
->BufferLength
)
374 data_length
= (*Header
)->common
.frag_len
- hdr_length
;
375 if (((*Header
)->common
.flags
& RPC_FLG_FIRST
) != first_flag
||
376 data_length
+ buffer_length
> pMsg
->BufferLength
) {
377 TRACE("invalid packet flags or buffer length\n");
378 status
= RPC_S_PROTOCOL_ERROR
;
382 if (data_length
== 0) dwRead
= 0; else
383 if (!ReadFile(Connection
->conn
, buffer_ptr
, data_length
, &dwRead
, NULL
)) {
384 if (GetLastError() != ERROR_MORE_DATA
) {
385 WARN("ReadFile failed with error %ld\n", GetLastError());
386 status
= RPC_S_PROTOCOL_ERROR
;
390 if (dwRead
!= data_length
) {
391 status
= RPC_S_PROTOCOL_ERROR
;
395 if (buffer_length
== pMsg
->BufferLength
&&
396 ((*Header
)->common
.flags
& RPC_FLG_LAST
) == 0) {
397 status
= RPC_S_PROTOCOL_ERROR
;
401 buffer_length
+= data_length
;
402 if (buffer_length
< pMsg
->BufferLength
) {
403 TRACE("next header\n");
405 /* read the header of next packet */
406 if (!ReadFile(Connection
->conn
, *Header
, hdr_length
, &dwRead
, NULL
)) {
407 if (GetLastError() != ERROR_MORE_DATA
) {
408 WARN("ReadFile failed with error %ld\n", GetLastError());
409 status
= GetLastError();
413 if (dwRead
!= hdr_length
) {
414 WARN("invalid packet header size (%ld)\n", dwRead
);
415 status
= RPC_S_PROTOCOL_ERROR
;
419 buffer_ptr
+= data_length
;
428 if (status
!= RPC_S_OK
&& *Header
) {
429 RPCRT4_FreeHeader(*Header
);
435 /***********************************************************************
436 * I_RpcGetBuffer [RPCRT4.@]
438 RPC_STATUS WINAPI
I_RpcGetBuffer(PRPC_MESSAGE pMsg
)
440 RpcBinding
* bind
= (RpcBinding
*)pMsg
->Handle
;
442 TRACE("(%p): BufferLength=%d\n", pMsg
, pMsg
->BufferLength
);
443 /* FIXME: pfnAllocate? */
445 /* it turns out that the original buffer data must still be available
446 * while the RPC server is marshalling a reply, so we should not deallocate
447 * it, we'll leave deallocating the original buffer to the RPC server */
448 pMsg
->Buffer
= HeapAlloc(GetProcessHeap(), 0, pMsg
->BufferLength
);
450 HeapFree(GetProcessHeap(), 0, pMsg
->Buffer
);
451 pMsg
->Buffer
= HeapAlloc(GetProcessHeap(), 0, pMsg
->BufferLength
);
453 TRACE("Buffer=%p\n", pMsg
->Buffer
);
454 /* FIXME: which errors to return? */
455 return pMsg
->Buffer
? S_OK
: E_OUTOFMEMORY
;
458 /***********************************************************************
459 * I_RpcFreeBuffer [RPCRT4.@]
461 RPC_STATUS WINAPI
I_RpcFreeBuffer(PRPC_MESSAGE pMsg
)
463 TRACE("(%p) Buffer=%p\n", pMsg
, pMsg
->Buffer
);
464 /* FIXME: pfnFree? */
465 HeapFree(GetProcessHeap(), 0, pMsg
->Buffer
);
470 /***********************************************************************
471 * I_RpcSend [RPCRT4.@]
473 RPC_STATUS WINAPI
I_RpcSend(PRPC_MESSAGE pMsg
)
475 RpcBinding
* bind
= (RpcBinding
*)pMsg
->Handle
;
477 RPC_CLIENT_INTERFACE
* cif
= NULL
;
478 RPC_SERVER_INTERFACE
* sif
= NULL
;
482 TRACE("(%p)\n", pMsg
);
483 if (!bind
) return RPC_S_INVALID_BINDING
;
486 sif
= pMsg
->RpcInterfaceInformation
;
487 if (!sif
) return RPC_S_INTERFACE_NOT_FOUND
; /* ? */
488 status
= RPCRT4_OpenBinding(bind
, &conn
, &sif
->TransferSyntax
,
491 cif
= pMsg
->RpcInterfaceInformation
;
492 if (!cif
) return RPC_S_INTERFACE_NOT_FOUND
; /* ? */
493 status
= RPCRT4_OpenBinding(bind
, &conn
, &cif
->TransferSyntax
,
497 if (status
!= RPC_S_OK
) return status
;
500 if (pMsg
->RpcFlags
& WINE_RPCFLAG_EXCEPTION
) {
501 hdr
= RPCRT4_BuildFaultHeader(pMsg
->DataRepresentation
,
504 hdr
= RPCRT4_BuildResponseHeader(pMsg
->DataRepresentation
,
508 hdr
= RPCRT4_BuildRequestHeader(pMsg
->DataRepresentation
,
509 pMsg
->BufferLength
, pMsg
->ProcNum
,
513 status
= RPCRT4_Send(conn
, hdr
, pMsg
->Buffer
, pMsg
->BufferLength
);
515 RPCRT4_FreeHeader(hdr
);
519 /* save the connection, so the response can be read from it */
520 pMsg
->ReservedForRuntime
= conn
;
523 RPCRT4_CloseBinding(bind
, conn
);
529 /***********************************************************************
530 * I_RpcReceive [RPCRT4.@]
532 RPC_STATUS WINAPI
I_RpcReceive(PRPC_MESSAGE pMsg
)
534 RpcBinding
* bind
= (RpcBinding
*)pMsg
->Handle
;
536 RPC_CLIENT_INTERFACE
* cif
= NULL
;
537 RPC_SERVER_INTERFACE
* sif
= NULL
;
539 RpcPktHdr
*hdr
= NULL
;
541 TRACE("(%p)\n", pMsg
);
542 if (!bind
) return RPC_S_INVALID_BINDING
;
544 if (pMsg
->ReservedForRuntime
) {
545 conn
= pMsg
->ReservedForRuntime
;
546 pMsg
->ReservedForRuntime
= NULL
;
549 sif
= pMsg
->RpcInterfaceInformation
;
550 if (!sif
) return RPC_S_INTERFACE_NOT_FOUND
; /* ? */
551 status
= RPCRT4_OpenBinding(bind
, &conn
, &sif
->TransferSyntax
,
554 cif
= pMsg
->RpcInterfaceInformation
;
555 if (!cif
) return RPC_S_INTERFACE_NOT_FOUND
; /* ? */
556 status
= RPCRT4_OpenBinding(bind
, &conn
, &cif
->TransferSyntax
,
559 if (status
!= RPC_S_OK
) return status
;
562 status
= RPCRT4_Receive(conn
, &hdr
, pMsg
);
563 if (status
!= RPC_S_OK
) {
564 WARN("receive failed with error %lx\n", status
);
568 status
= RPC_S_PROTOCOL_ERROR
;
570 switch (hdr
->common
.ptype
) {
572 if (bind
->server
) goto fail
;
575 if (!bind
->server
) goto fail
;
578 pMsg
->RpcFlags
|= WINE_RPCFLAG_EXCEPTION
;
579 ERR ("we got fault packet with status %lx\n", hdr
->fault
.status
);
580 status
= RPC_S_CALL_FAILED
; /* ? */
591 RPCRT4_FreeHeader(hdr
);
593 RPCRT4_CloseBinding(bind
, conn
);
597 /***********************************************************************
598 * I_RpcSendReceive [RPCRT4.@]
600 RPC_STATUS WINAPI
I_RpcSendReceive(PRPC_MESSAGE pMsg
)
604 TRACE("(%p)\n", pMsg
);
605 status
= I_RpcSend(pMsg
);
606 if (status
== RPC_S_OK
)
607 status
= I_RpcReceive(pMsg
);