Commit 2db1d0de authored by Rob Shearman's avatar Rob Shearman Committed by Alexandre Julliard

ole32: Make sure to reset the message state if it has been changed in…

ole32: Make sure to reset the message state if it has been changed in RPC_ExecuteCall before returning. Fix the hrFault parameter passed into the ClientNotify function in ClientRpcChannelBuffer_SendReceive so that it is correct in the case of messages dispatched with PostMessage. Only unmarshal ORPCTHAT for non-fault packets.
parent d57dc9da
...@@ -740,6 +740,7 @@ static HRESULT WINAPI ClientRpcChannelBuffer_SendReceive(LPRPCCHANNELBUFFER ifac ...@@ -740,6 +740,7 @@ static HRESULT WINAPI ClientRpcChannelBuffer_SendReceive(LPRPCCHANNELBUFFER ifac
ORPCTHAT orpcthat; ORPCTHAT orpcthat;
ORPC_EXTENT_ARRAY orpc_ext_array; ORPC_EXTENT_ARRAY orpc_ext_array;
WIRE_ORPC_EXTENT *first_wire_orpc_extent = NULL; WIRE_ORPC_EXTENT *first_wire_orpc_extent = NULL;
HRESULT hrFault = S_OK;
TRACE("(%p) iMethod=%d\n", olemsg, olemsg->iMethod); TRACE("(%p) iMethod=%d\n", olemsg, olemsg->iMethod);
...@@ -836,7 +837,9 @@ static HRESULT WINAPI ClientRpcChannelBuffer_SendReceive(LPRPCCHANNELBUFFER ifac ...@@ -836,7 +837,9 @@ static HRESULT WINAPI ClientRpcChannelBuffer_SendReceive(LPRPCCHANNELBUFFER ifac
} }
ClientRpcChannelBuffer_ReleaseEventHandle(This, params->handle); ClientRpcChannelBuffer_ReleaseEventHandle(This, params->handle);
if (hr == S_OK) hr = params->hr; /* for WM shortcut, faults are returned in params->hr */
if (hr == S_OK)
hrFault = params->hr;
status = params->status; status = params->status;
HeapFree(GetProcessHeap(), 0, params); HeapFree(GetProcessHeap(), 0, params);
...@@ -845,7 +848,21 @@ static HRESULT WINAPI ClientRpcChannelBuffer_SendReceive(LPRPCCHANNELBUFFER ifac ...@@ -845,7 +848,21 @@ static HRESULT WINAPI ClientRpcChannelBuffer_SendReceive(LPRPCCHANNELBUFFER ifac
orpcthat.flags = ORPCF_NULL; orpcthat.flags = ORPCF_NULL;
orpcthat.extensions = NULL; orpcthat.extensions = NULL;
if (status == RPC_S_OK && msg->BufferLength > FIELD_OFFSET(ORPCTHAT, extensions) + 4) /* for normal RPC calls, faults are returned in first 4 bytes of the
* buffer */
TRACE("RPC call status: 0x%lx\n", status);
if (status == RPC_S_CALL_FAILED)
hrFault = *(HRESULT *)olemsg->Buffer;
else if (status != RPC_S_OK)
hr = HRESULT_FROM_WIN32(status);
TRACE("hrFault = 0x%08x\n", hrFault);
/* FIXME: this condition should be
* "hr == S_OK && (!hrFault || msg->BufferLength > FIELD_OFFSET(ORPCTHAT, extentions) + 4)"
* but we don't currently reset the message length for PostMessage
* dispatched calls */
if (hr == S_OK && hrFault == S_OK)
{ {
HRESULT hr2; HRESULT hr2;
char *original_buffer = msg->Buffer; char *original_buffer = msg->Buffer;
...@@ -862,26 +879,22 @@ static HRESULT WINAPI ClientRpcChannelBuffer_SendReceive(LPRPCCHANNELBUFFER ifac ...@@ -862,26 +879,22 @@ static HRESULT WINAPI ClientRpcChannelBuffer_SendReceive(LPRPCCHANNELBUFFER ifac
else else
message_state->prefix_data_len = 0; message_state->prefix_data_len = 0;
ChannelHooks_ClientNotify(&message_state->channel_hook_info, if (hr == S_OK)
msg->DataRepresentation, {
first_wire_orpc_extent, ChannelHooks_ClientNotify(&message_state->channel_hook_info,
orpcthat.extensions && first_wire_orpc_extent ? orpcthat.extensions->size : 0, msg->DataRepresentation,
status == RPC_S_CALL_FAILED && msg->BufferLength >= sizeof(HRESULT) ? *(HRESULT *)msg->Buffer : S_OK); first_wire_orpc_extent,
orpcthat.extensions && first_wire_orpc_extent ? orpcthat.extensions->size : 0,
hrFault);
}
/* save away the message state again */ /* save away the message state again */
msg->Handle = message_state; msg->Handle = message_state;
if (hr) return hr;
if (pstatus) *pstatus = status; if (pstatus) *pstatus = status;
TRACE("RPC call status: 0x%lx\n", status); if (hr == S_OK)
if (status == RPC_S_OK) hr = hrFault;
hr = S_OK;
else if (status == RPC_S_CALL_FAILED)
hr = *(HRESULT *)olemsg->Buffer;
else
hr = HRESULT_FROM_WIN32(status);
TRACE("-- 0x%08x\n", hr); TRACE("-- 0x%08x\n", hr);
...@@ -1228,12 +1241,16 @@ void RPC_ExecuteCall(struct dispatch_params *params) ...@@ -1228,12 +1241,16 @@ void RPC_ExecuteCall(struct dispatch_params *params)
params->hr = unmarshal_ORPCTHIS(msg, &orpcthis, &orpc_ext_array, &first_wire_orpc_extent); params->hr = unmarshal_ORPCTHIS(msg, &orpcthis, &orpc_ext_array, &first_wire_orpc_extent);
if (params->hr != S_OK) if (params->hr != S_OK)
{
msg->Buffer = original_buffer;
goto exit; goto exit;
}
message_state = HeapAlloc(GetProcessHeap(), 0, sizeof(*message_state)); message_state = HeapAlloc(GetProcessHeap(), 0, sizeof(*message_state));
if (!message_state) if (!message_state)
{ {
params->hr = E_OUTOFMEMORY; params->hr = E_OUTOFMEMORY;
msg->Buffer = original_buffer;
goto exit; goto exit;
} }
...@@ -1283,11 +1300,11 @@ void RPC_ExecuteCall(struct dispatch_params *params) ...@@ -1283,11 +1300,11 @@ void RPC_ExecuteCall(struct dispatch_params *params)
{ {
case SERVERCALL_REJECTED: case SERVERCALL_REJECTED:
params->hr = RPC_E_CALL_REJECTED; params->hr = RPC_E_CALL_REJECTED;
goto exit; goto exit_reset_state;
case SERVERCALL_RETRYLATER: case SERVERCALL_RETRYLATER:
#if 0 /* FIXME: handle retries on the client side before enabling this code */ #if 0 /* FIXME: handle retries on the client side before enabling this code */
params->hr = RPC_E_RETRY; params->hr = RPC_E_RETRY;
goto exit; goto exit_reset_state;
#else #else
FIXME("retry call later not implemented\n"); FIXME("retry call later not implemented\n");
break; break;
...@@ -1310,6 +1327,7 @@ void RPC_ExecuteCall(struct dispatch_params *params) ...@@ -1310,6 +1327,7 @@ void RPC_ExecuteCall(struct dispatch_params *params)
COM_CurrentInfo()->pending_call_count_server--; COM_CurrentInfo()->pending_call_count_server--;
COM_CurrentInfo()->causality_id = old_causality_id; COM_CurrentInfo()->causality_id = old_causality_id;
exit_reset_state:
message_state = (struct message_state *)msg->Handle; message_state = (struct message_state *)msg->Handle;
msg->Handle = message_state->binding_handle; msg->Handle = message_state->binding_handle;
msg->Buffer = (char *)msg->Buffer - message_state->prefix_data_len; msg->Buffer = (char *)msg->Buffer - message_state->prefix_data_len;
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment