4 * Copyright 2001 Ove Kåven, TransGaming Technologies
5 * Copyright 2002 Marcus Meissner
6 * Copyright 2005 Mike Hearn, Rob Shearman for CodeWeavers
8 * This library is free software; you can redistribute it and/or
9 * modify it under the terms of the GNU Lesser General Public
10 * License as published by the Free Software Foundation; either
11 * version 2.1 of the License, or (at your option) any later version.
13 * This library is distributed in the hope that it will be useful,
14 * but WITHOUT ANY WARRANTY; without even the implied warranty of
15 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
16 * Lesser General Public License for more details.
18 * You should have received a copy of the GNU Lesser General Public
19 * License along with this library; if not, write to the Free Software
20 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
23 #define WIN32_NO_STATUS
27 //#include "wine/port.h"
33 #define NONAMELESSUNION
34 #define NONAMELESSSTRUCT
38 //#include "winuser.h"
40 //#include "objbase.h"
43 //#include "winerror.h"
46 #include <wine/unicode.h>
48 #include "compobj_private.h"
50 #include <wine/debug.h>
52 WINE_DEFAULT_DEBUG_CHANNEL(ole
);
54 static void __RPC_STUB
dispatch_rpc(RPC_MESSAGE
*msg
);
56 /* we only use one function to dispatch calls for all methods - we use the
57 * RPC_IF_OLE flag to tell the RPC runtime that this is the case */
58 static RPC_DISPATCH_FUNCTION rpc_dispatch_table
[1] = { dispatch_rpc
}; /* (RO) */
59 static RPC_DISPATCH_TABLE rpc_dispatch
= { 1, rpc_dispatch_table
}; /* (RO) */
61 static struct list registered_interfaces
= LIST_INIT(registered_interfaces
); /* (CS csRegIf) */
62 static CRITICAL_SECTION csRegIf
;
63 static CRITICAL_SECTION_DEBUG csRegIf_debug
=
66 { &csRegIf_debug
.ProcessLocksList
, &csRegIf_debug
.ProcessLocksList
},
67 0, 0, { (DWORD_PTR
)(__FILE__
": dcom registered server interfaces") }
69 static CRITICAL_SECTION csRegIf
= { &csRegIf_debug
, -1, 0, 0, 0, 0 };
71 static struct list channel_hooks
= LIST_INIT(channel_hooks
); /* (CS csChannelHook) */
72 static CRITICAL_SECTION csChannelHook
;
73 static CRITICAL_SECTION_DEBUG csChannelHook_debug
=
76 { &csChannelHook_debug
.ProcessLocksList
, &csChannelHook_debug
.ProcessLocksList
},
77 0, 0, { (DWORD_PTR
)(__FILE__
": channel hooks") }
79 static CRITICAL_SECTION csChannelHook
= { &csChannelHook_debug
, -1, 0, 0, 0, 0 };
81 static WCHAR wszRpcTransport
[] = {'n','c','a','l','r','p','c',0};
87 DWORD refs
; /* ref count */
88 RPC_SERVER_INTERFACE If
; /* interface registered with the RPC runtime */
91 /* get the pipe endpoint specified of the specified apartment */
92 static inline void get_rpc_endpoint(LPWSTR endpoint
, const OXID
*oxid
)
94 /* FIXME: should get endpoint from rpcss */
95 static const WCHAR wszEndpointFormat
[] = {'\\','p','i','p','e','\\','O','L','E','_','%','0','8','l','x','%','0','8','l','x',0};
96 wsprintfW(endpoint
, wszEndpointFormat
, (DWORD
)(*oxid
>> 32),(DWORD
)*oxid
);
101 IRpcChannelBuffer IRpcChannelBuffer_iface
;
104 DWORD dest_context
; /* returned from GetDestCtx */
105 void *dest_context_data
; /* returned from GetDestCtx */
110 RpcChannelBuffer super
; /* superclass */
112 RPC_BINDING_HANDLE bind
; /* handle to the remote server */
113 OXID oxid
; /* apartment in which the channel is valid */
114 DWORD server_pid
; /* id of server process */
115 HANDLE event
; /* cached event handle */
116 } ClientRpcChannelBuffer
;
118 struct dispatch_params
120 RPCOLEMESSAGE
*msg
; /* message */
121 IRpcStubBuffer
*stub
; /* stub buffer, if applicable */
122 IRpcChannelBuffer
*chan
; /* server channel buffer, if applicable */
123 IID iid
; /* ID of interface being called */
124 IUnknown
*iface
; /* interface being called */
125 HANDLE handle
; /* handle that will become signaled when call finishes */
126 BOOL bypass_rpcrt
; /* bypass RPC runtime? */
127 RPC_STATUS status
; /* status (out) */
128 HRESULT hr
; /* hresult (out) */
133 RPC_BINDING_HANDLE binding_handle
;
134 ULONG prefix_data_len
;
135 SChannelHookCallInfo channel_hook_info
;
141 struct dispatch_params params
;
146 ULONG conformance
; /* NDR */
149 /* [size_is((size+7)&~7)] */ unsigned char data
[1];
156 unsigned char extent
[1];
157 } WIRE_ORPC_EXTENT_ARRAY
;
165 unsigned char extensions
[1];
171 unsigned char extensions
[1];
174 struct channel_hook_entry
181 struct channel_hook_buffer_data
184 ULONG extension_size
;
188 static HRESULT
unmarshal_ORPCTHAT(RPC_MESSAGE
*msg
, ORPCTHAT
*orpcthat
,
189 ORPC_EXTENT_ARRAY
*orpc_ext_array
, WIRE_ORPC_EXTENT
**first_wire_orpc_extent
);
191 /* Channel Hook Functions */
193 static ULONG
ChannelHooks_ClientGetSize(SChannelHookCallInfo
*info
,
194 struct channel_hook_buffer_data
**data
, unsigned int *hook_count
,
195 ULONG
*extension_count
)
197 struct channel_hook_entry
*entry
;
198 ULONG total_size
= 0;
199 unsigned int hook_index
= 0;
202 *extension_count
= 0;
204 EnterCriticalSection(&csChannelHook
);
206 LIST_FOR_EACH_ENTRY(entry
, &channel_hooks
, struct channel_hook_entry
, entry
)
210 *data
= HeapAlloc(GetProcessHeap(), 0, *hook_count
* sizeof(struct channel_hook_buffer_data
));
214 LIST_FOR_EACH_ENTRY(entry
, &channel_hooks
, struct channel_hook_entry
, entry
)
216 ULONG extension_size
= 0;
218 IChannelHook_ClientGetSize(entry
->hook
, &entry
->id
, &info
->iid
, &extension_size
);
220 TRACE("%s: extension_size = %u\n", debugstr_guid(&entry
->id
), extension_size
);
222 extension_size
= (extension_size
+7)&~7;
223 (*data
)[hook_index
].id
= entry
->id
;
224 (*data
)[hook_index
].extension_size
= extension_size
;
226 /* an extension is only put onto the wire if it has data to write */
229 total_size
+= FIELD_OFFSET(WIRE_ORPC_EXTENT
, data
[extension_size
]);
230 (*extension_count
)++;
236 LeaveCriticalSection(&csChannelHook
);
241 static unsigned char * ChannelHooks_ClientFillBuffer(SChannelHookCallInfo
*info
,
242 unsigned char *buffer
, struct channel_hook_buffer_data
*data
,
243 unsigned int hook_count
)
245 struct channel_hook_entry
*entry
;
247 EnterCriticalSection(&csChannelHook
);
249 LIST_FOR_EACH_ENTRY(entry
, &channel_hooks
, struct channel_hook_entry
, entry
)
252 ULONG extension_size
= 0;
253 WIRE_ORPC_EXTENT
*wire_orpc_extent
= (WIRE_ORPC_EXTENT
*)buffer
;
255 for (i
= 0; i
< hook_count
; i
++)
256 if (IsEqualGUID(&entry
->id
, &data
[i
].id
))
257 extension_size
= data
[i
].extension_size
;
259 /* an extension is only put onto the wire if it has data to write */
263 IChannelHook_ClientFillBuffer(entry
->hook
, &entry
->id
, &info
->iid
,
264 &extension_size
, buffer
+ FIELD_OFFSET(WIRE_ORPC_EXTENT
, data
[0]));
266 TRACE("%s: extension_size = %u\n", debugstr_guid(&entry
->id
), extension_size
);
268 /* FIXME: set unused portion of wire_orpc_extent->data to 0? */
270 wire_orpc_extent
->conformance
= (extension_size
+7)&~7;
271 wire_orpc_extent
->size
= extension_size
;
272 wire_orpc_extent
->id
= entry
->id
;
273 buffer
+= FIELD_OFFSET(WIRE_ORPC_EXTENT
, data
[wire_orpc_extent
->conformance
]);
276 LeaveCriticalSection(&csChannelHook
);
281 static void ChannelHooks_ServerNotify(SChannelHookCallInfo
*info
,
282 DWORD lDataRep
, WIRE_ORPC_EXTENT
*first_wire_orpc_extent
,
283 ULONG extension_count
)
285 struct channel_hook_entry
*entry
;
288 EnterCriticalSection(&csChannelHook
);
290 LIST_FOR_EACH_ENTRY(entry
, &channel_hooks
, struct channel_hook_entry
, entry
)
292 WIRE_ORPC_EXTENT
*wire_orpc_extent
;
293 for (i
= 0, wire_orpc_extent
= first_wire_orpc_extent
;
295 i
++, wire_orpc_extent
= (WIRE_ORPC_EXTENT
*)&wire_orpc_extent
->data
[wire_orpc_extent
->conformance
])
297 if (IsEqualGUID(&entry
->id
, &wire_orpc_extent
->id
))
300 if (i
== extension_count
) wire_orpc_extent
= NULL
;
302 IChannelHook_ServerNotify(entry
->hook
, &entry
->id
, &info
->iid
,
303 wire_orpc_extent
? wire_orpc_extent
->size
: 0,
304 wire_orpc_extent
? wire_orpc_extent
->data
: NULL
,
308 LeaveCriticalSection(&csChannelHook
);
311 static ULONG
ChannelHooks_ServerGetSize(SChannelHookCallInfo
*info
,
312 struct channel_hook_buffer_data
**data
, unsigned int *hook_count
,
313 ULONG
*extension_count
)
315 struct channel_hook_entry
*entry
;
316 ULONG total_size
= 0;
317 unsigned int hook_index
= 0;
320 *extension_count
= 0;
322 EnterCriticalSection(&csChannelHook
);
324 LIST_FOR_EACH_ENTRY(entry
, &channel_hooks
, struct channel_hook_entry
, entry
)
328 *data
= HeapAlloc(GetProcessHeap(), 0, *hook_count
* sizeof(struct channel_hook_buffer_data
));
332 LIST_FOR_EACH_ENTRY(entry
, &channel_hooks
, struct channel_hook_entry
, entry
)
334 ULONG extension_size
= 0;
336 IChannelHook_ServerGetSize(entry
->hook
, &entry
->id
, &info
->iid
, S_OK
,
339 TRACE("%s: extension_size = %u\n", debugstr_guid(&entry
->id
), extension_size
);
341 extension_size
= (extension_size
+7)&~7;
342 (*data
)[hook_index
].id
= entry
->id
;
343 (*data
)[hook_index
].extension_size
= extension_size
;
345 /* an extension is only put onto the wire if it has data to write */
348 total_size
+= FIELD_OFFSET(WIRE_ORPC_EXTENT
, data
[extension_size
]);
349 (*extension_count
)++;
355 LeaveCriticalSection(&csChannelHook
);
360 static unsigned char * ChannelHooks_ServerFillBuffer(SChannelHookCallInfo
*info
,
361 unsigned char *buffer
, struct channel_hook_buffer_data
*data
,
362 unsigned int hook_count
)
364 struct channel_hook_entry
*entry
;
366 EnterCriticalSection(&csChannelHook
);
368 LIST_FOR_EACH_ENTRY(entry
, &channel_hooks
, struct channel_hook_entry
, entry
)
371 ULONG extension_size
= 0;
372 WIRE_ORPC_EXTENT
*wire_orpc_extent
= (WIRE_ORPC_EXTENT
*)buffer
;
374 for (i
= 0; i
< hook_count
; i
++)
375 if (IsEqualGUID(&entry
->id
, &data
[i
].id
))
376 extension_size
= data
[i
].extension_size
;
378 /* an extension is only put onto the wire if it has data to write */
382 IChannelHook_ServerFillBuffer(entry
->hook
, &entry
->id
, &info
->iid
,
383 &extension_size
, buffer
+ FIELD_OFFSET(WIRE_ORPC_EXTENT
, data
[0]),
386 TRACE("%s: extension_size = %u\n", debugstr_guid(&entry
->id
), extension_size
);
388 /* FIXME: set unused portion of wire_orpc_extent->data to 0? */
390 wire_orpc_extent
->conformance
= (extension_size
+7)&~7;
391 wire_orpc_extent
->size
= extension_size
;
392 wire_orpc_extent
->id
= entry
->id
;
393 buffer
+= FIELD_OFFSET(WIRE_ORPC_EXTENT
, data
[wire_orpc_extent
->conformance
]);
396 LeaveCriticalSection(&csChannelHook
);
401 static void ChannelHooks_ClientNotify(SChannelHookCallInfo
*info
,
402 DWORD lDataRep
, WIRE_ORPC_EXTENT
*first_wire_orpc_extent
,
403 ULONG extension_count
, HRESULT hrFault
)
405 struct channel_hook_entry
*entry
;
408 EnterCriticalSection(&csChannelHook
);
410 LIST_FOR_EACH_ENTRY(entry
, &channel_hooks
, struct channel_hook_entry
, entry
)
412 WIRE_ORPC_EXTENT
*wire_orpc_extent
;
413 for (i
= 0, wire_orpc_extent
= first_wire_orpc_extent
;
415 i
++, wire_orpc_extent
= (WIRE_ORPC_EXTENT
*)&wire_orpc_extent
->data
[wire_orpc_extent
->conformance
])
417 if (IsEqualGUID(&entry
->id
, &wire_orpc_extent
->id
))
420 if (i
== extension_count
) wire_orpc_extent
= NULL
;
422 IChannelHook_ClientNotify(entry
->hook
, &entry
->id
, &info
->iid
,
423 wire_orpc_extent
? wire_orpc_extent
->size
: 0,
424 wire_orpc_extent
? wire_orpc_extent
->data
: NULL
,
428 LeaveCriticalSection(&csChannelHook
);
431 HRESULT
RPC_RegisterChannelHook(REFGUID rguid
, IChannelHook
*hook
)
433 struct channel_hook_entry
*entry
;
435 TRACE("(%s, %p)\n", debugstr_guid(rguid
), hook
);
437 entry
= HeapAlloc(GetProcessHeap(), 0, sizeof(*entry
));
439 return E_OUTOFMEMORY
;
443 IChannelHook_AddRef(hook
);
445 EnterCriticalSection(&csChannelHook
);
446 list_add_tail(&channel_hooks
, &entry
->entry
);
447 LeaveCriticalSection(&csChannelHook
);
452 void RPC_UnregisterAllChannelHooks(void)
454 struct channel_hook_entry
*cursor
;
455 struct channel_hook_entry
*cursor2
;
457 EnterCriticalSection(&csChannelHook
);
458 LIST_FOR_EACH_ENTRY_SAFE(cursor
, cursor2
, &channel_hooks
, struct channel_hook_entry
, entry
)
459 HeapFree(GetProcessHeap(), 0, cursor
);
460 LeaveCriticalSection(&csChannelHook
);
461 DeleteCriticalSection(&csChannelHook
);
462 DeleteCriticalSection(&csRegIf
);
465 /* RPC Channel Buffer Functions */
467 static HRESULT WINAPI
RpcChannelBuffer_QueryInterface(IRpcChannelBuffer
*iface
, REFIID riid
, LPVOID
*ppv
)
470 if (IsEqualIID(riid
,&IID_IRpcChannelBuffer
) || IsEqualIID(riid
,&IID_IUnknown
))
473 IRpcChannelBuffer_AddRef(iface
);
476 return E_NOINTERFACE
;
479 static ULONG WINAPI
RpcChannelBuffer_AddRef(LPRPCCHANNELBUFFER iface
)
481 RpcChannelBuffer
*This
= (RpcChannelBuffer
*)iface
;
482 return InterlockedIncrement(&This
->refs
);
485 static ULONG WINAPI
ServerRpcChannelBuffer_Release(LPRPCCHANNELBUFFER iface
)
487 RpcChannelBuffer
*This
= (RpcChannelBuffer
*)iface
;
490 ref
= InterlockedDecrement(&This
->refs
);
494 HeapFree(GetProcessHeap(), 0, This
);
498 static ULONG WINAPI
ClientRpcChannelBuffer_Release(LPRPCCHANNELBUFFER iface
)
500 ClientRpcChannelBuffer
*This
= (ClientRpcChannelBuffer
*)iface
;
503 ref
= InterlockedDecrement(&This
->super
.refs
);
507 if (This
->event
) CloseHandle(This
->event
);
508 RpcBindingFree(&This
->bind
);
509 HeapFree(GetProcessHeap(), 0, This
);
513 static HRESULT WINAPI
ServerRpcChannelBuffer_GetBuffer(LPRPCCHANNELBUFFER iface
, RPCOLEMESSAGE
* olemsg
, REFIID riid
)
515 RpcChannelBuffer
*This
= (RpcChannelBuffer
*)iface
;
516 RPC_MESSAGE
*msg
= (RPC_MESSAGE
*)olemsg
;
519 struct message_state
*message_state
;
520 ULONG extensions_size
;
521 struct channel_hook_buffer_data
*channel_hook_data
;
522 unsigned int channel_hook_count
;
523 ULONG extension_count
;
525 TRACE("(%p)->(%p,%s)\n", This
, olemsg
, debugstr_guid(riid
));
527 message_state
= msg
->Handle
;
528 /* restore the binding handle and the real start of data */
529 msg
->Handle
= message_state
->binding_handle
;
530 msg
->Buffer
= (char *)msg
->Buffer
- message_state
->prefix_data_len
;
532 extensions_size
= ChannelHooks_ServerGetSize(&message_state
->channel_hook_info
,
533 &channel_hook_data
, &channel_hook_count
, &extension_count
);
535 msg
->BufferLength
+= FIELD_OFFSET(WIRE_ORPCTHAT
, extensions
) + sizeof(DWORD
);
538 msg
->BufferLength
+= FIELD_OFFSET(WIRE_ORPC_EXTENT_ARRAY
, extent
[2*sizeof(DWORD
) + extensions_size
]);
539 if (extension_count
& 1)
540 msg
->BufferLength
+= FIELD_OFFSET(WIRE_ORPC_EXTENT
, data
[0]);
543 if (message_state
->bypass_rpcrt
)
545 msg
->Buffer
= HeapAlloc(GetProcessHeap(), 0, msg
->BufferLength
);
550 HeapFree(GetProcessHeap(), 0, channel_hook_data
);
551 return E_OUTOFMEMORY
;
555 status
= I_RpcGetBuffer(msg
);
557 orpcthat
= msg
->Buffer
;
558 msg
->Buffer
= (char *)msg
->Buffer
+ FIELD_OFFSET(WIRE_ORPCTHAT
, extensions
);
560 orpcthat
->flags
= ORPCF_NULL
/* FIXME? */;
562 /* NDR representation of orpcthat->extensions */
563 *(DWORD
*)msg
->Buffer
= extensions_size
? 1 : 0;
564 msg
->Buffer
= (char *)msg
->Buffer
+ sizeof(DWORD
);
568 WIRE_ORPC_EXTENT_ARRAY
*orpc_extent_array
= msg
->Buffer
;
569 orpc_extent_array
->size
= extension_count
;
570 orpc_extent_array
->reserved
= 0;
571 msg
->Buffer
= (char *)msg
->Buffer
+ FIELD_OFFSET(WIRE_ORPC_EXTENT_ARRAY
, extent
);
572 /* NDR representation of orpc_extent_array->extent */
573 *(DWORD
*)msg
->Buffer
= 1;
574 msg
->Buffer
= (char *)msg
->Buffer
+ sizeof(DWORD
);
575 /* NDR representation of [size_is] attribute of orpc_extent_array->extent */
576 *(DWORD
*)msg
->Buffer
= (extension_count
+ 1) & ~1;
577 msg
->Buffer
= (char *)msg
->Buffer
+ sizeof(DWORD
);
579 msg
->Buffer
= ChannelHooks_ServerFillBuffer(&message_state
->channel_hook_info
,
580 msg
->Buffer
, channel_hook_data
, channel_hook_count
);
582 /* we must add a dummy extension if there is an odd extension
583 * count to meet the contract specified by the size_is attribute */
584 if (extension_count
& 1)
586 WIRE_ORPC_EXTENT
*wire_orpc_extent
= msg
->Buffer
;
587 wire_orpc_extent
->conformance
= 0;
588 wire_orpc_extent
->id
= GUID_NULL
;
589 wire_orpc_extent
->size
= 0;
590 msg
->Buffer
= (char *)msg
->Buffer
+ FIELD_OFFSET(WIRE_ORPC_EXTENT
, data
[0]);
594 HeapFree(GetProcessHeap(), 0, channel_hook_data
);
596 /* store the prefixed data length so that we can restore the real buffer
598 message_state
->prefix_data_len
= (char *)msg
->Buffer
- (char *)orpcthat
;
599 msg
->BufferLength
-= message_state
->prefix_data_len
;
600 /* save away the message state again */
601 msg
->Handle
= message_state
;
603 TRACE("-- %d\n", status
);
605 return HRESULT_FROM_WIN32(status
);
608 static HANDLE
ClientRpcChannelBuffer_GetEventHandle(ClientRpcChannelBuffer
*This
)
610 HANDLE event
= InterlockedExchangePointer(&This
->event
, NULL
);
612 /* Note: must be auto-reset event so we can reuse it without a call
614 if (!event
) event
= CreateEventW(NULL
, FALSE
, FALSE
, NULL
);
619 static void ClientRpcChannelBuffer_ReleaseEventHandle(ClientRpcChannelBuffer
*This
, HANDLE event
)
621 if (InterlockedCompareExchangePointer(&This
->event
, event
, NULL
))
622 /* already a handle cached in This */
626 static HRESULT WINAPI
ClientRpcChannelBuffer_GetBuffer(LPRPCCHANNELBUFFER iface
, RPCOLEMESSAGE
* olemsg
, REFIID riid
)
628 ClientRpcChannelBuffer
*This
= (ClientRpcChannelBuffer
*)iface
;
629 RPC_MESSAGE
*msg
= (RPC_MESSAGE
*)olemsg
;
630 RPC_CLIENT_INTERFACE
*cif
;
633 struct message_state
*message_state
;
634 ULONG extensions_size
;
635 struct channel_hook_buffer_data
*channel_hook_data
;
636 unsigned int channel_hook_count
;
637 ULONG extension_count
;
640 APARTMENT
*apt
= NULL
;
642 TRACE("(%p)->(%p,%s)\n", This
, olemsg
, debugstr_guid(riid
));
644 cif
= HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY
, sizeof(RPC_CLIENT_INTERFACE
));
646 return E_OUTOFMEMORY
;
648 message_state
= HeapAlloc(GetProcessHeap(), 0, sizeof(*message_state
));
651 HeapFree(GetProcessHeap(), 0, cif
);
652 return E_OUTOFMEMORY
;
655 cif
->Length
= sizeof(RPC_CLIENT_INTERFACE
);
656 /* RPC interface ID = COM interface ID */
657 cif
->InterfaceId
.SyntaxGUID
= *riid
;
658 /* COM objects always have a version of 0.0 */
659 cif
->InterfaceId
.SyntaxVersion
.MajorVersion
= 0;
660 cif
->InterfaceId
.SyntaxVersion
.MinorVersion
= 0;
661 msg
->Handle
= This
->bind
;
662 msg
->RpcInterfaceInformation
= cif
;
664 message_state
->prefix_data_len
= 0;
665 message_state
->binding_handle
= This
->bind
;
667 message_state
->channel_hook_info
.iid
= *riid
;
668 message_state
->channel_hook_info
.cbSize
= sizeof(message_state
->channel_hook_info
);
669 message_state
->channel_hook_info
.uCausality
= COM_CurrentCausalityId();
670 message_state
->channel_hook_info
.dwServerPid
= This
->server_pid
;
671 message_state
->channel_hook_info
.iMethod
= msg
->ProcNum
;
672 message_state
->channel_hook_info
.pObject
= NULL
; /* only present on server-side */
673 message_state
->target_hwnd
= NULL
;
674 message_state
->target_tid
= 0;
675 memset(&message_state
->params
, 0, sizeof(message_state
->params
));
677 extensions_size
= ChannelHooks_ClientGetSize(&message_state
->channel_hook_info
,
678 &channel_hook_data
, &channel_hook_count
, &extension_count
);
680 msg
->BufferLength
+= FIELD_OFFSET(WIRE_ORPCTHIS
, extensions
) + sizeof(DWORD
);
683 msg
->BufferLength
+= FIELD_OFFSET(WIRE_ORPC_EXTENT_ARRAY
, extent
[2*sizeof(DWORD
) + extensions_size
]);
684 if (extension_count
& 1)
685 msg
->BufferLength
+= FIELD_OFFSET(WIRE_ORPC_EXTENT
, data
[0]);
688 RpcBindingInqObject(message_state
->binding_handle
, &ipid
);
689 hr
= ipid_get_dispatch_params(&ipid
, &apt
, &message_state
->params
.stub
,
690 &message_state
->params
.chan
,
691 &message_state
->params
.iid
,
692 &message_state
->params
.iface
);
695 /* stub, chan, iface and iid are unneeded in multi-threaded case as we go
696 * via the RPC runtime */
697 if (apt
->multi_threaded
)
699 IRpcStubBuffer_Release(message_state
->params
.stub
);
700 message_state
->params
.stub
= NULL
;
701 IRpcChannelBuffer_Release(message_state
->params
.chan
);
702 message_state
->params
.chan
= NULL
;
703 message_state
->params
.iface
= NULL
;
707 message_state
->params
.bypass_rpcrt
= TRUE
;
708 message_state
->target_hwnd
= apartment_getwindow(apt
);
709 message_state
->target_tid
= apt
->tid
;
710 /* we assume later on that this being non-NULL is the indicator that
711 * means call directly instead of going through RPC runtime */
712 if (!message_state
->target_hwnd
)
713 ERR("window for apartment %s is NULL\n", wine_dbgstr_longlong(apt
->oxid
));
716 if (apt
) apartment_release(apt
);
717 message_state
->params
.handle
= ClientRpcChannelBuffer_GetEventHandle(This
);
718 /* Note: message_state->params.msg is initialised in
719 * ClientRpcChannelBuffer_SendReceive */
721 /* shortcut the RPC runtime */
722 if (message_state
->target_hwnd
)
724 msg
->Buffer
= HeapAlloc(GetProcessHeap(), 0, msg
->BufferLength
);
728 status
= ERROR_OUTOFMEMORY
;
731 status
= I_RpcGetBuffer(msg
);
733 msg
->Handle
= message_state
;
735 if (status
== RPC_S_OK
)
737 orpcthis
= msg
->Buffer
;
738 msg
->Buffer
= (char *)msg
->Buffer
+ FIELD_OFFSET(WIRE_ORPCTHIS
, extensions
);
740 orpcthis
->version
.MajorVersion
= COM_MAJOR_VERSION
;
741 orpcthis
->version
.MinorVersion
= COM_MINOR_VERSION
;
742 orpcthis
->flags
= message_state
->channel_hook_info
.dwServerPid
? ORPCF_LOCAL
: ORPCF_NULL
;
743 orpcthis
->reserved1
= 0;
744 orpcthis
->cid
= message_state
->channel_hook_info
.uCausality
;
746 /* NDR representation of orpcthis->extensions */
747 *(DWORD
*)msg
->Buffer
= extensions_size
? 1 : 0;
748 msg
->Buffer
= (char *)msg
->Buffer
+ sizeof(DWORD
);
752 ORPC_EXTENT_ARRAY
*orpc_extent_array
= msg
->Buffer
;
753 orpc_extent_array
->size
= extension_count
;
754 orpc_extent_array
->reserved
= 0;
755 msg
->Buffer
= (char *)msg
->Buffer
+ FIELD_OFFSET(WIRE_ORPC_EXTENT_ARRAY
, extent
);
756 /* NDR representation of orpc_extent_array->extent */
757 *(DWORD
*)msg
->Buffer
= 1;
758 msg
->Buffer
= (char *)msg
->Buffer
+ sizeof(DWORD
);
759 /* NDR representation of [size_is] attribute of orpc_extent_array->extent */
760 *(DWORD
*)msg
->Buffer
= (extension_count
+ 1) & ~1;
761 msg
->Buffer
= (char *)msg
->Buffer
+ sizeof(DWORD
);
763 msg
->Buffer
= ChannelHooks_ClientFillBuffer(&message_state
->channel_hook_info
,
764 msg
->Buffer
, channel_hook_data
, channel_hook_count
);
766 /* we must add a dummy extension if there is an odd extension
767 * count to meet the contract specified by the size_is attribute */
768 if (extension_count
& 1)
770 WIRE_ORPC_EXTENT
*wire_orpc_extent
= msg
->Buffer
;
771 wire_orpc_extent
->conformance
= 0;
772 wire_orpc_extent
->id
= GUID_NULL
;
773 wire_orpc_extent
->size
= 0;
774 msg
->Buffer
= (char *)msg
->Buffer
+ FIELD_OFFSET(WIRE_ORPC_EXTENT
, data
[0]);
778 /* store the prefixed data length so that we can restore the real buffer
779 * pointer in ClientRpcChannelBuffer_SendReceive. */
780 message_state
->prefix_data_len
= (char *)msg
->Buffer
- (char *)orpcthis
;
781 msg
->BufferLength
-= message_state
->prefix_data_len
;
784 HeapFree(GetProcessHeap(), 0, channel_hook_data
);
786 TRACE("-- %d\n", status
);
788 return HRESULT_FROM_WIN32(status
);
791 static HRESULT WINAPI
ServerRpcChannelBuffer_SendReceive(LPRPCCHANNELBUFFER iface
, RPCOLEMESSAGE
*olemsg
, ULONG
*pstatus
)
797 /* this thread runs an outgoing RPC */
798 static DWORD WINAPI
rpc_sendreceive_thread(LPVOID param
)
800 struct dispatch_params
*data
= param
;
802 /* Note: I_RpcSendReceive doesn't raise exceptions like the higher-level
803 * RPC functions do */
804 data
->status
= I_RpcSendReceive((RPC_MESSAGE
*)data
->msg
);
806 TRACE("completed with status 0x%x\n", data
->status
);
808 SetEvent(data
->handle
);
813 static inline HRESULT
ClientRpcChannelBuffer_IsCorrectApartment(ClientRpcChannelBuffer
*This
, APARTMENT
*apt
)
818 if (apartment_getoxid(apt
, &oxid
) != S_OK
)
820 if (This
->oxid
!= oxid
)
825 static HRESULT WINAPI
ClientRpcChannelBuffer_SendReceive(LPRPCCHANNELBUFFER iface
, RPCOLEMESSAGE
*olemsg
, ULONG
*pstatus
)
827 ClientRpcChannelBuffer
*This
= (ClientRpcChannelBuffer
*)iface
;
829 RPC_MESSAGE
*msg
= (RPC_MESSAGE
*)olemsg
;
832 struct message_state
*message_state
;
834 ORPC_EXTENT_ARRAY orpc_ext_array
;
835 WIRE_ORPC_EXTENT
*first_wire_orpc_extent
= NULL
;
836 HRESULT hrFault
= S_OK
;
838 TRACE("(%p) iMethod=%d\n", olemsg
, olemsg
->iMethod
);
840 hr
= ClientRpcChannelBuffer_IsCorrectApartment(This
, COM_CurrentApt());
843 ERR("called from wrong apartment, should have been 0x%s\n",
844 wine_dbgstr_longlong(This
->oxid
));
845 return RPC_E_WRONG_THREAD
;
847 /* This situation should be impossible in multi-threaded apartments,
848 * because the calling thread isn't re-enterable.
849 * Note: doing a COM call during the processing of a sent message is
850 * only disallowed if a client call is already being waited for
852 if (!COM_CurrentApt()->multi_threaded
&&
853 COM_CurrentInfo()->pending_call_count_client
&&
856 ERR("can't make an outgoing COM call in response to a sent message\n");
857 return RPC_E_CANTCALLOUT_ININPUTSYNCCALL
;
860 message_state
= msg
->Handle
;
861 /* restore the binding handle and the real start of data */
862 msg
->Handle
= message_state
->binding_handle
;
863 msg
->Buffer
= (char *)msg
->Buffer
- message_state
->prefix_data_len
;
864 msg
->BufferLength
+= message_state
->prefix_data_len
;
866 /* Note: this is an optimization in the Microsoft OLE runtime that we need
867 * to copy, as shown by the test_no_couninitialize_client test. without
868 * short-circuiting the RPC runtime in the case below, the test will
869 * deadlock on the loader lock due to the RPC runtime needing to create
870 * a thread to process the RPC when this function is called indirectly
873 message_state
->params
.msg
= olemsg
;
874 if (message_state
->params
.bypass_rpcrt
)
876 TRACE("Calling apartment thread 0x%08x...\n", message_state
->target_tid
);
878 msg
->ProcNum
&= ~RPC_FLAGS_VALID_BIT
;
880 if (!PostMessageW(message_state
->target_hwnd
, DM_EXECUTERPC
, 0,
881 (LPARAM
)&message_state
->params
))
883 ERR("PostMessage failed with error %u\n", GetLastError());
885 /* Note: message_state->params.iface doesn't have a reference and
886 * so doesn't need to be released */
888 hr
= HRESULT_FROM_WIN32(GetLastError());
893 /* we use a separate thread here because we need to be able to
894 * pump the message loop in the application thread: if we do not,
895 * any windows created by this thread will hang and RPCs that try
896 * and re-enter this STA from an incoming server thread will
897 * deadlock. InstallShield is an example of that.
899 if (!QueueUserWorkItem(rpc_sendreceive_thread
, &message_state
->params
, WT_EXECUTEDEFAULT
))
901 ERR("QueueUserWorkItem failed with error %u\n", GetLastError());
910 if (WaitForSingleObject(message_state
->params
.handle
, 0))
912 COM_CurrentInfo()->pending_call_count_client
++;
913 hr
= CoWaitForMultipleHandles(0, INFINITE
, 1, &message_state
->params
.handle
, &index
);
914 COM_CurrentInfo()->pending_call_count_client
--;
917 ClientRpcChannelBuffer_ReleaseEventHandle(This
, message_state
->params
.handle
);
919 /* for WM shortcut, faults are returned in params->hr */
921 hrFault
= message_state
->params
.hr
;
923 status
= message_state
->params
.status
;
925 orpcthat
.flags
= ORPCF_NULL
;
926 orpcthat
.extensions
= NULL
;
928 TRACE("RPC call status: 0x%x\n", status
);
929 if (status
!= RPC_S_OK
)
930 hr
= HRESULT_FROM_WIN32(status
);
932 TRACE("hrFault = 0x%08x\n", hrFault
);
934 /* FIXME: this condition should be
935 * "hr == S_OK && (!hrFault || msg->BufferLength > FIELD_OFFSET(ORPCTHAT, extensions) + 4)"
936 * but we don't currently reset the message length for PostMessage
937 * dispatched calls */
938 if (hr
== S_OK
&& hrFault
== S_OK
)
941 char *original_buffer
= msg
->Buffer
;
943 /* handle ORPCTHAT and client extensions */
945 hr2
= unmarshal_ORPCTHAT(msg
, &orpcthat
, &orpc_ext_array
, &first_wire_orpc_extent
);
949 message_state
->prefix_data_len
= (char *)msg
->Buffer
- original_buffer
;
950 msg
->BufferLength
-= message_state
->prefix_data_len
;
953 message_state
->prefix_data_len
= 0;
957 ChannelHooks_ClientNotify(&message_state
->channel_hook_info
,
958 msg
->DataRepresentation
,
959 first_wire_orpc_extent
,
960 orpcthat
.extensions
&& first_wire_orpc_extent
? orpcthat
.extensions
->size
: 0,
964 /* save away the message state again */
965 msg
->Handle
= message_state
;
967 if (pstatus
) *pstatus
= status
;
972 TRACE("-- 0x%08x\n", hr
);
977 static HRESULT WINAPI
ServerRpcChannelBuffer_FreeBuffer(LPRPCCHANNELBUFFER iface
, RPCOLEMESSAGE
* olemsg
)
979 RPC_MESSAGE
*msg
= (RPC_MESSAGE
*)olemsg
;
981 struct message_state
*message_state
;
983 TRACE("(%p)\n", msg
);
985 message_state
= msg
->Handle
;
986 /* restore the binding handle and the real start of data */
987 msg
->Handle
= message_state
->binding_handle
;
988 msg
->Buffer
= (char *)msg
->Buffer
- message_state
->prefix_data_len
;
989 msg
->BufferLength
+= message_state
->prefix_data_len
;
990 message_state
->prefix_data_len
= 0;
992 if (message_state
->bypass_rpcrt
)
994 HeapFree(GetProcessHeap(), 0, msg
->Buffer
);
998 status
= I_RpcFreeBuffer(msg
);
1000 msg
->Handle
= message_state
;
1002 TRACE("-- %d\n", status
);
1004 return HRESULT_FROM_WIN32(status
);
1007 static HRESULT WINAPI
ClientRpcChannelBuffer_FreeBuffer(LPRPCCHANNELBUFFER iface
, RPCOLEMESSAGE
* olemsg
)
1009 RPC_MESSAGE
*msg
= (RPC_MESSAGE
*)olemsg
;
1011 struct message_state
*message_state
;
1013 TRACE("(%p)\n", msg
);
1015 message_state
= msg
->Handle
;
1016 /* restore the binding handle and the real start of data */
1017 msg
->Handle
= message_state
->binding_handle
;
1018 msg
->Buffer
= (char *)msg
->Buffer
- message_state
->prefix_data_len
;
1019 msg
->BufferLength
+= message_state
->prefix_data_len
;
1021 if (message_state
->params
.bypass_rpcrt
)
1023 HeapFree(GetProcessHeap(), 0, msg
->Buffer
);
1027 status
= I_RpcFreeBuffer(msg
);
1029 HeapFree(GetProcessHeap(), 0, msg
->RpcInterfaceInformation
);
1030 msg
->RpcInterfaceInformation
= NULL
;
1032 if (message_state
->params
.stub
)
1033 IRpcStubBuffer_Release(message_state
->params
.stub
);
1034 if (message_state
->params
.chan
)
1035 IRpcChannelBuffer_Release(message_state
->params
.chan
);
1036 HeapFree(GetProcessHeap(), 0, message_state
);
1038 TRACE("-- %d\n", status
);
1040 return HRESULT_FROM_WIN32(status
);
1043 static HRESULT WINAPI
ClientRpcChannelBuffer_GetDestCtx(LPRPCCHANNELBUFFER iface
, DWORD
* pdwDestContext
, void** ppvDestContext
)
1045 ClientRpcChannelBuffer
*This
= (ClientRpcChannelBuffer
*)iface
;
1047 TRACE("(%p,%p)\n", pdwDestContext
, ppvDestContext
);
1049 *pdwDestContext
= This
->super
.dest_context
;
1050 *ppvDestContext
= This
->super
.dest_context_data
;
1055 static HRESULT WINAPI
ServerRpcChannelBuffer_GetDestCtx(LPRPCCHANNELBUFFER iface
, DWORD
* dest_context
, void** dest_context_data
)
1057 RpcChannelBuffer
*This
= (RpcChannelBuffer
*)iface
;
1059 TRACE("(%p,%p)\n", dest_context
, dest_context_data
);
1061 *dest_context
= This
->dest_context
;
1062 *dest_context_data
= This
->dest_context_data
;
1066 static HRESULT WINAPI
RpcChannelBuffer_IsConnected(LPRPCCHANNELBUFFER iface
)
1069 /* native does nothing too */
1073 static const IRpcChannelBufferVtbl ClientRpcChannelBufferVtbl
=
1075 RpcChannelBuffer_QueryInterface
,
1076 RpcChannelBuffer_AddRef
,
1077 ClientRpcChannelBuffer_Release
,
1078 ClientRpcChannelBuffer_GetBuffer
,
1079 ClientRpcChannelBuffer_SendReceive
,
1080 ClientRpcChannelBuffer_FreeBuffer
,
1081 ClientRpcChannelBuffer_GetDestCtx
,
1082 RpcChannelBuffer_IsConnected
1085 static const IRpcChannelBufferVtbl ServerRpcChannelBufferVtbl
=
1087 RpcChannelBuffer_QueryInterface
,
1088 RpcChannelBuffer_AddRef
,
1089 ServerRpcChannelBuffer_Release
,
1090 ServerRpcChannelBuffer_GetBuffer
,
1091 ServerRpcChannelBuffer_SendReceive
,
1092 ServerRpcChannelBuffer_FreeBuffer
,
1093 ServerRpcChannelBuffer_GetDestCtx
,
1094 RpcChannelBuffer_IsConnected
1097 /* returns a channel buffer for proxies */
1098 HRESULT
RPC_CreateClientChannel(const OXID
*oxid
, const IPID
*ipid
,
1099 const OXID_INFO
*oxid_info
,
1100 DWORD dest_context
, void *dest_context_data
,
1101 IRpcChannelBuffer
**chan
)
1103 ClientRpcChannelBuffer
*This
;
1104 WCHAR endpoint
[200];
1105 RPC_BINDING_HANDLE bind
;
1107 LPWSTR string_binding
;
1109 /* FIXME: get the endpoint from oxid_info->psa instead */
1110 get_rpc_endpoint(endpoint
, oxid
);
1112 TRACE("proxy pipe: connecting to endpoint: %s\n", debugstr_w(endpoint
));
1114 status
= RpcStringBindingComposeW(
1122 if (status
== RPC_S_OK
)
1124 status
= RpcBindingFromStringBindingW(string_binding
, &bind
);
1126 if (status
== RPC_S_OK
)
1128 IPID ipid2
= *ipid
; /* why can't RpcBindingSetObject take a const? */
1129 status
= RpcBindingSetObject(bind
, &ipid2
);
1130 if (status
!= RPC_S_OK
)
1131 RpcBindingFree(&bind
);
1134 RpcStringFreeW(&string_binding
);
1137 if (status
!= RPC_S_OK
)
1139 ERR("Couldn't get binding for endpoint %s, status = %d\n", debugstr_w(endpoint
), status
);
1140 return HRESULT_FROM_WIN32(status
);
1143 This
= HeapAlloc(GetProcessHeap(), 0, sizeof(*This
));
1146 RpcBindingFree(&bind
);
1147 return E_OUTOFMEMORY
;
1150 This
->super
.IRpcChannelBuffer_iface
.lpVtbl
= &ClientRpcChannelBufferVtbl
;
1151 This
->super
.refs
= 1;
1152 This
->super
.dest_context
= dest_context
;
1153 This
->super
.dest_context_data
= dest_context_data
;
1155 apartment_getoxid(COM_CurrentApt(), &This
->oxid
);
1156 This
->server_pid
= oxid_info
->dwPid
;
1159 *chan
= &This
->super
.IRpcChannelBuffer_iface
;
1164 HRESULT
RPC_CreateServerChannel(DWORD dest_context
, void *dest_context_data
, IRpcChannelBuffer
**chan
)
1166 RpcChannelBuffer
*This
= HeapAlloc(GetProcessHeap(), 0, sizeof(*This
));
1168 return E_OUTOFMEMORY
;
1170 This
->IRpcChannelBuffer_iface
.lpVtbl
= &ServerRpcChannelBufferVtbl
;
1172 This
->dest_context
= dest_context
;
1173 This
->dest_context_data
= dest_context_data
;
1175 *chan
= &This
->IRpcChannelBuffer_iface
;
1180 /* unmarshals ORPC_EXTENT_ARRAY according to NDR rules, but doesn't allocate
1182 static HRESULT
unmarshal_ORPC_EXTENT_ARRAY(RPC_MESSAGE
*msg
, const char *end
,
1183 ORPC_EXTENT_ARRAY
*extensions
,
1184 WIRE_ORPC_EXTENT
**first_wire_orpc_extent
)
1189 memcpy(extensions
, msg
->Buffer
, FIELD_OFFSET(WIRE_ORPC_EXTENT_ARRAY
, extent
));
1190 msg
->Buffer
= (char *)msg
->Buffer
+ FIELD_OFFSET(WIRE_ORPC_EXTENT_ARRAY
, extent
);
1192 if ((const char *)msg
->Buffer
+ 2 * sizeof(DWORD
) > end
)
1193 return RPC_E_INVALID_HEADER
;
1195 pointer_id
= *(DWORD
*)msg
->Buffer
;
1196 msg
->Buffer
= (char *)msg
->Buffer
+ sizeof(DWORD
);
1197 extensions
->extent
= NULL
;
1201 WIRE_ORPC_EXTENT
*wire_orpc_extent
;
1204 if (*(DWORD
*)msg
->Buffer
!= ((extensions
->size
+1)&~1))
1205 return RPC_S_INVALID_BOUND
;
1207 msg
->Buffer
= (char *)msg
->Buffer
+ sizeof(DWORD
);
1209 /* arbitrary limit for security (don't know what native does) */
1210 if (extensions
->size
> 256)
1212 ERR("too many extensions: %d\n", extensions
->size
);
1213 return RPC_S_INVALID_BOUND
;
1216 *first_wire_orpc_extent
= wire_orpc_extent
= msg
->Buffer
;
1217 for (i
= 0; i
< ((extensions
->size
+1)&~1); i
++)
1219 if ((const char *)&wire_orpc_extent
->data
[0] > end
)
1220 return RPC_S_INVALID_BOUND
;
1221 if (wire_orpc_extent
->conformance
!= ((wire_orpc_extent
->size
+7)&~7))
1222 return RPC_S_INVALID_BOUND
;
1223 if ((const char *)&wire_orpc_extent
->data
[wire_orpc_extent
->conformance
] > end
)
1224 return RPC_S_INVALID_BOUND
;
1225 TRACE("size %u, guid %s\n", wire_orpc_extent
->size
, debugstr_guid(&wire_orpc_extent
->id
));
1226 wire_orpc_extent
= (WIRE_ORPC_EXTENT
*)&wire_orpc_extent
->data
[wire_orpc_extent
->conformance
];
1228 msg
->Buffer
= wire_orpc_extent
;
1234 /* unmarshals ORPCTHIS according to NDR rules, but doesn't allocate any memory */
1235 static HRESULT
unmarshal_ORPCTHIS(RPC_MESSAGE
*msg
, ORPCTHIS
*orpcthis
,
1236 ORPC_EXTENT_ARRAY
*orpc_ext_array
, WIRE_ORPC_EXTENT
**first_wire_orpc_extent
)
1238 const char *end
= (char *)msg
->Buffer
+ msg
->BufferLength
;
1240 *first_wire_orpc_extent
= NULL
;
1242 if (msg
->BufferLength
< FIELD_OFFSET(WIRE_ORPCTHIS
, extensions
) + sizeof(DWORD
))
1244 ERR("invalid buffer length\n");
1245 return RPC_E_INVALID_HEADER
;
1248 memcpy(orpcthis
, msg
->Buffer
, FIELD_OFFSET(WIRE_ORPCTHIS
, extensions
));
1249 msg
->Buffer
= (char *)msg
->Buffer
+ FIELD_OFFSET(WIRE_ORPCTHIS
, extensions
);
1251 if ((const char *)msg
->Buffer
+ sizeof(DWORD
) > end
)
1252 return RPC_E_INVALID_HEADER
;
1254 if (*(DWORD
*)msg
->Buffer
)
1255 orpcthis
->extensions
= orpc_ext_array
;
1257 orpcthis
->extensions
= NULL
;
1259 msg
->Buffer
= (char *)msg
->Buffer
+ sizeof(DWORD
);
1261 if (orpcthis
->extensions
)
1263 HRESULT hr
= unmarshal_ORPC_EXTENT_ARRAY(msg
, end
, orpc_ext_array
,
1264 first_wire_orpc_extent
);
1269 if ((orpcthis
->version
.MajorVersion
!= COM_MAJOR_VERSION
) ||
1270 (orpcthis
->version
.MinorVersion
> COM_MINOR_VERSION
))
1272 ERR("COM version {%d, %d} not supported\n",
1273 orpcthis
->version
.MajorVersion
, orpcthis
->version
.MinorVersion
);
1274 return RPC_E_VERSION_MISMATCH
;
1277 if (orpcthis
->flags
& ~(ORPCF_LOCAL
|ORPCF_RESERVED1
|ORPCF_RESERVED2
|ORPCF_RESERVED3
|ORPCF_RESERVED4
))
1279 ERR("invalid flags 0x%x\n", orpcthis
->flags
& ~(ORPCF_LOCAL
|ORPCF_RESERVED1
|ORPCF_RESERVED2
|ORPCF_RESERVED3
|ORPCF_RESERVED4
));
1280 return RPC_E_INVALID_HEADER
;
1286 static HRESULT
unmarshal_ORPCTHAT(RPC_MESSAGE
*msg
, ORPCTHAT
*orpcthat
,
1287 ORPC_EXTENT_ARRAY
*orpc_ext_array
, WIRE_ORPC_EXTENT
**first_wire_orpc_extent
)
1289 const char *end
= (char *)msg
->Buffer
+ msg
->BufferLength
;
1291 *first_wire_orpc_extent
= NULL
;
1293 if (msg
->BufferLength
< FIELD_OFFSET(WIRE_ORPCTHAT
, extensions
) + sizeof(DWORD
))
1295 ERR("invalid buffer length\n");
1296 return RPC_E_INVALID_HEADER
;
1299 memcpy(orpcthat
, msg
->Buffer
, FIELD_OFFSET(WIRE_ORPCTHAT
, extensions
));
1300 msg
->Buffer
= (char *)msg
->Buffer
+ FIELD_OFFSET(WIRE_ORPCTHAT
, extensions
);
1302 if ((const char *)msg
->Buffer
+ sizeof(DWORD
) > end
)
1303 return RPC_E_INVALID_HEADER
;
1305 if (*(DWORD
*)msg
->Buffer
)
1306 orpcthat
->extensions
= orpc_ext_array
;
1308 orpcthat
->extensions
= NULL
;
1310 msg
->Buffer
= (char *)msg
->Buffer
+ sizeof(DWORD
);
1312 if (orpcthat
->extensions
)
1314 HRESULT hr
= unmarshal_ORPC_EXTENT_ARRAY(msg
, end
, orpc_ext_array
,
1315 first_wire_orpc_extent
);
1320 if (orpcthat
->flags
& ~(ORPCF_LOCAL
|ORPCF_RESERVED1
|ORPCF_RESERVED2
|ORPCF_RESERVED3
|ORPCF_RESERVED4
))
1322 ERR("invalid flags 0x%x\n", orpcthat
->flags
& ~(ORPCF_LOCAL
|ORPCF_RESERVED1
|ORPCF_RESERVED2
|ORPCF_RESERVED3
|ORPCF_RESERVED4
));
1323 return RPC_E_INVALID_HEADER
;
1329 void RPC_ExecuteCall(struct dispatch_params
*params
)
1331 struct message_state
*message_state
= NULL
;
1332 RPC_MESSAGE
*msg
= (RPC_MESSAGE
*)params
->msg
;
1333 char *original_buffer
= msg
->Buffer
;
1335 ORPC_EXTENT_ARRAY orpc_ext_array
;
1336 WIRE_ORPC_EXTENT
*first_wire_orpc_extent
;
1337 GUID old_causality_id
;
1339 /* handle ORPCTHIS and server extensions */
1341 params
->hr
= unmarshal_ORPCTHIS(msg
, &orpcthis
, &orpc_ext_array
, &first_wire_orpc_extent
);
1342 if (params
->hr
!= S_OK
)
1344 msg
->Buffer
= original_buffer
;
1348 message_state
= HeapAlloc(GetProcessHeap(), 0, sizeof(*message_state
));
1351 params
->hr
= E_OUTOFMEMORY
;
1352 msg
->Buffer
= original_buffer
;
1356 message_state
->prefix_data_len
= (char *)msg
->Buffer
- original_buffer
;
1357 message_state
->binding_handle
= msg
->Handle
;
1358 message_state
->bypass_rpcrt
= params
->bypass_rpcrt
;
1360 message_state
->channel_hook_info
.iid
= params
->iid
;
1361 message_state
->channel_hook_info
.cbSize
= sizeof(message_state
->channel_hook_info
);
1362 message_state
->channel_hook_info
.uCausality
= orpcthis
.cid
;
1363 message_state
->channel_hook_info
.dwServerPid
= GetCurrentProcessId();
1364 message_state
->channel_hook_info
.iMethod
= msg
->ProcNum
;
1365 message_state
->channel_hook_info
.pObject
= params
->iface
;
1367 if (orpcthis
.extensions
&& first_wire_orpc_extent
&&
1368 orpcthis
.extensions
->size
)
1369 ChannelHooks_ServerNotify(&message_state
->channel_hook_info
, msg
->DataRepresentation
, first_wire_orpc_extent
, orpcthis
.extensions
->size
);
1371 msg
->Handle
= message_state
;
1372 msg
->BufferLength
-= message_state
->prefix_data_len
;
1374 /* call message filter */
1376 if (COM_CurrentApt()->filter
)
1379 INTERFACEINFO interface_info
;
1382 interface_info
.pUnk
= params
->iface
;
1383 interface_info
.iid
= params
->iid
;
1384 interface_info
.wMethod
= msg
->ProcNum
;
1386 if (IsEqualGUID(&orpcthis
.cid
, &COM_CurrentInfo()->causality_id
))
1387 calltype
= CALLTYPE_NESTED
;
1388 else if (COM_CurrentInfo()->pending_call_count_server
== 0)
1389 calltype
= CALLTYPE_TOPLEVEL
;
1391 calltype
= CALLTYPE_TOPLEVEL_CALLPENDING
;
1393 handlecall
= IMessageFilter_HandleInComingCall(COM_CurrentApt()->filter
,
1395 UlongToHandle(GetCurrentProcessId()),
1398 TRACE("IMessageFilter_HandleInComingCall returned %d\n", handlecall
);
1401 case SERVERCALL_REJECTED
:
1402 params
->hr
= RPC_E_CALL_REJECTED
;
1403 goto exit_reset_state
;
1404 case SERVERCALL_RETRYLATER
:
1405 #if 0 /* FIXME: handle retries on the client side before enabling this code */
1406 params
->hr
= RPC_E_RETRY
;
1407 goto exit_reset_state
;
1409 FIXME("retry call later not implemented\n");
1412 case SERVERCALL_ISHANDLED
:
1418 /* invoke the method */
1420 /* save the old causality ID - note: any calls executed while processing
1421 * messages received during the SendReceive will appear to originate from
1422 * this call - this should be checked with what Windows does */
1423 old_causality_id
= COM_CurrentInfo()->causality_id
;
1424 COM_CurrentInfo()->causality_id
= orpcthis
.cid
;
1425 COM_CurrentInfo()->pending_call_count_server
++;
1426 params
->hr
= IRpcStubBuffer_Invoke(params
->stub
, params
->msg
, params
->chan
);
1427 COM_CurrentInfo()->pending_call_count_server
--;
1428 COM_CurrentInfo()->causality_id
= old_causality_id
;
1430 /* the invoke allocated a new buffer, so free the old one */
1431 if (message_state
->bypass_rpcrt
&& original_buffer
!= msg
->Buffer
)
1432 HeapFree(GetProcessHeap(), 0, original_buffer
);
1435 message_state
= msg
->Handle
;
1436 msg
->Handle
= message_state
->binding_handle
;
1437 msg
->Buffer
= (char *)msg
->Buffer
- message_state
->prefix_data_len
;
1438 msg
->BufferLength
+= message_state
->prefix_data_len
;
1441 HeapFree(GetProcessHeap(), 0, message_state
);
1442 if (params
->handle
) SetEvent(params
->handle
);
1445 static void __RPC_STUB
dispatch_rpc(RPC_MESSAGE
*msg
)
1447 struct dispatch_params
*params
;
1452 RpcBindingInqObject(msg
->Handle
, &ipid
);
1454 TRACE("ipid = %s, iMethod = %d\n", debugstr_guid(&ipid
), msg
->ProcNum
);
1456 params
= HeapAlloc(GetProcessHeap(), 0, sizeof(*params
));
1459 RpcRaiseException(E_OUTOFMEMORY
);
1463 hr
= ipid_get_dispatch_params(&ipid
, &apt
, ¶ms
->stub
, ¶ms
->chan
,
1464 ¶ms
->iid
, ¶ms
->iface
);
1467 ERR("no apartment found for ipid %s\n", debugstr_guid(&ipid
));
1468 HeapFree(GetProcessHeap(), 0, params
);
1469 RpcRaiseException(hr
);
1473 params
->msg
= (RPCOLEMESSAGE
*)msg
;
1474 params
->status
= RPC_S_OK
;
1476 params
->handle
= NULL
;
1477 params
->bypass_rpcrt
= FALSE
;
1479 /* Note: this is the important difference between STAs and MTAs - we
1480 * always execute RPCs to STAs in the thread that originally created the
1481 * apartment (i.e. the one that pumps messages to the window) */
1482 if (!apt
->multi_threaded
)
1484 params
->handle
= CreateEventW(NULL
, FALSE
, FALSE
, NULL
);
1486 TRACE("Calling apartment thread 0x%08x...\n", apt
->tid
);
1488 if (PostMessageW(apartment_getwindow(apt
), DM_EXECUTERPC
, 0, (LPARAM
)params
))
1489 WaitForSingleObject(params
->handle
, INFINITE
);
1492 ERR("PostMessage failed with error %u\n", GetLastError());
1493 IRpcChannelBuffer_Release(params
->chan
);
1494 IRpcStubBuffer_Release(params
->stub
);
1496 CloseHandle(params
->handle
);
1500 BOOL joined
= FALSE
;
1501 if (!COM_CurrentInfo()->apt
)
1503 apartment_joinmta();
1506 RPC_ExecuteCall(params
);
1509 apartment_release(COM_CurrentInfo()->apt
);
1510 COM_CurrentInfo()->apt
= NULL
;
1516 IRpcChannelBuffer_Release(params
->chan
);
1518 IRpcStubBuffer_Release(params
->stub
);
1519 HeapFree(GetProcessHeap(), 0, params
);
1521 apartment_release(apt
);
1523 /* if IRpcStubBuffer_Invoke fails, we should raise an exception to tell
1524 * the RPC runtime that the call failed */
1525 if (hr
!= S_OK
) RpcRaiseException(hr
);
1528 /* stub registration */
1529 HRESULT
RPC_RegisterInterface(REFIID riid
)
1531 struct registered_if
*rif
;
1535 TRACE("(%s)\n", debugstr_guid(riid
));
1537 EnterCriticalSection(&csRegIf
);
1538 LIST_FOR_EACH_ENTRY(rif
, ®istered_interfaces
, struct registered_if
, entry
)
1540 if (IsEqualGUID(&rif
->If
.InterfaceId
.SyntaxGUID
, riid
))
1549 TRACE("Creating new interface\n");
1551 rif
= HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY
, sizeof(*rif
));
1557 rif
->If
.Length
= sizeof(RPC_SERVER_INTERFACE
);
1558 /* RPC interface ID = COM interface ID */
1559 rif
->If
.InterfaceId
.SyntaxGUID
= *riid
;
1560 rif
->If
.DispatchTable
= &rpc_dispatch
;
1561 /* all other fields are 0, including the version asCOM objects
1562 * always have a version of 0.0 */
1563 status
= RpcServerRegisterIfEx(
1564 (RPC_IF_HANDLE
)&rif
->If
,
1566 RPC_IF_OLE
| RPC_IF_AUTOLISTEN
,
1567 RPC_C_LISTEN_MAX_CALLS_DEFAULT
,
1569 if (status
== RPC_S_OK
)
1570 list_add_tail(®istered_interfaces
, &rif
->entry
);
1573 ERR("RpcServerRegisterIfEx failed with error %d\n", status
);
1574 HeapFree(GetProcessHeap(), 0, rif
);
1575 hr
= HRESULT_FROM_WIN32(status
);
1581 LeaveCriticalSection(&csRegIf
);
1585 /* stub unregistration */
1586 void RPC_UnregisterInterface(REFIID riid
)
1588 struct registered_if
*rif
;
1589 EnterCriticalSection(&csRegIf
);
1590 LIST_FOR_EACH_ENTRY(rif
, ®istered_interfaces
, struct registered_if
, entry
)
1592 if (IsEqualGUID(&rif
->If
.InterfaceId
.SyntaxGUID
, riid
))
1596 RpcServerUnregisterIf((RPC_IF_HANDLE
)&rif
->If
, NULL
, TRUE
);
1597 list_remove(&rif
->entry
);
1598 HeapFree(GetProcessHeap(), 0, rif
);
1603 LeaveCriticalSection(&csRegIf
);
1606 /* get the info for an OXID, including the IPID for the rem unknown interface
1607 * and the string binding */
1608 HRESULT
RPC_ResolveOxid(OXID oxid
, OXID_INFO
*oxid_info
)
1610 TRACE("%s\n", wine_dbgstr_longlong(oxid
));
1612 oxid_info
->dwTid
= 0;
1613 oxid_info
->dwPid
= 0;
1614 oxid_info
->dwAuthnHint
= RPC_C_AUTHN_LEVEL_NONE
;
1615 /* FIXME: this is a hack around not having an OXID resolver yet -
1616 * this function should contact the machine's OXID resolver and then it
1617 * should give us the IPID of the IRemUnknown interface */
1618 oxid_info
->ipidRemUnknown
.Data1
= 0xffffffff;
1619 oxid_info
->ipidRemUnknown
.Data2
= 0xffff;
1620 oxid_info
->ipidRemUnknown
.Data3
= 0xffff;
1621 memcpy(oxid_info
->ipidRemUnknown
.Data4
, &oxid
, sizeof(OXID
));
1622 oxid_info
->psa
= NULL
/* FIXME */;
1627 /* make the apartment reachable by other threads and processes and create the
1628 * IRemUnknown object */
1629 void RPC_StartRemoting(struct apartment
*apt
)
1631 if (!InterlockedExchange(&apt
->remoting_started
, TRUE
))
1633 WCHAR endpoint
[200];
1636 get_rpc_endpoint(endpoint
, &apt
->oxid
);
1638 status
= RpcServerUseProtseqEpW(
1640 RPC_C_PROTSEQ_MAX_REQS_DEFAULT
,
1643 if (status
!= RPC_S_OK
)
1644 ERR("Couldn't register endpoint %s\n", debugstr_w(endpoint
));
1646 /* FIXME: move remote unknown exporting into this function */
1648 start_apartment_remote_unknown();
1652 static HRESULT
create_server(REFCLSID rclsid
, HANDLE
*process
)
1654 static const WCHAR wszLocalServer32
[] = { 'L','o','c','a','l','S','e','r','v','e','r','3','2',0 };
1655 static const WCHAR embedding
[] = { ' ', '-','E','m','b','e','d','d','i','n','g',0 };
1658 WCHAR command
[MAX_PATH
+sizeof(embedding
)/sizeof(WCHAR
)];
1659 DWORD size
= (MAX_PATH
+1) * sizeof(WCHAR
);
1661 PROCESS_INFORMATION pinfo
;
1664 hres
= COM_OpenKeyForCLSID(rclsid
, wszLocalServer32
, KEY_READ
, &key
);
1666 ERR("class %s not registered\n", debugstr_guid(rclsid
));
1670 ret
= RegQueryValueExW(key
, NULL
, NULL
, NULL
, (LPBYTE
)command
, &size
);
1673 WARN("No default value for LocalServer32 key\n");
1674 return REGDB_E_CLASSNOTREG
; /* FIXME: check retval */
1677 memset(&sinfo
,0,sizeof(sinfo
));
1678 sinfo
.cb
= sizeof(sinfo
);
1680 /* EXE servers are started with the -Embedding switch. */
1682 strcatW(command
, embedding
);
1684 TRACE("activating local server %s for %s\n", debugstr_w(command
), debugstr_guid(rclsid
));
1686 /* FIXME: Win2003 supports a ServerExecutable value that is passed into
1688 if (!CreateProcessW(NULL
, command
, NULL
, NULL
, FALSE
, DETACHED_PROCESS
, NULL
, NULL
, &sinfo
, &pinfo
)) {
1689 WARN("failed to run local server %s\n", debugstr_w(command
));
1690 return HRESULT_FROM_WIN32(GetLastError());
1692 *process
= pinfo
.hProcess
;
1693 CloseHandle(pinfo
.hThread
);
1699 * start_local_service() - start a service given its name and parameters
1701 static DWORD
start_local_service(LPCWSTR name
, DWORD num
, LPCWSTR
*params
)
1703 SC_HANDLE handle
, hsvc
;
1704 DWORD r
= ERROR_FUNCTION_FAILED
;
1706 TRACE("Starting service %s %d params\n", debugstr_w(name
), num
);
1708 handle
= OpenSCManagerW(NULL
, NULL
, SC_MANAGER_CONNECT
);
1711 hsvc
= OpenServiceW(handle
, name
, SERVICE_START
);
1714 if(StartServiceW(hsvc
, num
, params
))
1718 if (r
== ERROR_SERVICE_ALREADY_RUNNING
)
1720 CloseServiceHandle(hsvc
);
1724 CloseServiceHandle(handle
);
1726 TRACE("StartService returned error %u (%s)\n", r
, (r
== ERROR_SUCCESS
) ? "ok":"failed");
1732 * create_local_service() - start a COM server in a service
1734 * To start a Local Service, we read the AppID value under
1735 * the class's CLSID key, then open the HKCR\\AppId key specified
1736 * there and check for a LocalService value.
1738 * Note: Local Services are not supported under Windows 9x
1740 static HRESULT
create_local_service(REFCLSID rclsid
)
1743 WCHAR buf
[CHARS_IN_GUID
];
1744 static const WCHAR szLocalService
[] = { 'L','o','c','a','l','S','e','r','v','i','c','e',0 };
1745 static const WCHAR szServiceParams
[] = {'S','e','r','v','i','c','e','P','a','r','a','m','s',0};
1750 TRACE("Attempting to start Local service for %s\n", debugstr_guid(rclsid
));
1752 hres
= COM_OpenKeyForAppIdFromCLSID(rclsid
, KEY_READ
, &hkey
);
1756 /* read the LocalService and ServiceParameters values from the AppID key */
1758 r
= RegQueryValueExW(hkey
, szLocalService
, NULL
, &type
, (LPBYTE
)buf
, &sz
);
1759 if (r
==ERROR_SUCCESS
&& type
==REG_SZ
)
1762 LPWSTR args
[1] = { NULL
};
1765 * FIXME: I'm not really sure how to deal with the service parameters.
1766 * I suspect that the string returned from RegQueryValueExW
1767 * should be split into a number of arguments by spaces.
1768 * It would make more sense if ServiceParams contained a
1769 * REG_MULTI_SZ here, but it's a REG_SZ for the services
1770 * that I'm interested in for the moment.
1772 r
= RegQueryValueExW(hkey
, szServiceParams
, NULL
, &type
, NULL
, &sz
);
1773 if (r
== ERROR_SUCCESS
&& type
== REG_SZ
&& sz
)
1775 args
[0] = HeapAlloc(GetProcessHeap(),HEAP_ZERO_MEMORY
,sz
);
1777 RegQueryValueExW(hkey
, szServiceParams
, NULL
, &type
, (LPBYTE
)args
[0], &sz
);
1779 r
= start_local_service(buf
, num_args
, (LPCWSTR
*)args
);
1780 if (r
!= ERROR_SUCCESS
)
1781 hres
= REGDB_E_CLASSNOTREG
; /* FIXME: check retval */
1782 HeapFree(GetProcessHeap(),0,args
[0]);
1786 WARN("No LocalService value\n");
1787 hres
= REGDB_E_CLASSNOTREG
; /* FIXME: check retval */
1795 static void get_localserver_pipe_name(WCHAR
*pipefn
, REFCLSID rclsid
)
1797 static const WCHAR wszPipeRef
[] = {'\\','\\','.','\\','p','i','p','e','\\',0};
1798 strcpyW(pipefn
, wszPipeRef
);
1799 StringFromGUID2(rclsid
, pipefn
+ sizeof(wszPipeRef
)/sizeof(wszPipeRef
[0]) - 1, CHARS_IN_GUID
);
1802 /* FIXME: should call to rpcss instead */
1803 HRESULT
RPC_GetLocalClassObject(REFCLSID rclsid
, REFIID iid
, LPVOID
*ppv
)
1808 DWORD res
, bufferlen
;
1809 char marshalbuffer
[200];
1811 LARGE_INTEGER seekto
;
1812 ULARGE_INTEGER newpos
;
1814 IServiceProvider
*local_server
;
1816 static const int MAXTRIES
= 30; /* 30 seconds */
1818 TRACE("rclsid=%s, iid=%s\n", debugstr_guid(rclsid
), debugstr_guid(iid
));
1820 get_localserver_pipe_name(pipefn
, rclsid
);
1822 while (tries
++ < MAXTRIES
) {
1823 TRACE("waiting for %s\n", debugstr_w(pipefn
));
1825 WaitNamedPipeW( pipefn
, NMPWAIT_WAIT_FOREVER
);
1826 hPipe
= CreateFileW(pipefn
, GENERIC_READ
| GENERIC_WRITE
, 0, NULL
, OPEN_EXISTING
, 0, 0);
1827 if (hPipe
== INVALID_HANDLE_VALUE
) {
1832 if ( (hres
= create_local_service(rclsid
)) &&
1833 (hres
= create_server(rclsid
, &process
)) )
1836 WARN("Connecting to %s, no response yet, retrying: le is %u\n", debugstr_w(pipefn
), GetLastError());
1838 /* wait for one second, even if messages arrive */
1839 start_ticks
= GetTickCount();
1841 if (SUCCEEDED(CoWaitForMultipleHandles(0, 1000, (process
!= 0),
1842 &process
, &index
)) && process
&& !index
)
1844 WARN( "server for %s failed to start\n", debugstr_guid(rclsid
) );
1845 CloseHandle( hPipe
);
1846 CloseHandle( process
);
1847 return E_NOINTERFACE
;
1849 } while (GetTickCount() - start_ticks
< 1000);
1850 if (process
) CloseHandle( process
);
1854 if (!ReadFile(hPipe
,marshalbuffer
,sizeof(marshalbuffer
),&bufferlen
,NULL
)) {
1855 FIXME("Failed to read marshal id from classfactory of %s.\n",debugstr_guid(rclsid
));
1859 TRACE("read marshal id from pipe\n");
1864 if (tries
>= MAXTRIES
)
1865 return E_NOINTERFACE
;
1867 hres
= CreateStreamOnHGlobal(0,TRUE
,&pStm
);
1868 if (hres
!= S_OK
) return hres
;
1869 hres
= IStream_Write(pStm
,marshalbuffer
,bufferlen
,&res
);
1870 if (hres
!= S_OK
) goto out
;
1871 seekto
.u
.LowPart
= 0;seekto
.u
.HighPart
= 0;
1872 hres
= IStream_Seek(pStm
,seekto
,STREAM_SEEK_SET
,&newpos
);
1874 TRACE("unmarshalling local server\n");
1875 hres
= CoUnmarshalInterface(pStm
, &IID_IServiceProvider
, (void**)&local_server
);
1877 hres
= IServiceProvider_QueryService(local_server
, rclsid
, iid
, ppv
);
1878 IServiceProvider_Release(local_server
);
1880 IStream_Release(pStm
);
1885 struct local_server_params
1895 /* FIXME: should call to rpcss instead */
1896 static DWORD WINAPI
local_server_thread(LPVOID param
)
1898 struct local_server_params
* lsp
= param
;
1901 IStream
*pStm
= lsp
->stream
;
1903 unsigned char *buffer
;
1905 LARGE_INTEGER seekto
;
1906 ULARGE_INTEGER newpos
;
1908 BOOL multi_use
= lsp
->multi_use
;
1910 HANDLE pipe_event
, hPipe
= lsp
->pipe
, new_pipe
;
1913 TRACE("Starting threader for %s.\n",debugstr_guid(&lsp
->clsid
));
1915 memset(&ovl
, 0, sizeof(ovl
));
1916 get_localserver_pipe_name(pipefn
, &lsp
->clsid
);
1917 ovl
.hEvent
= pipe_event
= CreateEventW(NULL
, FALSE
, FALSE
, NULL
);
1920 if (!ConnectNamedPipe(hPipe
, &ovl
))
1922 DWORD error
= GetLastError();
1923 if (error
== ERROR_IO_PENDING
)
1925 HANDLE handles
[2] = { pipe_event
, lsp
->stop_event
};
1927 ret
= WaitForMultipleObjects(2, handles
, FALSE
, INFINITE
);
1928 if (ret
!= WAIT_OBJECT_0
)
1931 /* client already connected isn't an error */
1932 else if (error
!= ERROR_PIPE_CONNECTED
)
1934 ERR("ConnectNamedPipe failed with error %d\n", GetLastError());
1939 TRACE("marshalling LocalServer to client\n");
1941 hres
= IStream_Stat(pStm
,&ststg
,STATFLAG_NONAME
);
1945 seekto
.u
.LowPart
= 0;
1946 seekto
.u
.HighPart
= 0;
1947 hres
= IStream_Seek(pStm
,seekto
,STREAM_SEEK_SET
,&newpos
);
1949 FIXME("IStream_Seek failed, %x\n",hres
);
1953 buflen
= ststg
.cbSize
.u
.LowPart
;
1954 buffer
= HeapAlloc(GetProcessHeap(),0,buflen
);
1956 hres
= IStream_Read(pStm
,buffer
,buflen
,&res
);
1958 FIXME("Stream Read failed, %x\n",hres
);
1959 HeapFree(GetProcessHeap(),0,buffer
);
1963 WriteFile(hPipe
,buffer
,buflen
,&res
,&ovl
);
1964 GetOverlappedResult(hPipe
, &ovl
, &bytes
, TRUE
);
1965 HeapFree(GetProcessHeap(),0,buffer
);
1967 FlushFileBuffers(hPipe
);
1968 DisconnectNamedPipe(hPipe
);
1969 TRACE("done marshalling LocalServer\n");
1973 TRACE("single use object, shutting down pipe %s\n", debugstr_w(pipefn
));
1976 new_pipe
= CreateNamedPipeW( pipefn
, PIPE_ACCESS_DUPLEX
| FILE_FLAG_OVERLAPPED
,
1977 PIPE_TYPE_BYTE
|PIPE_WAIT
, PIPE_UNLIMITED_INSTANCES
,
1978 4096, 4096, 500 /* 0.5 second timeout */, NULL
);
1979 if (new_pipe
== INVALID_HANDLE_VALUE
)
1981 FIXME("pipe creation failed for %s, le is %u\n", debugstr_w(pipefn
), GetLastError());
1988 CloseHandle(pipe_event
);
1993 /* starts listening for a local server */
1994 HRESULT
RPC_StartLocalServer(REFCLSID clsid
, IStream
*stream
, BOOL multi_use
, void **registration
)
1997 struct local_server_params
*lsp
;
2000 lsp
= HeapAlloc(GetProcessHeap(), 0, sizeof(*lsp
));
2002 return E_OUTOFMEMORY
;
2004 lsp
->clsid
= *clsid
;
2005 lsp
->stream
= stream
;
2006 IStream_AddRef(stream
);
2007 lsp
->stop_event
= CreateEventW(NULL
, FALSE
, FALSE
, NULL
);
2008 if (!lsp
->stop_event
)
2010 HeapFree(GetProcessHeap(), 0, lsp
);
2011 return HRESULT_FROM_WIN32(GetLastError());
2013 lsp
->multi_use
= multi_use
;
2015 get_localserver_pipe_name(pipefn
, &lsp
->clsid
);
2016 lsp
->pipe
= CreateNamedPipeW(pipefn
, PIPE_ACCESS_DUPLEX
| FILE_FLAG_OVERLAPPED
,
2017 PIPE_TYPE_BYTE
|PIPE_WAIT
, PIPE_UNLIMITED_INSTANCES
,
2018 4096, 4096, 500 /* 0.5 second timeout */, NULL
);
2019 if (lsp
->pipe
== INVALID_HANDLE_VALUE
)
2021 err
= GetLastError();
2022 FIXME("pipe creation failed for %s, le is %u\n", debugstr_w(pipefn
), GetLastError());
2023 CloseHandle(lsp
->stop_event
);
2024 HeapFree(GetProcessHeap(), 0, lsp
);
2025 return HRESULT_FROM_WIN32(err
);
2028 lsp
->thread
= CreateThread(NULL
, 0, local_server_thread
, lsp
, 0, &tid
);
2031 CloseHandle(lsp
->pipe
);
2032 CloseHandle(lsp
->stop_event
);
2033 HeapFree(GetProcessHeap(), 0, lsp
);
2034 return HRESULT_FROM_WIN32(GetLastError());
2037 *registration
= lsp
;
2041 /* stops listening for a local server */
2042 void RPC_StopLocalServer(void *registration
)
2044 struct local_server_params
*lsp
= registration
;
2046 /* signal local_server_thread to stop */
2047 SetEvent(lsp
->stop_event
);
2048 /* wait for it to exit */
2049 WaitForSingleObject(lsp
->thread
, INFINITE
);
2051 IStream_Release(lsp
->stream
);
2052 CloseHandle(lsp
->stop_event
);
2053 CloseHandle(lsp
->thread
);
2054 HeapFree(GetProcessHeap(), 0, lsp
);