2012-11-28 21:47:04 +04:00
|
|
|
/**
|
|
|
|
* FreeRDP: A Remote Desktop Protocol Implementation
|
|
|
|
* RPC Client
|
|
|
|
*
|
|
|
|
* Copyright 2012 Marc-Andre Moreau <marcandre.moreau@gmail.com>
|
|
|
|
*
|
|
|
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
|
|
* you may not use this file except in compliance with the License.
|
|
|
|
* You may obtain a copy of the License at
|
|
|
|
*
|
|
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
*
|
|
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
|
|
* See the License for the specific language governing permissions and
|
|
|
|
* limitations under the License.
|
|
|
|
*/
|
|
|
|
|
|
|
|
#ifdef HAVE_CONFIG_H
|
|
|
|
#include "config.h"
|
|
|
|
#endif
|
|
|
|
|
2014-09-12 16:36:29 +04:00
|
|
|
#include <freerdp/log.h>
|
2013-12-21 03:22:29 +04:00
|
|
|
|
2012-11-28 21:47:04 +04:00
|
|
|
#include <winpr/crt.h>
|
2012-12-13 00:55:42 +04:00
|
|
|
#include <winpr/print.h>
|
2012-11-28 21:47:04 +04:00
|
|
|
#include <winpr/synch.h>
|
|
|
|
#include <winpr/thread.h>
|
2012-12-11 00:43:07 +04:00
|
|
|
#include <winpr/stream.h>
|
2012-11-28 21:47:04 +04:00
|
|
|
|
2015-02-04 01:17:17 +03:00
|
|
|
#include "http.h"
|
|
|
|
#include "ncacn_http.h"
|
|
|
|
|
2015-02-02 16:19:07 +03:00
|
|
|
#include "rpc_bind.h"
|
2012-12-12 04:17:57 +04:00
|
|
|
#include "rpc_fault.h"
|
2012-11-28 21:47:04 +04:00
|
|
|
#include "rpc_client.h"
|
2013-05-22 12:58:11 +04:00
|
|
|
#include "../rdp.h"
|
|
|
|
|
2015-02-03 02:50:26 +03:00
|
|
|
#define TAG FREERDP_TAG("core.gateway.rpc")
|
2012-12-12 09:49:15 +04:00
|
|
|
|
2015-02-02 16:19:07 +03:00
|
|
|
static void rpc_pdu_reset(RPC_PDU* pdu)
|
|
|
|
{
|
|
|
|
pdu->Type = 0;
|
|
|
|
pdu->Flags = 0;
|
|
|
|
pdu->CallId = 0;
|
|
|
|
Stream_SetPosition(pdu->s, 0);
|
|
|
|
}
|
|
|
|
|
2015-02-02 04:47:43 +03:00
|
|
|
RPC_PDU* rpc_pdu_new()
|
2012-12-12 09:49:15 +04:00
|
|
|
{
|
2015-02-02 04:47:43 +03:00
|
|
|
RPC_PDU* pdu;
|
2012-12-12 09:49:15 +04:00
|
|
|
|
2015-02-02 04:47:43 +03:00
|
|
|
pdu = (RPC_PDU*) malloc(sizeof(RPC_PDU));
|
2012-12-12 09:49:15 +04:00
|
|
|
|
|
|
|
if (!pdu)
|
2015-02-02 04:47:43 +03:00
|
|
|
return NULL;
|
2014-08-18 19:22:43 +04:00
|
|
|
|
2015-02-02 04:47:43 +03:00
|
|
|
pdu->s = Stream_New(NULL, 4096);
|
2014-08-18 19:22:43 +04:00
|
|
|
|
2015-02-02 04:47:43 +03:00
|
|
|
if (!pdu->s)
|
|
|
|
{
|
|
|
|
free(pdu);
|
|
|
|
return NULL;
|
2012-12-12 09:49:15 +04:00
|
|
|
}
|
2012-12-12 04:17:57 +04:00
|
|
|
|
2015-02-02 16:19:07 +03:00
|
|
|
rpc_pdu_reset(pdu);
|
2015-02-02 04:47:43 +03:00
|
|
|
|
|
|
|
return pdu;
|
|
|
|
}
|
|
|
|
|
|
|
|
static void rpc_pdu_free(RPC_PDU* pdu)
|
|
|
|
{
|
|
|
|
if (!pdu)
|
|
|
|
return;
|
|
|
|
|
|
|
|
Stream_Free(pdu->s, TRUE);
|
|
|
|
free(pdu);
|
|
|
|
}
|
|
|
|
|
2015-02-01 21:09:28 +03:00
|
|
|
int rpc_client_receive_pipe_write(rdpRpc* rpc, const BYTE* buffer, size_t length)
|
2015-02-01 00:56:25 +03:00
|
|
|
{
|
2015-02-01 21:09:28 +03:00
|
|
|
int status = 0;
|
|
|
|
RpcClient* client = rpc->client;
|
|
|
|
|
|
|
|
EnterCriticalSection(&(rpc->client->PipeLock));
|
|
|
|
|
|
|
|
if (ringbuffer_write(&(client->ReceivePipe), buffer, length))
|
|
|
|
status += (int) length;
|
|
|
|
|
|
|
|
if (ringbuffer_used(&(client->ReceivePipe)) > 0)
|
|
|
|
SetEvent(client->PipeEvent);
|
|
|
|
|
|
|
|
LeaveCriticalSection(&(rpc->client->PipeLock));
|
|
|
|
|
|
|
|
return status;
|
|
|
|
}
|
|
|
|
|
|
|
|
int rpc_client_receive_pipe_read(rdpRpc* rpc, BYTE* buffer, size_t length)
|
|
|
|
{
|
|
|
|
int index = 0;
|
|
|
|
int status = 0;
|
|
|
|
int nchunks = 0;
|
|
|
|
DataChunk chunks[2];
|
|
|
|
RpcClient* client = rpc->client;
|
|
|
|
|
|
|
|
EnterCriticalSection(&(client->PipeLock));
|
|
|
|
|
|
|
|
nchunks = ringbuffer_peek(&(client->ReceivePipe), chunks, length);
|
|
|
|
|
|
|
|
for (index = 0; index < nchunks; index++)
|
|
|
|
{
|
|
|
|
CopyMemory(&buffer[status], chunks[index].data, chunks[index].size);
|
|
|
|
status += chunks[index].size;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (status > 0)
|
|
|
|
ringbuffer_commit_read_bytes(&(client->ReceivePipe), status);
|
|
|
|
|
|
|
|
if (ringbuffer_used(&(client->ReceivePipe)) < 1)
|
|
|
|
ResetEvent(client->PipeEvent);
|
|
|
|
|
|
|
|
LeaveCriticalSection(&(client->PipeLock));
|
|
|
|
|
|
|
|
return status;
|
|
|
|
}
|
|
|
|
|
2015-02-03 02:50:26 +03:00
|
|
|
int rpc_client_transition_to_state(rdpRpc* rpc, RPC_CLIENT_STATE state)
|
|
|
|
{
|
|
|
|
int status = 1;
|
|
|
|
const char* str = "RPC_CLIENT_STATE_UNKNOWN";
|
|
|
|
|
|
|
|
switch (state)
|
|
|
|
{
|
|
|
|
case RPC_CLIENT_STATE_INITIAL:
|
|
|
|
str = "RPC_CLIENT_STATE_INITIAL";
|
|
|
|
break;
|
|
|
|
|
|
|
|
case RPC_CLIENT_STATE_ESTABLISHED:
|
|
|
|
str = "RPC_CLIENT_STATE_ESTABLISHED";
|
|
|
|
break;
|
|
|
|
|
|
|
|
case RPC_CLIENT_STATE_WAIT_SECURE_BIND_ACK:
|
|
|
|
str = "RPC_CLIENT_STATE_WAIT_SECURE_BIND_ACK";
|
|
|
|
break;
|
|
|
|
|
|
|
|
case RPC_CLIENT_STATE_WAIT_UNSECURE_BIND_ACK:
|
|
|
|
str = "RPC_CLIENT_STATE_WAIT_UNSECURE_BIND_ACK";
|
|
|
|
break;
|
|
|
|
|
|
|
|
case RPC_CLIENT_STATE_WAIT_SECURE_ALTER_CONTEXT_RESPONSE:
|
|
|
|
str = "RPC_CLIENT_STATE_WAIT_SECURE_ALTER_CONTEXT_RESPONSE";
|
|
|
|
break;
|
|
|
|
|
|
|
|
case RPC_CLIENT_STATE_CONTEXT_NEGOTIATED:
|
|
|
|
str = "RPC_CLIENT_STATE_CONTEXT_NEGOTIATED";
|
|
|
|
break;
|
|
|
|
|
|
|
|
case RPC_CLIENT_STATE_WAIT_RESPONSE:
|
|
|
|
str = "RPC_CLIENT_STATE_WAIT_RESPONSE";
|
|
|
|
break;
|
|
|
|
|
|
|
|
case RPC_CLIENT_STATE_FINAL:
|
|
|
|
str = "RPC_CLIENT_STATE_FINAL";
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
|
|
|
|
rpc->State = state;
|
|
|
|
WLog_DBG(TAG, "%s", str);
|
|
|
|
|
|
|
|
return status;
|
|
|
|
}
|
|
|
|
|
2015-02-02 04:47:43 +03:00
|
|
|
int rpc_client_recv_pdu(rdpRpc* rpc, RPC_PDU* pdu)
|
2015-02-01 21:09:28 +03:00
|
|
|
{
|
2015-02-03 02:50:26 +03:00
|
|
|
int status = -1;
|
2015-02-02 16:19:07 +03:00
|
|
|
rpcconn_rts_hdr_t* rts;
|
|
|
|
rdpTsg* tsg = rpc->transport->tsg;
|
|
|
|
|
2015-02-04 00:33:45 +03:00
|
|
|
if (rpc->VirtualConnection->State < VIRTUAL_CONNECTION_STATE_OPENED)
|
2015-02-02 16:19:07 +03:00
|
|
|
{
|
|
|
|
switch (rpc->VirtualConnection->State)
|
|
|
|
{
|
|
|
|
case VIRTUAL_CONNECTION_STATE_INITIAL:
|
|
|
|
break;
|
|
|
|
|
|
|
|
case VIRTUAL_CONNECTION_STATE_OUT_CHANNEL_WAIT:
|
|
|
|
break;
|
|
|
|
|
|
|
|
case VIRTUAL_CONNECTION_STATE_WAIT_A3W:
|
|
|
|
|
|
|
|
rts = (rpcconn_rts_hdr_t*) Stream_Buffer(pdu->s);
|
|
|
|
|
|
|
|
if (!rts_match_pdu_signature(rpc, &RTS_PDU_CONN_A3_SIGNATURE, rts))
|
|
|
|
{
|
|
|
|
WLog_ERR(TAG, "unexpected RTS PDU: Expected CONN/A3");
|
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
|
2015-02-03 22:44:31 +03:00
|
|
|
status = rts_recv_CONN_A3_pdu(rpc, Stream_Buffer(pdu->s), Stream_Length(pdu->s));
|
|
|
|
|
|
|
|
if (status < 0)
|
|
|
|
{
|
|
|
|
WLog_ERR(TAG, "rts_recv_CONN_A3_pdu failure");
|
|
|
|
return -1;
|
|
|
|
}
|
2015-02-02 16:19:07 +03:00
|
|
|
|
2015-02-03 02:50:26 +03:00
|
|
|
rpc_client_virtual_connection_transition_to_state(rpc,
|
|
|
|
rpc->VirtualConnection, VIRTUAL_CONNECTION_STATE_WAIT_C2);
|
|
|
|
|
|
|
|
status = 1;
|
2015-02-02 16:19:07 +03:00
|
|
|
|
|
|
|
break;
|
|
|
|
|
|
|
|
case VIRTUAL_CONNECTION_STATE_WAIT_C2:
|
|
|
|
|
|
|
|
rts = (rpcconn_rts_hdr_t*) Stream_Buffer(pdu->s);
|
|
|
|
|
|
|
|
if (!rts_match_pdu_signature(rpc, &RTS_PDU_CONN_C2_SIGNATURE, rts))
|
|
|
|
{
|
|
|
|
WLog_ERR(TAG, "unexpected RTS PDU: Expected CONN/C2");
|
2015-02-03 22:44:31 +03:00
|
|
|
return -1;
|
2015-02-02 16:19:07 +03:00
|
|
|
}
|
|
|
|
|
2015-02-03 22:44:31 +03:00
|
|
|
status = rts_recv_CONN_C2_pdu(rpc, Stream_Buffer(pdu->s), Stream_Length(pdu->s));
|
|
|
|
|
|
|
|
if (status < 0)
|
|
|
|
{
|
|
|
|
WLog_ERR(TAG, "rts_recv_CONN_C2_pdu failure");
|
|
|
|
return -1;
|
|
|
|
}
|
2015-02-02 16:19:07 +03:00
|
|
|
|
2015-02-03 02:50:26 +03:00
|
|
|
rpc_client_virtual_connection_transition_to_state(rpc,
|
|
|
|
rpc->VirtualConnection, VIRTUAL_CONNECTION_STATE_OPENED);
|
2015-02-02 16:19:07 +03:00
|
|
|
|
2015-02-03 02:50:26 +03:00
|
|
|
rpc_client_transition_to_state(rpc, RPC_CLIENT_STATE_ESTABLISHED);
|
2015-02-02 16:19:07 +03:00
|
|
|
|
|
|
|
if (rpc_send_bind_pdu(rpc) < 0)
|
|
|
|
{
|
|
|
|
WLog_ERR(TAG, "rpc_send_bind_pdu failure");
|
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
|
2015-02-03 02:50:26 +03:00
|
|
|
rpc_client_transition_to_state(rpc, RPC_CLIENT_STATE_WAIT_SECURE_BIND_ACK);
|
|
|
|
|
|
|
|
status = 1;
|
2015-02-02 16:19:07 +03:00
|
|
|
|
|
|
|
break;
|
|
|
|
|
|
|
|
case VIRTUAL_CONNECTION_STATE_OPENED:
|
|
|
|
break;
|
|
|
|
|
|
|
|
case VIRTUAL_CONNECTION_STATE_FINAL:
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
}
|
2015-02-03 02:50:26 +03:00
|
|
|
else if (rpc->State < RPC_CLIENT_STATE_CONTEXT_NEGOTIATED)
|
2015-02-02 16:19:07 +03:00
|
|
|
{
|
2015-02-03 02:50:26 +03:00
|
|
|
if (rpc->State == RPC_CLIENT_STATE_WAIT_SECURE_BIND_ACK)
|
2015-02-02 16:19:07 +03:00
|
|
|
{
|
2015-02-03 02:50:26 +03:00
|
|
|
if (pdu->Type == PTYPE_BIND_ACK)
|
|
|
|
{
|
|
|
|
if (rpc_recv_bind_ack_pdu(rpc, Stream_Buffer(pdu->s), Stream_Length(pdu->s)) <= 0)
|
|
|
|
{
|
|
|
|
WLog_ERR(TAG, "rpc_recv_bind_ack_pdu failure");
|
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
else
|
2015-02-02 16:19:07 +03:00
|
|
|
{
|
2015-02-03 02:50:26 +03:00
|
|
|
WLog_ERR(TAG, "RPC_CLIENT_STATE_WAIT_SECURE_BIND_ACK unexpected pdu type: 0x%04X", pdu->Type);
|
2015-02-02 16:19:07 +03:00
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
|
2015-02-04 01:17:17 +03:00
|
|
|
if (rpc_send_rpc_auth_3_pdu(rpc) < 0)
|
2015-02-02 16:19:07 +03:00
|
|
|
{
|
|
|
|
WLog_ERR(TAG, "rpc_secure_bind: error sending rpc_auth_3 pdu!");
|
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
|
2015-02-03 02:50:26 +03:00
|
|
|
rpc_client_transition_to_state(rpc, RPC_CLIENT_STATE_CONTEXT_NEGOTIATED);
|
2015-02-02 16:19:07 +03:00
|
|
|
|
|
|
|
if (!TsProxyCreateTunnel(tsg, NULL, NULL, NULL, NULL))
|
|
|
|
{
|
|
|
|
WLog_ERR(TAG, "TsProxyCreateTunnel failure");
|
|
|
|
tsg->state = TSG_STATE_FINAL;
|
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
|
2015-02-03 02:50:26 +03:00
|
|
|
tsg_transition_to_state(tsg, TSG_STATE_INITIAL);
|
|
|
|
|
|
|
|
status = 1;
|
2015-02-02 16:19:07 +03:00
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
WLog_ERR(TAG, "rpc_client_recv_pdu: invalid rpc->State: %d", rpc->State);
|
|
|
|
}
|
|
|
|
}
|
2015-02-03 02:50:26 +03:00
|
|
|
else if (rpc->State >= RPC_CLIENT_STATE_CONTEXT_NEGOTIATED)
|
2015-02-02 16:19:07 +03:00
|
|
|
{
|
2015-02-03 02:50:26 +03:00
|
|
|
if (tsg->state != TSG_STATE_PIPE_CREATED)
|
|
|
|
{
|
|
|
|
status = tsg_recv_pdu(tsg, pdu);
|
|
|
|
}
|
2015-02-02 16:19:07 +03:00
|
|
|
}
|
|
|
|
|
2015-02-03 02:50:26 +03:00
|
|
|
return status;
|
2015-02-01 00:56:25 +03:00
|
|
|
}
|
|
|
|
|
2015-02-02 04:47:43 +03:00
|
|
|
int rpc_client_recv_fragment(rdpRpc* rpc, wStream* fragment)
|
2012-12-12 04:17:57 +04:00
|
|
|
{
|
2014-08-18 21:34:47 +04:00
|
|
|
BYTE* buffer;
|
2015-02-01 23:58:32 +03:00
|
|
|
RPC_PDU* pdu;
|
2012-12-12 04:17:57 +04:00
|
|
|
UINT32 StubOffset;
|
|
|
|
UINT32 StubLength;
|
2015-02-01 23:58:32 +03:00
|
|
|
RpcClientCall* call;
|
2014-08-18 21:34:47 +04:00
|
|
|
rpcconn_hdr_t* header;
|
2012-12-12 04:17:57 +04:00
|
|
|
|
2015-02-02 16:19:07 +03:00
|
|
|
pdu = rpc->client->pdu;
|
2014-08-18 21:34:47 +04:00
|
|
|
buffer = (BYTE*) Stream_Buffer(fragment);
|
|
|
|
header = (rpcconn_hdr_t*) Stream_Buffer(fragment);
|
2012-12-12 08:34:51 +04:00
|
|
|
|
2015-02-01 23:58:32 +03:00
|
|
|
if (header->common.ptype == PTYPE_RESPONSE)
|
2012-12-12 04:17:57 +04:00
|
|
|
{
|
2015-02-01 23:58:32 +03:00
|
|
|
rpc->VirtualConnection->DefaultOutChannel->BytesReceived += header->common.frag_length;
|
|
|
|
rpc->VirtualConnection->DefaultOutChannel->ReceiverAvailableWindow -= header->common.frag_length;
|
2012-12-12 04:17:57 +04:00
|
|
|
|
2015-02-01 23:58:32 +03:00
|
|
|
if (rpc->VirtualConnection->DefaultOutChannel->ReceiverAvailableWindow < (rpc->ReceiveWindow / 2))
|
|
|
|
{
|
|
|
|
rts_send_flow_control_ack_pdu(rpc);
|
|
|
|
}
|
2014-08-19 20:26:39 +04:00
|
|
|
|
2015-02-01 23:58:32 +03:00
|
|
|
if (!rpc_get_stub_data_info(rpc, buffer, &StubOffset, &StubLength))
|
|
|
|
{
|
|
|
|
WLog_ERR(TAG, "expected stub");
|
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (StubLength == 4)
|
|
|
|
{
|
|
|
|
/* received a disconnect request from the server? */
|
|
|
|
if ((header->common.call_id == rpc->PipeCallId) && (header->common.pfc_flags & PFC_LAST_FRAG))
|
2014-05-21 19:32:14 +04:00
|
|
|
{
|
2015-02-01 23:58:32 +03:00
|
|
|
TerminateEventArgs e;
|
|
|
|
|
|
|
|
rpc->result = *((UINT32*) &buffer[StubOffset]);
|
|
|
|
|
|
|
|
rpc->context->rdp->disconnect = TRUE;
|
|
|
|
rpc->transport->tsg->state = TSG_STATE_TUNNEL_CLOSE_PENDING;
|
|
|
|
EventArgsInit(&e, "freerdp");
|
|
|
|
e.code = 0;
|
|
|
|
PubSub_OnTerminate(rpc->context->pubSub, rpc->context, &e);
|
2014-05-21 19:32:14 +04:00
|
|
|
}
|
2014-08-18 19:22:43 +04:00
|
|
|
|
2014-05-21 19:32:14 +04:00
|
|
|
return 0;
|
2015-02-01 23:58:32 +03:00
|
|
|
}
|
2015-01-13 21:50:46 +03:00
|
|
|
|
2015-02-01 23:58:32 +03:00
|
|
|
if (rpc->StubFragCount == 0)
|
|
|
|
rpc->StubCallId = header->common.call_id;
|
2015-01-13 21:50:46 +03:00
|
|
|
|
2015-02-01 23:58:32 +03:00
|
|
|
if (rpc->StubCallId != header->common.call_id)
|
|
|
|
{
|
|
|
|
WLog_ERR(TAG, "invalid call_id: actual: %d, expected: %d, frag_count: %d",
|
|
|
|
rpc->StubCallId, header->common.call_id, rpc->StubFragCount);
|
|
|
|
}
|
2015-01-13 21:50:46 +03:00
|
|
|
|
2015-02-01 23:58:32 +03:00
|
|
|
call = rpc_client_call_find_by_id(rpc, rpc->StubCallId);
|
|
|
|
|
|
|
|
if (!call)
|
2014-05-21 19:32:14 +04:00
|
|
|
return -1;
|
2012-12-12 08:34:51 +04:00
|
|
|
|
2015-02-01 23:58:32 +03:00
|
|
|
if (call->OpNum != TsProxySetupReceivePipeOpnum)
|
|
|
|
{
|
|
|
|
Stream_EnsureCapacity(pdu->s, header->response.alloc_hint);
|
|
|
|
Stream_Write(pdu->s, &buffer[StubOffset], StubLength);
|
|
|
|
rpc->StubFragCount++;
|
2012-12-12 08:34:51 +04:00
|
|
|
|
2015-02-01 23:58:32 +03:00
|
|
|
if (header->response.alloc_hint == StubLength)
|
|
|
|
{
|
|
|
|
pdu->Flags = RPC_PDU_FLAG_STUB;
|
|
|
|
pdu->Type = PTYPE_RESPONSE;
|
|
|
|
pdu->CallId = rpc->StubCallId;
|
|
|
|
Stream_SealLength(pdu->s);
|
2015-02-02 04:47:43 +03:00
|
|
|
rpc_client_recv_pdu(rpc, pdu);
|
2015-02-02 16:19:07 +03:00
|
|
|
rpc_pdu_reset(pdu);
|
2015-02-01 23:58:32 +03:00
|
|
|
rpc->StubFragCount = 0;
|
|
|
|
rpc->StubCallId = 0;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
else
|
2013-06-19 00:55:23 +04:00
|
|
|
{
|
2015-02-01 23:58:32 +03:00
|
|
|
rpc_client_receive_pipe_write(rpc, &buffer[StubOffset], (size_t) StubLength);
|
|
|
|
rpc->StubFragCount++;
|
|
|
|
|
|
|
|
if (header->response.alloc_hint == StubLength)
|
|
|
|
{
|
|
|
|
rpc->StubFragCount = 0;
|
|
|
|
rpc->StubCallId = 0;
|
|
|
|
}
|
2013-06-19 00:55:23 +04:00
|
|
|
}
|
|
|
|
|
2015-02-02 16:19:07 +03:00
|
|
|
return 1;
|
2012-12-12 08:34:51 +04:00
|
|
|
}
|
2015-02-01 23:58:32 +03:00
|
|
|
else if (header->common.ptype == PTYPE_RTS)
|
2012-12-12 04:17:57 +04:00
|
|
|
{
|
2015-02-01 23:58:32 +03:00
|
|
|
if (rpc->State < RPC_CLIENT_STATE_CONTEXT_NEGOTIATED)
|
|
|
|
{
|
|
|
|
pdu->Flags = 0;
|
|
|
|
pdu->Type = header->common.ptype;
|
|
|
|
pdu->CallId = header->common.call_id;
|
|
|
|
Stream_EnsureCapacity(pdu->s, Stream_Length(fragment));
|
|
|
|
Stream_Write(pdu->s, buffer, Stream_Length(fragment));
|
|
|
|
Stream_SealLength(pdu->s);
|
2015-02-02 04:47:43 +03:00
|
|
|
rpc_client_recv_pdu(rpc, pdu);
|
2015-02-02 16:19:07 +03:00
|
|
|
rpc_pdu_reset(pdu);
|
2015-02-01 23:58:32 +03:00
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
if (rpc->VirtualConnection->State < VIRTUAL_CONNECTION_STATE_OPENED)
|
|
|
|
WLog_ERR(TAG, "warning: unhandled RTS PDU");
|
2012-12-12 04:17:57 +04:00
|
|
|
|
2015-02-01 23:58:32 +03:00
|
|
|
WLog_DBG(TAG, "Receiving Out-of-Sequence RTS PDU");
|
|
|
|
rts_recv_out_of_sequence_pdu(rpc, buffer, header->common.frag_length);
|
|
|
|
}
|
2015-02-02 16:19:07 +03:00
|
|
|
|
|
|
|
return 1;
|
2012-12-12 04:17:57 +04:00
|
|
|
}
|
2015-02-01 23:58:32 +03:00
|
|
|
else if (header->common.ptype == PTYPE_BIND_ACK)
|
2012-12-12 04:17:57 +04:00
|
|
|
{
|
2015-02-01 23:58:32 +03:00
|
|
|
pdu->Flags = 0;
|
|
|
|
pdu->Type = header->common.ptype;
|
|
|
|
pdu->CallId = header->common.call_id;
|
|
|
|
Stream_EnsureCapacity(pdu->s, Stream_Length(fragment));
|
|
|
|
Stream_Write(pdu->s, buffer, Stream_Length(fragment));
|
|
|
|
Stream_SealLength(pdu->s);
|
2015-02-02 04:47:43 +03:00
|
|
|
rpc_client_recv_pdu(rpc, pdu);
|
2015-02-02 16:19:07 +03:00
|
|
|
rpc_pdu_reset(pdu);
|
|
|
|
|
|
|
|
return 1;
|
2012-12-12 04:17:57 +04:00
|
|
|
}
|
2015-02-01 23:58:32 +03:00
|
|
|
else if (header->common.ptype == PTYPE_FAULT)
|
|
|
|
{
|
|
|
|
rpc_recv_fault_pdu(header);
|
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
WLog_ERR(TAG, "unexpected RPC PDU type 0x%04X", header->common.ptype);
|
2015-02-02 16:19:07 +03:00
|
|
|
return -1;
|
2015-02-01 23:58:32 +03:00
|
|
|
}
|
2012-12-12 04:17:57 +04:00
|
|
|
|
2015-02-02 16:19:07 +03:00
|
|
|
return 1;
|
2012-12-12 04:17:57 +04:00
|
|
|
}
|
|
|
|
|
2015-02-02 04:47:43 +03:00
|
|
|
int rpc_client_recv(rdpRpc* rpc)
|
2012-12-11 00:43:07 +04:00
|
|
|
{
|
|
|
|
int status = -1;
|
2015-02-03 02:50:26 +03:00
|
|
|
wStream* fragment;
|
2014-08-18 21:34:47 +04:00
|
|
|
rpcconn_common_hdr_t* header;
|
2012-12-11 00:43:07 +04:00
|
|
|
|
2015-02-03 02:50:26 +03:00
|
|
|
fragment = rpc->client->ReceiveFragment;
|
|
|
|
|
2014-05-21 19:32:14 +04:00
|
|
|
while (1)
|
2012-12-11 00:43:07 +04:00
|
|
|
{
|
2015-02-03 02:50:26 +03:00
|
|
|
while (Stream_GetPosition(fragment) < RPC_COMMON_FIELDS_LENGTH)
|
2012-12-11 00:43:07 +04:00
|
|
|
{
|
2015-02-03 02:50:26 +03:00
|
|
|
status = rpc_out_read(rpc, Stream_Pointer(fragment),
|
|
|
|
RPC_COMMON_FIELDS_LENGTH - Stream_GetPosition(fragment));
|
2014-05-21 19:32:14 +04:00
|
|
|
|
|
|
|
if (status < 0)
|
|
|
|
return -1;
|
|
|
|
|
|
|
|
if (!status)
|
|
|
|
return 0;
|
|
|
|
|
2015-02-03 02:50:26 +03:00
|
|
|
Stream_Seek(fragment, status);
|
2012-12-11 00:43:07 +04:00
|
|
|
}
|
|
|
|
|
2015-02-03 02:50:26 +03:00
|
|
|
if (Stream_GetPosition(fragment) < RPC_COMMON_FIELDS_LENGTH)
|
2014-05-21 19:32:14 +04:00
|
|
|
return status;
|
|
|
|
|
2015-02-03 02:50:26 +03:00
|
|
|
header = (rpcconn_common_hdr_t*) Stream_Buffer(fragment);
|
2012-12-11 00:43:07 +04:00
|
|
|
|
|
|
|
if (header->frag_length > rpc->max_recv_frag)
|
|
|
|
{
|
2015-02-03 02:50:26 +03:00
|
|
|
WLog_ERR(TAG, "rpc_client_recv: invalid fragment size: %d (max: %d)",
|
2014-09-12 16:36:29 +04:00
|
|
|
header->frag_length, rpc->max_recv_frag);
|
2015-02-03 02:50:26 +03:00
|
|
|
winpr_HexDump(TAG, WLOG_ERROR, Stream_Buffer(fragment), Stream_GetPosition(fragment));
|
2012-12-11 00:43:07 +04:00
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
|
2015-02-03 02:50:26 +03:00
|
|
|
while (Stream_GetPosition(fragment) < header->frag_length)
|
2012-12-11 00:43:07 +04:00
|
|
|
{
|
2015-02-03 02:50:26 +03:00
|
|
|
status = rpc_out_read(rpc, Stream_Pointer(fragment),
|
|
|
|
header->frag_length - Stream_GetPosition(fragment));
|
2012-12-11 00:43:07 +04:00
|
|
|
|
|
|
|
if (status < 0)
|
|
|
|
{
|
2014-09-12 16:36:29 +04:00
|
|
|
WLog_ERR(TAG, "error reading fragment body");
|
2012-12-11 00:43:07 +04:00
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
|
2014-05-21 19:32:14 +04:00
|
|
|
if (!status)
|
|
|
|
return 0;
|
|
|
|
|
2015-02-03 02:50:26 +03:00
|
|
|
Stream_Seek(fragment, status);
|
2012-12-11 00:43:07 +04:00
|
|
|
}
|
|
|
|
|
2014-05-21 19:32:14 +04:00
|
|
|
if (status < 0)
|
|
|
|
return -1;
|
2012-12-11 00:43:07 +04:00
|
|
|
|
2015-02-03 02:50:26 +03:00
|
|
|
if (Stream_GetPosition(fragment) >= header->frag_length)
|
2014-05-21 19:32:14 +04:00
|
|
|
{
|
|
|
|
/* complete fragment received */
|
2015-02-01 00:56:25 +03:00
|
|
|
|
2015-02-03 02:50:26 +03:00
|
|
|
Stream_SealLength(fragment);
|
|
|
|
Stream_SetPosition(fragment, 0);
|
2012-12-12 04:17:57 +04:00
|
|
|
|
2015-02-03 02:50:26 +03:00
|
|
|
status = rpc_client_recv_fragment(rpc, fragment);
|
2015-02-01 00:56:25 +03:00
|
|
|
|
|
|
|
if (status < 0)
|
|
|
|
return status;
|
|
|
|
|
2015-02-03 02:50:26 +03:00
|
|
|
Stream_SetPosition(fragment, 0);
|
2014-05-21 19:32:14 +04:00
|
|
|
}
|
2012-12-11 00:43:07 +04:00
|
|
|
}
|
|
|
|
|
2015-02-02 16:19:07 +03:00
|
|
|
return 1;
|
2012-12-11 00:43:07 +04:00
|
|
|
}
|
|
|
|
|
2015-02-04 01:17:17 +03:00
|
|
|
int rpc_client_out_channel_recv(rdpRpc* rpc)
|
|
|
|
{
|
|
|
|
int status = -1;
|
|
|
|
HttpResponse* response;
|
2015-02-04 04:39:47 +03:00
|
|
|
RpcInChannel* inChannel;
|
2015-02-04 01:17:17 +03:00
|
|
|
RpcOutChannel* outChannel;
|
|
|
|
|
2015-02-04 04:39:47 +03:00
|
|
|
inChannel = rpc->VirtualConnection->DefaultInChannel;
|
2015-02-04 01:17:17 +03:00
|
|
|
outChannel = rpc->VirtualConnection->DefaultOutChannel;
|
|
|
|
|
2015-02-04 04:39:47 +03:00
|
|
|
if (outChannel->State < CLIENT_OUT_CHANNEL_STATE_OPENED)
|
2015-02-04 01:17:17 +03:00
|
|
|
{
|
2015-02-11 22:27:29 +03:00
|
|
|
response = http_response_recv(outChannel->tls);
|
2015-02-04 01:17:17 +03:00
|
|
|
|
|
|
|
if (!response)
|
|
|
|
return -1;
|
|
|
|
|
|
|
|
if (outChannel->State == CLIENT_OUT_CHANNEL_STATE_SECURITY)
|
|
|
|
{
|
|
|
|
/* Receive OUT Channel Response */
|
|
|
|
|
|
|
|
if (rpc_ncacn_http_recv_out_channel_response(rpc, response) < 0)
|
|
|
|
{
|
|
|
|
WLog_ERR(TAG, "rpc_ncacn_http_recv_out_channel_response failure");
|
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
|
|
|
|
/* Send OUT Channel Request */
|
|
|
|
|
|
|
|
if (rpc_ncacn_http_send_out_channel_request(rpc) < 0)
|
|
|
|
{
|
|
|
|
WLog_ERR(TAG, "rpc_ncacn_http_send_out_channel_request failure");
|
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
|
|
|
|
rpc_ncacn_http_ntlm_uninit(rpc, TSG_CHANNEL_OUT);
|
|
|
|
|
|
|
|
rpc_client_out_channel_transition_to_state(outChannel,
|
|
|
|
CLIENT_OUT_CHANNEL_STATE_NEGOTIATED);
|
|
|
|
|
|
|
|
/* Send CONN/A1 PDU over OUT channel */
|
|
|
|
|
|
|
|
if (rts_send_CONN_A1_pdu(rpc) < 0)
|
|
|
|
{
|
|
|
|
WLog_ERR(TAG, "rpc_send_CONN_A1_pdu error!");
|
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
|
|
|
|
rpc_client_out_channel_transition_to_state(outChannel,
|
|
|
|
CLIENT_OUT_CHANNEL_STATE_OPENED);
|
2015-02-04 04:39:47 +03:00
|
|
|
|
2015-02-04 19:18:27 +03:00
|
|
|
if (inChannel->State == CLIENT_IN_CHANNEL_STATE_OPENED)
|
2015-02-04 04:39:47 +03:00
|
|
|
{
|
|
|
|
rpc_client_virtual_connection_transition_to_state(rpc,
|
|
|
|
rpc->VirtualConnection, VIRTUAL_CONNECTION_STATE_OUT_CHANNEL_WAIT);
|
|
|
|
}
|
|
|
|
|
|
|
|
status = 1;
|
2015-02-04 01:17:17 +03:00
|
|
|
}
|
|
|
|
|
|
|
|
http_response_free(response);
|
|
|
|
}
|
|
|
|
else if (rpc->VirtualConnection->State == VIRTUAL_CONNECTION_STATE_OUT_CHANNEL_WAIT)
|
|
|
|
{
|
|
|
|
/* Receive OUT channel response */
|
|
|
|
|
2015-02-11 22:27:29 +03:00
|
|
|
response = http_response_recv(outChannel->tls);
|
2015-02-04 01:17:17 +03:00
|
|
|
|
|
|
|
if (!response)
|
|
|
|
return -1;
|
|
|
|
|
|
|
|
if (response->StatusCode != HTTP_STATUS_OK)
|
|
|
|
{
|
|
|
|
WLog_ERR(TAG, "error! Status Code: %d", response->StatusCode);
|
|
|
|
http_response_print(response);
|
|
|
|
http_response_free(response);
|
|
|
|
|
|
|
|
if (response->StatusCode == HTTP_STATUS_DENIED)
|
|
|
|
{
|
|
|
|
if (!connectErrorCode)
|
|
|
|
connectErrorCode = AUTHENTICATIONERROR;
|
|
|
|
|
|
|
|
if (!freerdp_get_last_error(rpc->context))
|
|
|
|
{
|
|
|
|
freerdp_set_last_error(rpc->context, FREERDP_ERROR_AUTHENTICATION_FAILED);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
|
|
|
|
http_response_free(response);
|
|
|
|
|
|
|
|
rpc_client_virtual_connection_transition_to_state(rpc,
|
|
|
|
rpc->VirtualConnection, VIRTUAL_CONNECTION_STATE_WAIT_A3W);
|
2015-02-04 04:39:47 +03:00
|
|
|
|
|
|
|
status = 1;
|
2015-02-04 01:17:17 +03:00
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
status = rpc_client_recv(rpc);
|
|
|
|
}
|
|
|
|
|
|
|
|
return status;
|
|
|
|
}
|
|
|
|
|
|
|
|
int rpc_client_in_channel_recv(rdpRpc* rpc)
|
|
|
|
{
|
|
|
|
int status = -1;
|
|
|
|
HttpResponse* response;
|
|
|
|
RpcInChannel* inChannel;
|
2015-02-04 04:39:47 +03:00
|
|
|
RpcOutChannel* outChannel;
|
2015-02-04 01:17:17 +03:00
|
|
|
HANDLE InChannelEvent = NULL;
|
|
|
|
|
|
|
|
inChannel = rpc->VirtualConnection->DefaultInChannel;
|
2015-02-04 04:39:47 +03:00
|
|
|
outChannel = rpc->VirtualConnection->DefaultOutChannel;
|
|
|
|
|
2015-02-11 22:27:29 +03:00
|
|
|
BIO_get_event(inChannel->tls->bio, &InChannelEvent);
|
2015-02-04 01:17:17 +03:00
|
|
|
|
|
|
|
if (WaitForSingleObject(InChannelEvent, 0) != WAIT_OBJECT_0)
|
|
|
|
return 1;
|
|
|
|
|
|
|
|
if (inChannel->State < CLIENT_IN_CHANNEL_STATE_OPENED)
|
|
|
|
{
|
2015-02-11 22:27:29 +03:00
|
|
|
response = http_response_recv(inChannel->tls);
|
2015-02-04 01:17:17 +03:00
|
|
|
|
|
|
|
if (!response)
|
|
|
|
return -1;
|
|
|
|
|
|
|
|
if (inChannel->State == CLIENT_IN_CHANNEL_STATE_SECURITY)
|
|
|
|
{
|
|
|
|
if (rpc_ncacn_http_recv_in_channel_response(rpc, response) < 0)
|
|
|
|
{
|
|
|
|
WLog_ERR(TAG, "rpc_ncacn_http_recv_in_channel_response failure");
|
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
|
|
|
|
/* Send IN Channel Request */
|
|
|
|
|
|
|
|
if (rpc_ncacn_http_send_in_channel_request(rpc) < 0)
|
|
|
|
{
|
|
|
|
WLog_ERR(TAG, "rpc_ncacn_http_send_in_channel_request failure");
|
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
|
|
|
|
rpc_ncacn_http_ntlm_uninit(rpc, TSG_CHANNEL_IN);
|
|
|
|
|
|
|
|
rpc_client_in_channel_transition_to_state(inChannel,
|
|
|
|
CLIENT_IN_CHANNEL_STATE_NEGOTIATED);
|
|
|
|
|
|
|
|
/* Send CONN/B1 PDU over IN channel */
|
|
|
|
|
|
|
|
if (rts_send_CONN_B1_pdu(rpc) < 0)
|
|
|
|
{
|
|
|
|
WLog_ERR(TAG, "rpc_send_CONN_B1_pdu error!");
|
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
|
|
|
|
rpc_client_in_channel_transition_to_state(inChannel,
|
|
|
|
CLIENT_IN_CHANNEL_STATE_OPENED);
|
|
|
|
|
2015-02-04 19:18:27 +03:00
|
|
|
if (outChannel->State == CLIENT_OUT_CHANNEL_STATE_OPENED)
|
2015-02-04 04:39:47 +03:00
|
|
|
{
|
|
|
|
rpc_client_virtual_connection_transition_to_state(rpc,
|
|
|
|
rpc->VirtualConnection, VIRTUAL_CONNECTION_STATE_OUT_CHANNEL_WAIT);
|
|
|
|
}
|
|
|
|
|
2015-02-04 01:17:17 +03:00
|
|
|
status = 1;
|
|
|
|
}
|
2015-02-04 19:18:27 +03:00
|
|
|
|
|
|
|
http_response_free(response);
|
2015-02-04 01:17:17 +03:00
|
|
|
}
|
|
|
|
|
|
|
|
return status;
|
|
|
|
}
|
|
|
|
|
2012-12-08 06:09:55 +04:00
|
|
|
/**
|
|
|
|
* [MS-RPCE] Client Call:
|
|
|
|
* http://msdn.microsoft.com/en-us/library/gg593159/
|
|
|
|
*/
|
|
|
|
|
2014-08-18 21:34:47 +04:00
|
|
|
RpcClientCall* rpc_client_call_find_by_id(rdpRpc* rpc, UINT32 CallId)
|
2012-12-08 06:09:55 +04:00
|
|
|
{
|
|
|
|
int index;
|
|
|
|
int count;
|
2015-01-28 22:54:03 +03:00
|
|
|
RpcClientCall* clientCall = NULL;
|
|
|
|
|
2012-12-12 08:34:51 +04:00
|
|
|
ArrayList_Lock(rpc->client->ClientCallList);
|
2015-01-28 22:54:03 +03:00
|
|
|
|
2012-12-12 08:34:51 +04:00
|
|
|
count = ArrayList_Count(rpc->client->ClientCallList);
|
2012-12-08 06:09:55 +04:00
|
|
|
|
|
|
|
for (index = 0; index < count; index++)
|
|
|
|
{
|
2014-08-18 21:34:47 +04:00
|
|
|
clientCall = (RpcClientCall*) ArrayList_GetItem(rpc->client->ClientCallList, index);
|
2012-12-08 06:09:55 +04:00
|
|
|
|
2012-12-12 09:49:15 +04:00
|
|
|
if (clientCall->CallId == CallId)
|
2012-12-08 06:09:55 +04:00
|
|
|
break;
|
|
|
|
}
|
|
|
|
|
2012-12-12 08:34:51 +04:00
|
|
|
ArrayList_Unlock(rpc->client->ClientCallList);
|
2015-01-28 22:54:03 +03:00
|
|
|
|
2012-12-12 09:49:15 +04:00
|
|
|
return clientCall;
|
2012-12-08 06:09:55 +04:00
|
|
|
}
|
|
|
|
|
2014-08-18 21:34:47 +04:00
|
|
|
RpcClientCall* rpc_client_call_new(UINT32 CallId, UINT32 OpNum)
|
2012-12-08 06:09:55 +04:00
|
|
|
{
|
2014-08-18 21:34:47 +04:00
|
|
|
RpcClientCall* clientCall;
|
2015-01-28 22:54:03 +03:00
|
|
|
|
|
|
|
clientCall = (RpcClientCall*) calloc(1, sizeof(RpcClientCall));
|
2012-12-08 06:09:55 +04:00
|
|
|
|
2014-05-21 19:32:14 +04:00
|
|
|
if (!clientCall)
|
|
|
|
return NULL;
|
2012-12-08 06:09:55 +04:00
|
|
|
|
2014-05-21 19:32:14 +04:00
|
|
|
clientCall->CallId = CallId;
|
|
|
|
clientCall->OpNum = OpNum;
|
|
|
|
clientCall->State = RPC_CLIENT_CALL_STATE_SEND_PDUS;
|
2015-01-28 22:54:03 +03:00
|
|
|
|
2012-12-12 09:49:15 +04:00
|
|
|
return clientCall;
|
2012-12-08 06:09:55 +04:00
|
|
|
}
|
|
|
|
|
2014-08-18 21:34:47 +04:00
|
|
|
void rpc_client_call_free(RpcClientCall* clientCall)
|
2012-12-08 06:09:55 +04:00
|
|
|
{
|
2012-12-12 09:49:15 +04:00
|
|
|
free(clientCall);
|
2012-12-08 06:09:55 +04:00
|
|
|
}
|
|
|
|
|
2015-02-02 04:47:43 +03:00
|
|
|
int rpc_send_pdu(rdpRpc* rpc, BYTE* buffer, UINT32 length)
|
2012-11-28 21:47:04 +04:00
|
|
|
{
|
2013-12-21 03:22:29 +04:00
|
|
|
int status;
|
2015-02-02 04:47:43 +03:00
|
|
|
RpcInChannel* inChannel;
|
2014-08-18 21:34:47 +04:00
|
|
|
RpcClientCall* clientCall;
|
|
|
|
rpcconn_common_hdr_t* header;
|
2014-12-15 17:42:04 +03:00
|
|
|
|
2015-02-02 04:47:43 +03:00
|
|
|
inChannel = rpc->VirtualConnection->DefaultInChannel;
|
2012-11-28 21:47:04 +04:00
|
|
|
|
2015-02-02 04:47:43 +03:00
|
|
|
status = rpc_in_write(rpc, buffer, length);
|
2012-11-28 22:38:01 +04:00
|
|
|
|
2015-02-03 22:44:31 +03:00
|
|
|
if (status <= 0)
|
|
|
|
return -1;
|
|
|
|
|
2015-02-02 04:47:43 +03:00
|
|
|
header = (rpcconn_common_hdr_t*) buffer;
|
2012-12-12 09:49:15 +04:00
|
|
|
clientCall = rpc_client_call_find_by_id(rpc, header->call_id);
|
|
|
|
clientCall->State = RPC_CLIENT_CALL_STATE_DISPATCHED;
|
2012-11-28 21:47:04 +04:00
|
|
|
|
|
|
|
/*
|
|
|
|
* This protocol specifies that only RPC PDUs are subject to the flow control abstract
|
|
|
|
* data model. RTS PDUs and the HTTP request and response headers are not subject to flow control.
|
|
|
|
* Implementations of this protocol MUST NOT include them when computing any of the variables
|
|
|
|
* specified by this abstract data model.
|
|
|
|
*/
|
2012-12-13 07:03:40 +04:00
|
|
|
|
|
|
|
if (header->ptype == PTYPE_REQUEST)
|
|
|
|
{
|
2014-05-21 19:32:14 +04:00
|
|
|
inChannel->BytesSent += status;
|
|
|
|
inChannel->SenderAvailableWindow -= status;
|
2012-12-13 07:03:40 +04:00
|
|
|
}
|
2012-11-28 21:47:04 +04:00
|
|
|
|
|
|
|
return status;
|
|
|
|
}
|
|
|
|
|
2014-08-18 21:34:47 +04:00
|
|
|
int rpc_client_new(rdpRpc* rpc)
|
2012-11-28 21:47:04 +04:00
|
|
|
{
|
2014-12-11 19:25:34 +03:00
|
|
|
RpcClient* client;
|
2014-08-18 19:22:43 +04:00
|
|
|
|
2014-12-11 19:25:34 +03:00
|
|
|
client = (RpcClient*) calloc(1, sizeof(RpcClient));
|
2012-11-28 21:47:04 +04:00
|
|
|
|
2014-12-11 19:25:34 +03:00
|
|
|
rpc->client = client;
|
2014-08-18 19:22:43 +04:00
|
|
|
|
2014-12-11 19:25:34 +03:00
|
|
|
if (!client)
|
2014-05-21 19:32:14 +04:00
|
|
|
return -1;
|
2012-11-28 21:47:04 +04:00
|
|
|
|
2015-02-02 16:19:07 +03:00
|
|
|
client->pdu = rpc_pdu_new();
|
2015-02-01 00:56:25 +03:00
|
|
|
|
2015-02-02 16:19:07 +03:00
|
|
|
if (!client->pdu)
|
2014-05-21 19:32:14 +04:00
|
|
|
return -1;
|
2012-12-11 03:56:53 +04:00
|
|
|
|
2015-02-01 21:09:28 +03:00
|
|
|
client->ReceiveFragment = Stream_New(NULL, rpc->max_recv_frag);
|
|
|
|
|
|
|
|
if (!client->ReceiveFragment)
|
|
|
|
return -1;
|
|
|
|
|
|
|
|
client->PipeEvent = CreateEvent(NULL, TRUE, FALSE, NULL);
|
|
|
|
|
|
|
|
if (!client->PipeEvent)
|
|
|
|
return -1;
|
|
|
|
|
|
|
|
if (!ringbuffer_init(&(client->ReceivePipe), 4096))
|
|
|
|
return -1;
|
|
|
|
|
|
|
|
if (!InitializeCriticalSectionAndSpinCount(&(client->PipeLock), 4000))
|
|
|
|
return -1;
|
2014-08-18 19:22:43 +04:00
|
|
|
|
2014-05-21 19:32:14 +04:00
|
|
|
client->ClientCallList = ArrayList_New(TRUE);
|
2014-08-18 19:22:43 +04:00
|
|
|
|
2014-05-21 19:32:14 +04:00
|
|
|
if (!client->ClientCallList)
|
|
|
|
return -1;
|
2014-08-18 19:22:43 +04:00
|
|
|
|
2014-05-21 19:32:14 +04:00
|
|
|
ArrayList_Object(client->ClientCallList)->fnObjectFree = (OBJECT_FREE_FN) rpc_client_call_free;
|
2012-11-29 05:30:03 +04:00
|
|
|
|
2015-02-02 16:19:07 +03:00
|
|
|
return 1;
|
2012-12-12 08:34:51 +04:00
|
|
|
}
|
|
|
|
|
2015-02-02 16:19:07 +03:00
|
|
|
void rpc_client_free(rdpRpc* rpc)
|
2012-12-12 08:34:51 +04:00
|
|
|
{
|
2014-12-15 17:42:04 +03:00
|
|
|
RpcClient* client = rpc->client;
|
2012-12-12 09:49:15 +04:00
|
|
|
|
2014-05-21 19:32:14 +04:00
|
|
|
if (!client)
|
2015-02-02 16:19:07 +03:00
|
|
|
return;
|
2014-05-21 19:32:14 +04:00
|
|
|
|
2015-02-01 21:09:28 +03:00
|
|
|
if (client->ReceiveFragment)
|
|
|
|
Stream_Free(client->ReceiveFragment, TRUE);
|
|
|
|
|
|
|
|
if (client->PipeEvent)
|
|
|
|
CloseHandle(client->PipeEvent);
|
|
|
|
|
|
|
|
ringbuffer_destroy(&(client->ReceivePipe));
|
|
|
|
|
|
|
|
DeleteCriticalSection(&(client->PipeLock));
|
2012-12-13 07:03:40 +04:00
|
|
|
|
2014-05-21 19:32:14 +04:00
|
|
|
if (client->pdu)
|
|
|
|
rpc_pdu_free(client->pdu);
|
2012-12-13 05:02:56 +04:00
|
|
|
|
2014-05-21 19:32:14 +04:00
|
|
|
if (client->ClientCallList)
|
2012-12-13 05:02:56 +04:00
|
|
|
ArrayList_Free(client->ClientCallList);
|
2012-12-12 09:49:15 +04:00
|
|
|
|
2014-05-21 19:32:14 +04:00
|
|
|
free(client);
|
2014-12-15 17:42:04 +03:00
|
|
|
rpc->client = NULL;
|
2012-12-12 08:34:51 +04:00
|
|
|
}
|