January 2008 - Wine-commits

Rob Shearman : rpcrt4: Call NdrCorrelationInitialize and NdrCorrelationFree from NdrClientCall2 if new correlation descriptors have been specified .
by Alexandre Julliard 07 Jan '08

07 Jan '08

Module: wine Branch: master Commit: 807a203f7704028caeda48810e1755b8e4cd9519 URL: http://source.winehq.org/git/wine.git/?a=commit;h=807a203f7704028caeda48810… Author: Rob Shearman <rob(a)codeweavers.com> Date: Mon Jan 7 15:20:25 2008 +0000 rpcrt4: Call NdrCorrelationInitialize and NdrCorrelationFree from NdrClientCall2 if new correlation descriptors have been specified. --- dlls/rpcrt4/ndr_stubless.c | 7 ++++--- 1 files changed, 4 insertions(+), 3 deletions(-) diff --git a/dlls/rpcrt4/ndr_stubless.c b/dlls/rpcrt4/ndr_stubless.c index 5c90aa0..fb76a2f 100644 --- a/dlls/rpcrt4/ndr_stubless.c +++ b/dlls/rpcrt4/ndr_stubless.c @@ -562,6 +562,8 @@ LONG_PTR WINAPIV NdrClientCall2(PMIDL_STUB_DESC pStubDesc, PFORMAT_STRING pForma /* the pointer to the object when in OLE mode */ void * This = NULL; PFORMAT_STRING pHandleFormat; + /* correlation cache */ + unsigned long NdrCorrCache[256]; TRACE("pStubDesc %p, pFormat %p, ...\n", pStubDesc, pFormat); @@ -663,8 +665,7 @@ LONG_PTR WINAPIV NdrClientCall2(PMIDL_STUB_DESC pStubDesc, PFORMAT_STRING pForma if (ext_flags.HasNewCorrDesc) { /* initialize extra correlation package */ - FIXME("new correlation description not implemented\n"); - stubMsg.fHasNewCorrDesc = TRUE; + NdrCorrelationInitialize(&stubMsg, NdrCorrCache, sizeof(NdrCorrCache), 0); } /* order of phases: @@ -744,7 +745,7 @@ LONG_PTR WINAPIV NdrClientCall2(PMIDL_STUB_DESC pStubDesc, PFORMAT_STRING pForma if (ext_flags.HasNewCorrDesc) { /* free extra correlation package */ - /* NdrCorrelationFree(&stubMsg); */ + NdrCorrelationFree(&stubMsg); } if (Oif_flags.HasPipes)

1 0

Rob Shearman : rpcrt4: Add stubs for I_RpcAsyncSetHandle and I_RpcAsyncAbortCall.
by Alexandre Julliard 07 Jan '08

07 Jan '08

Module: wine Branch: master Commit: b0cbf664f661fb51d0790608125ac1a2531bcadc URL: http://source.winehq.org/git/wine.git/?a=commit;h=b0cbf664f661fb51d07906081… Author: Rob Shearman <rob(a)codeweavers.com> Date: Mon Jan 7 15:20:40 2008 +0000 rpcrt4: Add stubs for I_RpcAsyncSetHandle and I_RpcAsyncAbortCall. --- dlls/rpcrt4/rpc_message.c | 39 +++++++++++++++++++++++++++++++++++++++ dlls/rpcrt4/rpcrt4.spec | 6 +++--- 2 files changed, 42 insertions(+), 3 deletions(-) diff --git a/dlls/rpcrt4/rpc_message.c b/dlls/rpcrt4/rpc_message.c index c7282e7..bbcc798 100644 --- a/dlls/rpcrt4/rpc_message.c +++ b/dlls/rpcrt4/rpc_message.c @@ -1154,3 +1154,42 @@ RPC_STATUS WINAPI I_RpcSendReceive(PRPC_MESSAGE pMsg) I_RpcFreeBuffer(&original_message); return status; } + +/*********************************************************************** + * I_RpcAsyncSetHandle [RPCRT4.@] + * + * Sets the asynchronous state of the handle contained in the RPC message + * structure. + * + * PARAMS + * pMsg [I] RPC Message structure. + * pAsync [I] Asynchronous state to set. + * + * RETURNS + * Success: RPC_S_OK. + * Failure: Any error code. + */ +RPC_STATUS WINAPI I_RpcAsyncSetHandle(PRPC_MESSAGE pMsg, PRPC_ASYNC_STATE pAsync) +{ + FIXME("(%p, %p): stub\n", pMsg, pAsync); + return RPC_S_INVALID_BINDING; +} + +/*********************************************************************** + * I_RpcAsyncAbortCall [RPCRT4.@] + * + * Aborts an asynchronous call. + * + * PARAMS + * pAsync [I] Asynchronous state. + * ExceptionCode [I] Exception code. + * + * RETURNS + * Success: RPC_S_OK. + * Failure: Any error code. + */ +RPC_STATUS WINAPI I_RpcAsyncAbortCall(PRPC_ASYNC_STATE pAsync, ULONG ExceptionCode) +{ + FIXME("(%p, %d): stub\n", pAsync, ExceptionCode); + return RPC_S_INVALID_ASYNC_HANDLE; +} diff --git a/dlls/rpcrt4/rpcrt4.spec b/dlls/rpcrt4/rpcrt4.spec index 8b3e502..224d3a8 100644 --- a/dlls/rpcrt4/rpcrt4.spec +++ b/dlls/rpcrt4/rpcrt4.spec @@ -18,11 +18,11 @@ @ stdcall IUnknown_AddRef_Proxy(ptr) @ stdcall IUnknown_QueryInterface_Proxy(ptr ptr ptr) @ stdcall IUnknown_Release_Proxy(ptr) -@ stub I_RpcAbortAsyncCall +@ stdcall I_RpcAbortAsyncCall(ptr long) I_RpcAsyncAbortCall @ stdcall I_RpcAllocate(long) -@ stub I_RpcAsyncAbortCall +@ stdcall I_RpcAsyncAbortCall(ptr long) @ stub I_RpcAsyncSendReceive # NT4 -@ stub I_RpcAsyncSetHandle +@ stdcall I_RpcAsyncSetHandle(ptr ptr) @ stub I_RpcBCacheAllocate @ stub I_RpcBCacheFree @ stub I_RpcBindingCopy

1 0

Rob Shearman : ole32: Fix a memory leak.
by Alexandre Julliard 07 Jan '08

07 Jan '08

Module: wine Branch: master Commit: caa9f89a18a43bf1183891e59757967aba7cfbc4 URL: http://source.winehq.org/git/wine.git/?a=commit;h=caa9f89a18a43bf1183891e59… Author: Rob Shearman <rob(a)codeweavers.com> Date: Mon Jan 7 17:15:08 2008 +0000 ole32: Fix a memory leak. --- dlls/ole32/oleproxy.c | 8 +++++--- 1 files changed, 5 insertions(+), 3 deletions(-) diff --git a/dlls/ole32/oleproxy.c b/dlls/ole32/oleproxy.c index bc87997..d37c2cf 100644 --- a/dlls/ole32/oleproxy.c +++ b/dlls/ole32/oleproxy.c @@ -574,9 +574,11 @@ static HRESULT WINAPI RemUnkStub_Invoke(LPRPCSTUBBUFFER iface, *(HRESULT *)buf = hr; buf += sizeof(HRESULT); - if (hr) return hr; - /* FIXME: pQIResults is a unique pointer so pQIResults can be NULL! */ - memcpy(buf, pQIResults, cIids * sizeof(REMQIRESULT)); + if (hr == S_OK) + /* FIXME: pQIResults is a unique pointer so pQIResults can be NULL! */ + memcpy(buf, pQIResults, cIids * sizeof(REMQIRESULT)); + + CoTaskMemFree(pQIResults); break; }

1 0

Rob Shearman : ole32: Fix memory leaks caused by not calling the channel hook function when extension_size is zero .
by Alexandre Julliard 07 Jan '08

07 Jan '08

Module: wine Branch: master Commit: f2b1a1106b04948c4e8a3409585138c3af4a49ee URL: http://source.winehq.org/git/wine.git/?a=commit;h=f2b1a1106b04948c4e8a34095… Author: Rob Shearman <rob(a)codeweavers.com> Date: Mon Jan 7 17:15:17 2008 +0000 ole32: Fix memory leaks caused by not calling the channel hook function when extension_size is zero. Instead free the allocated memory outside of the channel hook function. --- dlls/ole32/rpc.c | 8 ++++---- 1 files changed, 4 insertions(+), 4 deletions(-) diff --git a/dlls/ole32/rpc.c b/dlls/ole32/rpc.c index ad6a39f..31b9eb1 100644 --- a/dlls/ole32/rpc.c +++ b/dlls/ole32/rpc.c @@ -248,8 +248,6 @@ static unsigned char * ChannelHooks_ClientFillBuffer(SChannelHookCallInfo *info, LeaveCriticalSection(&csChannelHook); - HeapFree(GetProcessHeap(), 0, data); - return buffer; } @@ -370,8 +368,6 @@ static unsigned char * ChannelHooks_ServerFillBuffer(SChannelHookCallInfo *info, LeaveCriticalSection(&csChannelHook); - HeapFree(GetProcessHeap(), 0, data); - return buffer; } @@ -563,6 +559,8 @@ static HRESULT WINAPI ServerRpcChannelBuffer_GetBuffer(LPRPCCHANNELBUFFER iface, } } + HeapFree(GetProcessHeap(), 0, channel_hook_data); + /* store the prefixed data length so that we can restore the real buffer * later */ message_state->prefix_data_len = (char *)msg->Buffer - (char *)orpcthat; @@ -751,6 +749,8 @@ static HRESULT WINAPI ClientRpcChannelBuffer_GetBuffer(LPRPCCHANNELBUFFER iface, msg->BufferLength -= message_state->prefix_data_len; } + HeapFree(GetProcessHeap(), 0, channel_hook_data); + TRACE("-- %ld\n", status); return HRESULT_FROM_WIN32(status);

1 0

Rob Shearman : rpcrt4: Add tests for the RPC_MESSAGE members set by NdrClientInitializeNew.
by Alexandre Julliard 07 Jan '08

07 Jan '08

Module: wine Branch: master Commit: 4d6aa7ca9595e117f1af5028513ac18c9c58800d URL: http://source.winehq.org/git/wine.git/?a=commit;h=4d6aa7ca9595e117f1af50285… Author: Rob Shearman <rob(a)codeweavers.com> Date: Mon Jan 7 15:19:49 2008 +0000 rpcrt4: Add tests for the RPC_MESSAGE members set by NdrClientInitializeNew. --- dlls/rpcrt4/tests/ndr_marshall.c | 19 ++++++++++++++++++- 1 files changed, 18 insertions(+), 1 deletions(-) diff --git a/dlls/rpcrt4/tests/ndr_marshall.c b/dlls/rpcrt4/tests/ndr_marshall.c index d02135b..eebed49 100644 --- a/dlls/rpcrt4/tests/ndr_marshall.c +++ b/dlls/rpcrt4/tests/ndr_marshall.c @@ -891,11 +891,28 @@ static void test_client_init(void) MIDL_STUB_MESSAGE stubMsg; RPC_MESSAGE rpcMsg; - memset(&rpcMsg, 0, sizeof(rpcMsg)); + memset(&rpcMsg, 0xcc, sizeof(rpcMsg)); memset(&stubMsg, 0xcc, sizeof(stubMsg)); NdrClientInitializeNew(&rpcMsg, &stubMsg, &Object_StubDesc, 1); +#define TEST_POINTER_UNSET(field) ok(rpcMsg.field == (void *)0xcccccccc, #field " should have been unset instead of %p\n", rpcMsg.field) + + ok(rpcMsg.Handle == NULL, "rpcMsg.Handle should have been NULL instead of %p\n", rpcMsg.Handle); + TEST_POINTER_UNSET(Buffer); + ok(rpcMsg.BufferLength == 0xcccccccc, "rpcMsg.BufferLength should have been unset instead of %d\n", rpcMsg.BufferLength); + todo_wine + ok(rpcMsg.ProcNum == 0x8001, "rpcMsg.ProcNum should have been 0x8001 instead of 0x%x\n", rpcMsg.ProcNum); + TEST_POINTER_UNSET(TransferSyntax); + ok(rpcMsg.RpcInterfaceInformation == Object_StubDesc.RpcInterfaceInformation, + "rpcMsg.RpcInterfaceInformation should have been %p instead of %p\n", + Object_StubDesc.RpcInterfaceInformation, rpcMsg.RpcInterfaceInformation); + /* Note: ReservedForRuntime not tested */ + TEST_POINTER_UNSET(ManagerEpv); + TEST_POINTER_UNSET(ImportContext); + ok(rpcMsg.RpcFlags == 0, "rpcMsg.RpcFlags should have been 0 instead of 0x%lx\n", rpcMsg.RpcFlags); +#undef TEST_POINTER_UNSET + #define TEST_ZERO(field, fmt) ok(stubMsg.field == 0, #field " should have been set to zero instead of " fmt "\n", stubMsg.field) #define TEST_POINTER_UNSET(field) ok(stubMsg.field == (void *)0xcccccccc, #field " should have been unset instead of %p\n", stubMsg.field) #define TEST_ULONG_UNSET(field) ok(stubMsg.field == 0xcccccccc, #field " should have been unset instead of 0x%x\n", stubMsg.field)

1 0

Rob Shearman : rpcrt4: Set the ProcNum of RPC_MESSAGE in NdrClientInitializeNew to include RPC_VALID_FLAGS_BIT .
by Alexandre Julliard 07 Jan '08

07 Jan '08

Module: wine Branch: master Commit: 31676530c2ec59886c167404467e1eb8840337f5 URL: http://source.winehq.org/git/wine.git/?a=commit;h=31676530c2ec59886c1674044… Author: Rob Shearman <rob(a)codeweavers.com> Date: Mon Jan 7 15:19:57 2008 +0000 rpcrt4: Set the ProcNum of RPC_MESSAGE in NdrClientInitializeNew to include RPC_VALID_FLAGS_BIT. Mask out RPC_VALID_FLAGS_BIT from ProcNum when it is used to build a packet to send to the server. --- dlls/rpcrt4/ndr_clientserver.c | 2 +- dlls/rpcrt4/rpc_message.c | 3 ++- dlls/rpcrt4/tests/ndr_marshall.c | 1 - include/rpcdcep.h | 3 +++ 4 files changed, 6 insertions(+), 3 deletions(-) diff --git a/dlls/rpcrt4/ndr_clientserver.c b/dlls/rpcrt4/ndr_clientserver.c index 809d4cb..8cce9d7 100644 --- a/dlls/rpcrt4/ndr_clientserver.c +++ b/dlls/rpcrt4/ndr_clientserver.c @@ -54,7 +54,7 @@ void WINAPI NdrClientInitializeNew( PRPC_MESSAGE pRpcMessage, PMIDL_STUB_MESSAGE pRpcMessage, pStubMsg, pStubDesc, ProcNum); pRpcMessage->Handle = NULL; - pRpcMessage->ProcNum = ProcNum; + pRpcMessage->ProcNum = ProcNum | RPC_FLAGS_VALID_BIT; pRpcMessage->RpcInterfaceInformation = pStubDesc->RpcInterfaceInformation; pRpcMessage->RpcFlags = 0; pRpcMessage->DataRepresentation = NDR_LOCAL_DATA_REPRESENTATION; diff --git a/dlls/rpcrt4/rpc_message.c b/dlls/rpcrt4/rpc_message.c index 9f0df05..c7282e7 100644 --- a/dlls/rpcrt4/rpc_message.c +++ b/dlls/rpcrt4/rpc_message.c @@ -1000,7 +1000,8 @@ RPC_STATUS WINAPI I_RpcSend(PRPC_MESSAGE pMsg) if (status != RPC_S_OK) return status; hdr = RPCRT4_BuildRequestHeader(pMsg->DataRepresentation, - pMsg->BufferLength, pMsg->ProcNum, + pMsg->BufferLength, + pMsg->ProcNum & ~RPC_FLAGS_VALID_BIT, &bind->ObjectUuid); if (!hdr) { diff --git a/dlls/rpcrt4/tests/ndr_marshall.c b/dlls/rpcrt4/tests/ndr_marshall.c index eebed49..c9b8fa9 100644 --- a/dlls/rpcrt4/tests/ndr_marshall.c +++ b/dlls/rpcrt4/tests/ndr_marshall.c @@ -901,7 +901,6 @@ static void test_client_init(void) ok(rpcMsg.Handle == NULL, "rpcMsg.Handle should have been NULL instead of %p\n", rpcMsg.Handle); TEST_POINTER_UNSET(Buffer); ok(rpcMsg.BufferLength == 0xcccccccc, "rpcMsg.BufferLength should have been unset instead of %d\n", rpcMsg.BufferLength); - todo_wine ok(rpcMsg.ProcNum == 0x8001, "rpcMsg.ProcNum should have been 0x8001 instead of 0x%x\n", rpcMsg.ProcNum); TEST_POINTER_UNSET(TransferSyntax); ok(rpcMsg.RpcInterfaceInformation == Object_StubDesc.RpcInterfaceInformation, diff --git a/include/rpcdcep.h b/include/rpcdcep.h index b93c9a6..f0b78c1 100644 --- a/include/rpcdcep.h +++ b/include/rpcdcep.h @@ -48,6 +48,9 @@ typedef struct _RPC_MESSAGE unsigned long RpcFlags; } RPC_MESSAGE, *PRPC_MESSAGE; +/* or'd with ProcNum */ +#define RPC_FLAGS_VALID_BIT 0x00008000 + #define RPC_CONTEXT_HANDLE_DEFAULT_GUARD ((void *)0xfffff00d) #define RPC_CONTEXT_HANDLE_DEFAULT_FLAGS 0x00000000

1 0

Rob Shearman : rpcrt4: ndr_stubless.h should include ndrtypes. h as it depends on types it defines.
by Alexandre Julliard 07 Jan '08

07 Jan '08

Module: wine Branch: master Commit: f9a54a58d5401b22f859c4fd1e08f9dbf6c4f993 URL: http://source.winehq.org/git/wine.git/?a=commit;h=f9a54a58d5401b22f859c4fd1… Author: Rob Shearman <rob(a)codeweavers.com> Date: Mon Jan 7 15:20:05 2008 +0000 rpcrt4: ndr_stubless.h should include ndrtypes.h as it depends on types it defines. --- dlls/rpcrt4/ndr_stubless.c | 1 - dlls/rpcrt4/ndr_stubless.h | 2 ++ 2 files changed, 2 insertions(+), 1 deletions(-) diff --git a/dlls/rpcrt4/ndr_stubless.c b/dlls/rpcrt4/ndr_stubless.c index c21b809..5c90aa0 100644 --- a/dlls/rpcrt4/ndr_stubless.c +++ b/dlls/rpcrt4/ndr_stubless.c @@ -37,7 +37,6 @@ #include "objbase.h" #include "rpc.h" #include "rpcproxy.h" -#include "ndrtypes.h" #include "wine/debug.h" #include "wine/rpcfc.h" diff --git a/dlls/rpcrt4/ndr_stubless.h b/dlls/rpcrt4/ndr_stubless.h index 1bd3a95..02850d1 100644 --- a/dlls/rpcrt4/ndr_stubless.h +++ b/dlls/rpcrt4/ndr_stubless.h @@ -18,6 +18,8 @@ * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA */ +#include "ndrtypes.h" + /* there can't be any alignment with the structures in this file */ #include "pshpack1.h"

1 0

Rob Shearman : ole32: Clear the RPC_FLAGS_VALID_BIT flag from the RPC_MESSAGE ProcNum when shortcutting the RPC runtime .
by Alexandre Julliard 07 Jan '08

07 Jan '08

Module: wine Branch: master Commit: e3e704bc43cb9851a65763f32b9bba4f4600e237 URL: http://source.winehq.org/git/wine.git/?a=commit;h=e3e704bc43cb9851a65763f32… Author: Rob Shearman <rob(a)codeweavers.com> Date: Mon Jan 7 17:09:52 2008 +0000 ole32: Clear the RPC_FLAGS_VALID_BIT flag from the RPC_MESSAGE ProcNum when shortcutting the RPC runtime. --- dlls/ole32/rpc.c | 2 ++ 1 files changed, 2 insertions(+), 0 deletions(-) diff --git a/dlls/ole32/rpc.c b/dlls/ole32/rpc.c index fdb49cb..2ee8a9c 100644 --- a/dlls/ole32/rpc.c +++ b/dlls/ole32/rpc.c @@ -821,6 +821,8 @@ static HRESULT WINAPI ClientRpcChannelBuffer_SendReceive(LPRPCCHANNELBUFFER ifac { TRACE("Calling apartment thread 0x%08x...\n", message_state->target_tid); + msg->ProcNum &= ~RPC_FLAGS_VALID_BIT; + if (!PostMessageW(message_state->target_hwnd, DM_EXECUTERPC, 0, (LPARAM)&message_state->params)) {

1 0

Rob Shearman : ole32: Shortcut the RPC runtime for allocation of the message when processing the COM call by just sending a message to a window in the process .
by Alexandre Julliard 07 Jan '08

07 Jan '08

Module: wine Branch: master Commit: b6553e7e297fc234aaaa632ff84ab4936c544292 URL: http://source.winehq.org/git/wine.git/?a=commit;h=b6553e7e297fc234aaaa632ff… Author: Rob Shearman <rob(a)codeweavers.com> Date: Mon Jan 7 17:14:54 2008 +0000 ole32: Shortcut the RPC runtime for allocation of the message when processing the COM call by just sending a message to a window in the process. --- dlls/ole32/rpc.c | 50 +++++++++++++++++++++++++++++++++++++++++++++----- 1 files changed, 45 insertions(+), 5 deletions(-) diff --git a/dlls/ole32/rpc.c b/dlls/ole32/rpc.c index 2ee8a9c..ad6a39f 100644 --- a/dlls/ole32/rpc.c +++ b/dlls/ole32/rpc.c @@ -118,6 +118,7 @@ struct dispatch_params IID iid; /* ID of interface being called */ IUnknown *iface; /* interface being called */ HANDLE handle; /* handle that will become signaled when call finishes */ + BOOL bypass_rpcrt; /* bypass RPC runtime? */ RPC_STATUS status; /* status (out) */ HRESULT hr; /* hresult (out) */ }; @@ -127,6 +128,7 @@ struct message_state RPC_BINDING_HANDLE binding_handle; ULONG prefix_data_len; SChannelHookCallInfo channel_hook_info; + BOOL bypass_rpcrt; /* client only */ HWND target_hwnd; @@ -513,7 +515,16 @@ static HRESULT WINAPI ServerRpcChannelBuffer_GetBuffer(LPRPCCHANNELBUFFER iface, msg->BufferLength += FIELD_OFFSET(WIRE_ORPC_EXTENT, data[0]); } - status = I_RpcGetBuffer(msg); + if (message_state->bypass_rpcrt) + { + msg->Buffer = HeapAlloc(GetProcessHeap(), 0, msg->BufferLength); + if (msg->Buffer) + status = RPC_S_OK; + else + status = ERROR_OUTOFMEMORY; + } + else + status = I_RpcGetBuffer(msg); orpcthat = (ORPCTHAT *)msg->Buffer; msg->Buffer = (char *)msg->Buffer + FIELD_OFFSET(ORPCTHAT, extensions); @@ -663,6 +674,7 @@ static HRESULT WINAPI ClientRpcChannelBuffer_GetBuffer(LPRPCCHANNELBUFFER iface, } else { + message_state->params.bypass_rpcrt = TRUE; message_state->target_hwnd = apartment_getwindow(apt); message_state->target_tid = apt->tid; /* we assume later on that this being non-NULL is the indicator that @@ -676,7 +688,17 @@ static HRESULT WINAPI ClientRpcChannelBuffer_GetBuffer(LPRPCCHANNELBUFFER iface, /* Note: message_state->params.msg is initialised in * ClientRpcChannelBuffer_SendReceive */ - status = I_RpcGetBuffer(msg); + /* shortcut the RPC runtime */ + if (message_state->target_hwnd) + { + msg->Buffer = HeapAlloc(GetProcessHeap(), 0, msg->BufferLength); + if (msg->Buffer) + status = RPC_S_OK; + else + status = ERROR_OUTOFMEMORY; + } + else + status = I_RpcGetBuffer(msg); msg->Handle = message_state; @@ -817,7 +839,7 @@ static HRESULT WINAPI ClientRpcChannelBuffer_SendReceive(LPRPCCHANNELBUFFER ifac * from DllMain */ message_state->params.msg = olemsg; - if (message_state->target_hwnd) + if (message_state->params.bypass_rpcrt) { TRACE("Calling apartment thread 0x%08x...\n", message_state->target_tid); @@ -939,7 +961,13 @@ static HRESULT WINAPI ServerRpcChannelBuffer_FreeBuffer(LPRPCCHANNELBUFFER iface msg->BufferLength += message_state->prefix_data_len; message_state->prefix_data_len = 0; - status = I_RpcFreeBuffer(msg); + if (message_state->bypass_rpcrt) + { + HeapFree(GetProcessHeap(), 0, msg->Buffer); + status = RPC_S_OK; + } + else + status = I_RpcFreeBuffer(msg); msg->Handle = message_state; @@ -962,7 +990,13 @@ static HRESULT WINAPI ClientRpcChannelBuffer_FreeBuffer(LPRPCCHANNELBUFFER iface msg->Buffer = (char *)msg->Buffer - message_state->prefix_data_len; msg->BufferLength += message_state->prefix_data_len; - status = I_RpcFreeBuffer(msg); + if (message_state->params.bypass_rpcrt) + { + HeapFree(GetProcessHeap(), 0, msg->Buffer); + status = RPC_S_OK; + } + else + status = I_RpcFreeBuffer(msg); HeapFree(GetProcessHeap(), 0, msg->RpcInterfaceInformation); msg->RpcInterfaceInformation = NULL; @@ -1286,6 +1320,7 @@ void RPC_ExecuteCall(struct dispatch_params *params) message_state->prefix_data_len = (char *)msg->Buffer - original_buffer; message_state->binding_handle = msg->Handle; + message_state->bypass_rpcrt = params->bypass_rpcrt; message_state->channel_hook_info.iid = params->iid; message_state->channel_hook_info.cbSize = sizeof(message_state->channel_hook_info); @@ -1357,6 +1392,10 @@ void RPC_ExecuteCall(struct dispatch_params *params) COM_CurrentInfo()->pending_call_count_server--; COM_CurrentInfo()->causality_id = old_causality_id; + /* the invoke allocated a new buffer, so free the old one */ + if (message_state->bypass_rpcrt && original_buffer != msg->Buffer) + HeapFree(GetProcessHeap(), 0, original_buffer); + exit_reset_state: message_state = (struct message_state *)msg->Handle; msg->Handle = message_state->binding_handle; @@ -1400,6 +1439,7 @@ static void __RPC_STUB dispatch_rpc(RPC_MESSAGE *msg) params->status = RPC_S_OK; params->hr = S_OK; params->handle = NULL; + params->bypass_rpcrt = FALSE; /* Note: this is the important difference between STAs and MTAs - we * always execute RPCs to STAs in the thread that originally created the

1 0

Alexandre Julliard : server: Output incorrectly-terminated strings in hex format.
by Alexandre Julliard 07 Jan '08

07 Jan '08

Module: wine Branch: master Commit: b647ded03dcbc036249aeb0318cab038131f191c URL: http://source.winehq.org/git/wine.git/?a=commit;h=b647ded03dcbc036249aeb031… Author: Alexandre Julliard <julliard(a)winehq.org> Date: Mon Jan 7 21:06:49 2008 +0100 server: Output incorrectly-terminated strings in hex format. --- server/registry.c | 50 +++++++++++++++++++++++++------------------------- 1 files changed, 25 insertions(+), 25 deletions(-) diff --git a/server/registry.c b/server/registry.c index bfcd3d2..d402f2c 100644 --- a/server/registry.c +++ b/server/registry.c @@ -182,7 +182,7 @@ static void dump_path( const struct key *key, const struct key *base, FILE *f ) /* dump a value to a text file */ static void dump_value( const struct key_value *value, FILE *f ) { - unsigned int i; + unsigned int i, dw; int count; if (value->namelen) @@ -198,37 +198,37 @@ static void dump_value( const struct key_value *value, FILE *f ) case REG_SZ: case REG_EXPAND_SZ: case REG_MULTI_SZ: - if (value->type != REG_SZ) fprintf( f, "str(%d):", value->type ); + /* only output properly terminated strings in string format */ + if (value->len < sizeof(WCHAR)) break; + if (value->len % sizeof(WCHAR)) break; + if (((WCHAR *)value->data)[value->len / sizeof(WCHAR) - 1]) break; + if (value->type != REG_SZ) fprintf( f, "str(%x):", value->type ); fputc( '\"', f ); if (value->data) dump_strW( (WCHAR *)value->data, value->len / sizeof(WCHAR), f, "\"\"" ); - fputc( '\"', f ); - break; + fprintf( f, "\"\n" ); + return; + case REG_DWORD: - if (value->len == sizeof(DWORD)) - { - DWORD dw; - memcpy( &dw, value->data, sizeof(DWORD) ); - fprintf( f, "dword:%08x", dw ); - break; - } - /* else fall through */ - default: - if (value->type == REG_BINARY) count += fprintf( f, "hex:" ); - else count += fprintf( f, "hex(%x):", value->type ); - for (i = 0; i < value->len; i++) + if (value->len != sizeof(dw)) break; + memcpy( &dw, value->data, sizeof(dw) ); + fprintf( f, "dword:%08x\n", dw ); + return; + } + + if (value->type == REG_BINARY) count += fprintf( f, "hex:" ); + else count += fprintf( f, "hex(%x):", value->type ); + for (i = 0; i < value->len; i++) + { + count += fprintf( f, "%02x", *((unsigned char *)value->data + i) ); + if (i < value->len-1) { - count += fprintf( f, "%02x", *((unsigned char *)value->data + i) ); - if (i < value->len-1) + fputc( ',', f ); + if (++count > 76) { - fputc( ',', f ); - if (++count > 76) - { - fprintf( f, "\\\n " ); - count = 2; - } + fprintf( f, "\\\n " ); + count = 2; } } - break; } fputc( '\n', f ); }

1 0