|
Packit Service |
fa4841 |
/**
|
|
Packit Service |
fa4841 |
* FreeRDP: A Remote Desktop Protocol Implementation
|
|
Packit Service |
fa4841 |
* RPC Client
|
|
Packit Service |
fa4841 |
*
|
|
Packit Service |
fa4841 |
* Copyright 2012 Marc-Andre Moreau <marcandre.moreau@gmail.com>
|
|
Packit Service |
fa4841 |
*
|
|
Packit Service |
fa4841 |
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
Packit Service |
fa4841 |
* you may not use this file except in compliance with the License.
|
|
Packit Service |
fa4841 |
* You may obtain a copy of the License at
|
|
Packit Service |
fa4841 |
*
|
|
Packit Service |
fa4841 |
* http://www.apache.org/licenses/LICENSE-2.0
|
|
Packit Service |
fa4841 |
*
|
|
Packit Service |
fa4841 |
* Unless required by applicable law or agreed to in writing, software
|
|
Packit Service |
fa4841 |
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
Packit Service |
fa4841 |
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
Packit Service |
fa4841 |
* See the License for the specific language governing permissions and
|
|
Packit Service |
fa4841 |
* limitations under the License.
|
|
Packit Service |
fa4841 |
*/
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
#ifdef HAVE_CONFIG_H
|
|
Packit Service |
fa4841 |
#include "config.h"
|
|
Packit Service |
fa4841 |
#endif
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
#include <freerdp/log.h>
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
#include <winpr/crt.h>
|
|
Packit Service |
fa4841 |
#include <winpr/print.h>
|
|
Packit Service |
fa4841 |
#include <winpr/synch.h>
|
|
Packit Service |
fa4841 |
#include <winpr/thread.h>
|
|
Packit Service |
fa4841 |
#include <winpr/stream.h>
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
#include "http.h"
|
|
Packit Service |
fa4841 |
#include "ncacn_http.h"
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
#include "rpc_bind.h"
|
|
Packit Service |
fa4841 |
#include "rpc_fault.h"
|
|
Packit Service |
fa4841 |
#include "rpc_client.h"
|
|
Packit Service |
fa4841 |
#include "../rdp.h"
|
|
Packit Service |
fa4841 |
#include "../proxy.h"
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
#define TAG FREERDP_TAG("core.gateway.rpc")
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
static void rpc_pdu_reset(RPC_PDU* pdu)
|
|
Packit Service |
fa4841 |
{
|
|
Packit Service |
fa4841 |
pdu->Type = 0;
|
|
Packit Service |
fa4841 |
pdu->Flags = 0;
|
|
Packit Service |
fa4841 |
pdu->CallId = 0;
|
|
Packit Service |
fa4841 |
Stream_SetPosition(pdu->s, 0);
|
|
Packit Service |
fa4841 |
}
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
static RPC_PDU* rpc_pdu_new(void)
|
|
Packit Service |
fa4841 |
{
|
|
Packit Service |
fa4841 |
RPC_PDU* pdu;
|
|
Packit Service |
fa4841 |
pdu = (RPC_PDU*)malloc(sizeof(RPC_PDU));
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
if (!pdu)
|
|
Packit Service |
fa4841 |
return NULL;
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
pdu->s = Stream_New(NULL, 4096);
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
if (!pdu->s)
|
|
Packit Service |
fa4841 |
{
|
|
Packit Service |
fa4841 |
free(pdu);
|
|
Packit Service |
fa4841 |
return NULL;
|
|
Packit Service |
fa4841 |
}
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
rpc_pdu_reset(pdu);
|
|
Packit Service |
fa4841 |
return pdu;
|
|
Packit Service |
fa4841 |
}
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
static void rpc_pdu_free(RPC_PDU* pdu)
|
|
Packit Service |
fa4841 |
{
|
|
Packit Service |
fa4841 |
if (!pdu)
|
|
Packit Service |
fa4841 |
return;
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
Stream_Free(pdu->s, TRUE);
|
|
Packit Service |
fa4841 |
free(pdu);
|
|
Packit Service |
fa4841 |
}
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
static int rpc_client_receive_pipe_write(RpcClient* client, const BYTE* buffer, size_t length)
|
|
Packit Service |
fa4841 |
{
|
|
Packit Service |
fa4841 |
int status = 0;
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
if (!client || !buffer)
|
|
Packit Service |
fa4841 |
return -1;
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
EnterCriticalSection(&(client->PipeLock));
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
if (ringbuffer_write(&(client->ReceivePipe), buffer, length))
|
|
Packit Service |
fa4841 |
status += (int)length;
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
if (ringbuffer_used(&(client->ReceivePipe)) > 0)
|
|
Packit Service |
fa4841 |
SetEvent(client->PipeEvent);
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
LeaveCriticalSection(&(client->PipeLock));
|
|
Packit Service |
fa4841 |
return status;
|
|
Packit Service |
fa4841 |
}
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
int rpc_client_receive_pipe_read(RpcClient* client, BYTE* buffer, size_t length)
|
|
Packit Service |
fa4841 |
{
|
|
Packit Service |
fa4841 |
int index = 0;
|
|
Packit Service |
fa4841 |
int status = 0;
|
|
Packit Service |
fa4841 |
int nchunks = 0;
|
|
Packit Service |
fa4841 |
DataChunk chunks[2];
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
if (!client || !buffer)
|
|
Packit Service |
fa4841 |
return -1;
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
EnterCriticalSection(&(client->PipeLock));
|
|
Packit Service |
fa4841 |
nchunks = ringbuffer_peek(&(client->ReceivePipe), chunks, length);
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
for (index = 0; index < nchunks; index++)
|
|
Packit Service |
fa4841 |
{
|
|
Packit Service |
fa4841 |
CopyMemory(&buffer[status], chunks[index].data, chunks[index].size);
|
|
Packit Service |
fa4841 |
status += chunks[index].size;
|
|
Packit Service |
fa4841 |
}
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
if (status > 0)
|
|
Packit Service |
fa4841 |
ringbuffer_commit_read_bytes(&(client->ReceivePipe), status);
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
if (ringbuffer_used(&(client->ReceivePipe)) < 1)
|
|
Packit Service |
fa4841 |
ResetEvent(client->PipeEvent);
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
LeaveCriticalSection(&(client->PipeLock));
|
|
Packit Service |
fa4841 |
return status;
|
|
Packit Service |
fa4841 |
}
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
static int rpc_client_transition_to_state(rdpRpc* rpc, RPC_CLIENT_STATE state)
|
|
Packit Service |
fa4841 |
{
|
|
Packit Service |
fa4841 |
int status = 1;
|
|
Packit Service |
fa4841 |
const char* str = "RPC_CLIENT_STATE_UNKNOWN";
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
switch (state)
|
|
Packit Service |
fa4841 |
{
|
|
Packit Service |
fa4841 |
case RPC_CLIENT_STATE_INITIAL:
|
|
Packit Service |
fa4841 |
str = "RPC_CLIENT_STATE_INITIAL";
|
|
Packit Service |
fa4841 |
break;
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
case RPC_CLIENT_STATE_ESTABLISHED:
|
|
Packit Service |
fa4841 |
str = "RPC_CLIENT_STATE_ESTABLISHED";
|
|
Packit Service |
fa4841 |
break;
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
case RPC_CLIENT_STATE_WAIT_SECURE_BIND_ACK:
|
|
Packit Service |
fa4841 |
str = "RPC_CLIENT_STATE_WAIT_SECURE_BIND_ACK";
|
|
Packit Service |
fa4841 |
break;
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
case RPC_CLIENT_STATE_WAIT_UNSECURE_BIND_ACK:
|
|
Packit Service |
fa4841 |
str = "RPC_CLIENT_STATE_WAIT_UNSECURE_BIND_ACK";
|
|
Packit Service |
fa4841 |
break;
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
case RPC_CLIENT_STATE_WAIT_SECURE_ALTER_CONTEXT_RESPONSE:
|
|
Packit Service |
fa4841 |
str = "RPC_CLIENT_STATE_WAIT_SECURE_ALTER_CONTEXT_RESPONSE";
|
|
Packit Service |
fa4841 |
break;
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
case RPC_CLIENT_STATE_CONTEXT_NEGOTIATED:
|
|
Packit Service |
fa4841 |
str = "RPC_CLIENT_STATE_CONTEXT_NEGOTIATED";
|
|
Packit Service |
fa4841 |
break;
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
case RPC_CLIENT_STATE_WAIT_RESPONSE:
|
|
Packit Service |
fa4841 |
str = "RPC_CLIENT_STATE_WAIT_RESPONSE";
|
|
Packit Service |
fa4841 |
break;
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
case RPC_CLIENT_STATE_FINAL:
|
|
Packit Service |
fa4841 |
str = "RPC_CLIENT_STATE_FINAL";
|
|
Packit Service |
fa4841 |
break;
|
|
Packit Service |
fa4841 |
}
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
rpc->State = state;
|
|
Packit Service |
fa4841 |
WLog_DBG(TAG, "%s", str);
|
|
Packit Service |
fa4841 |
return status;
|
|
Packit Service |
fa4841 |
}
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
static int rpc_client_recv_pdu(rdpRpc* rpc, RPC_PDU* pdu)
|
|
Packit Service |
fa4841 |
{
|
|
Packit Service |
fa4841 |
int status = -1;
|
|
Packit Service |
fa4841 |
rpcconn_rts_hdr_t* rts;
|
|
Packit Service |
fa4841 |
rdpTsg* tsg = rpc->transport->tsg;
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
if (rpc->VirtualConnection->State < VIRTUAL_CONNECTION_STATE_OPENED)
|
|
Packit Service |
fa4841 |
{
|
|
Packit Service |
fa4841 |
switch (rpc->VirtualConnection->State)
|
|
Packit Service |
fa4841 |
{
|
|
Packit Service |
fa4841 |
case VIRTUAL_CONNECTION_STATE_INITIAL:
|
|
Packit Service |
fa4841 |
break;
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
case VIRTUAL_CONNECTION_STATE_OUT_CHANNEL_WAIT:
|
|
Packit Service |
fa4841 |
break;
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
case VIRTUAL_CONNECTION_STATE_WAIT_A3W:
|
|
Packit Service |
fa4841 |
rts = (rpcconn_rts_hdr_t*)Stream_Buffer(pdu->s);
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
if (!rts_match_pdu_signature(&RTS_PDU_CONN_A3_SIGNATURE, rts))
|
|
Packit Service |
fa4841 |
{
|
|
Packit Service |
fa4841 |
WLog_ERR(TAG, "unexpected RTS PDU: Expected CONN/A3");
|
|
Packit Service |
fa4841 |
return -1;
|
|
Packit Service |
fa4841 |
}
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
status = rts_recv_CONN_A3_pdu(rpc, Stream_Buffer(pdu->s), Stream_Length(pdu->s));
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
if (status < 0)
|
|
Packit Service |
fa4841 |
{
|
|
Packit Service |
fa4841 |
WLog_ERR(TAG, "rts_recv_CONN_A3_pdu failure");
|
|
Packit Service |
fa4841 |
return -1;
|
|
Packit Service |
fa4841 |
}
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
rpc_virtual_connection_transition_to_state(rpc, rpc->VirtualConnection,
|
|
Packit Service |
fa4841 |
VIRTUAL_CONNECTION_STATE_WAIT_C2);
|
|
Packit Service |
fa4841 |
status = 1;
|
|
Packit Service |
fa4841 |
break;
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
case VIRTUAL_CONNECTION_STATE_WAIT_C2:
|
|
Packit Service |
fa4841 |
rts = (rpcconn_rts_hdr_t*)Stream_Buffer(pdu->s);
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
if (!rts_match_pdu_signature(&RTS_PDU_CONN_C2_SIGNATURE, rts))
|
|
Packit Service |
fa4841 |
{
|
|
Packit Service |
fa4841 |
WLog_ERR(TAG, "unexpected RTS PDU: Expected CONN/C2");
|
|
Packit Service |
fa4841 |
return -1;
|
|
Packit Service |
fa4841 |
}
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
status = rts_recv_CONN_C2_pdu(rpc, Stream_Buffer(pdu->s), Stream_Length(pdu->s));
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
if (status < 0)
|
|
Packit Service |
fa4841 |
{
|
|
Packit Service |
fa4841 |
WLog_ERR(TAG, "rts_recv_CONN_C2_pdu failure");
|
|
Packit Service |
fa4841 |
return -1;
|
|
Packit Service |
fa4841 |
}
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
rpc_virtual_connection_transition_to_state(rpc, rpc->VirtualConnection,
|
|
Packit Service |
fa4841 |
VIRTUAL_CONNECTION_STATE_OPENED);
|
|
Packit Service |
fa4841 |
rpc_client_transition_to_state(rpc, RPC_CLIENT_STATE_ESTABLISHED);
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
if (rpc_send_bind_pdu(rpc) < 0)
|
|
Packit Service |
fa4841 |
{
|
|
Packit Service |
fa4841 |
WLog_ERR(TAG, "rpc_send_bind_pdu failure");
|
|
Packit Service |
fa4841 |
return -1;
|
|
Packit Service |
fa4841 |
}
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
rpc_client_transition_to_state(rpc, RPC_CLIENT_STATE_WAIT_SECURE_BIND_ACK);
|
|
Packit Service |
fa4841 |
status = 1;
|
|
Packit Service |
fa4841 |
break;
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
case VIRTUAL_CONNECTION_STATE_OPENED:
|
|
Packit Service |
fa4841 |
break;
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
case VIRTUAL_CONNECTION_STATE_FINAL:
|
|
Packit Service |
fa4841 |
break;
|
|
Packit Service |
fa4841 |
}
|
|
Packit Service |
fa4841 |
}
|
|
Packit Service |
fa4841 |
else if (rpc->State < RPC_CLIENT_STATE_CONTEXT_NEGOTIATED)
|
|
Packit Service |
fa4841 |
{
|
|
Packit Service |
fa4841 |
if (rpc->State == RPC_CLIENT_STATE_WAIT_SECURE_BIND_ACK)
|
|
Packit Service |
fa4841 |
{
|
|
Packit Service |
fa4841 |
if (pdu->Type == PTYPE_BIND_ACK)
|
|
Packit Service |
fa4841 |
{
|
|
Packit Service |
fa4841 |
if (rpc_recv_bind_ack_pdu(rpc, Stream_Buffer(pdu->s), Stream_Length(pdu->s)) <= 0)
|
|
Packit Service |
fa4841 |
{
|
|
Packit Service |
fa4841 |
WLog_ERR(TAG, "rpc_recv_bind_ack_pdu failure");
|
|
Packit Service |
fa4841 |
return -1;
|
|
Packit Service |
fa4841 |
}
|
|
Packit Service |
fa4841 |
}
|
|
Packit Service |
fa4841 |
else
|
|
Packit Service |
fa4841 |
{
|
|
Packit Service |
fa4841 |
WLog_ERR(TAG,
|
|
Packit Service |
fa4841 |
"RPC_CLIENT_STATE_WAIT_SECURE_BIND_ACK unexpected pdu type: 0x%08" PRIX32
|
|
Packit Service |
fa4841 |
"",
|
|
Packit Service |
fa4841 |
pdu->Type);
|
|
Packit Service |
fa4841 |
return -1;
|
|
Packit Service |
fa4841 |
}
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
if (rpc_send_rpc_auth_3_pdu(rpc) < 0)
|
|
Packit Service |
fa4841 |
{
|
|
Packit Service |
fa4841 |
WLog_ERR(TAG, "rpc_secure_bind: error sending rpc_auth_3 pdu!");
|
|
Packit Service |
fa4841 |
return -1;
|
|
Packit Service |
fa4841 |
}
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
rpc_client_transition_to_state(rpc, RPC_CLIENT_STATE_CONTEXT_NEGOTIATED);
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
if (!tsg_proxy_begin(tsg))
|
|
Packit Service |
fa4841 |
{
|
|
Packit Service |
fa4841 |
WLog_ERR(TAG, "tsg_proxy_begin failure");
|
|
Packit Service |
fa4841 |
return -1;
|
|
Packit Service |
fa4841 |
}
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
status = 1;
|
|
Packit Service |
fa4841 |
}
|
|
Packit Service |
fa4841 |
else
|
|
Packit Service |
fa4841 |
{
|
|
Packit Service |
fa4841 |
WLog_ERR(TAG, "rpc_client_recv_pdu: invalid rpc->State: %d", rpc->State);
|
|
Packit Service |
fa4841 |
}
|
|
Packit Service |
fa4841 |
}
|
|
Packit Service |
fa4841 |
else if (rpc->State >= RPC_CLIENT_STATE_CONTEXT_NEGOTIATED)
|
|
Packit Service |
fa4841 |
{
|
|
Packit Service |
fa4841 |
if (!tsg_recv_pdu(tsg, pdu))
|
|
Packit Service |
fa4841 |
status = -1;
|
|
Packit Service |
fa4841 |
else
|
|
Packit Service |
fa4841 |
status = 1;
|
|
Packit Service |
fa4841 |
}
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
return status;
|
|
Packit Service |
fa4841 |
}
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
static int rpc_client_recv_fragment(rdpRpc* rpc, wStream* fragment)
|
|
Packit Service |
fa4841 |
{
|
|
Packit Service |
fa4841 |
BYTE* buffer;
|
|
Packit Service |
fa4841 |
RPC_PDU* pdu;
|
|
Packit Service |
fa4841 |
UINT32 StubOffset;
|
|
Packit Service |
fa4841 |
UINT32 StubLength;
|
|
Packit Service |
fa4841 |
RpcClientCall* call;
|
|
Packit Service |
fa4841 |
rpcconn_hdr_t* header;
|
|
Packit Service |
fa4841 |
pdu = rpc->client->pdu;
|
|
Packit Service |
fa4841 |
buffer = (BYTE*)Stream_Buffer(fragment);
|
|
Packit Service |
fa4841 |
header = (rpcconn_hdr_t*)Stream_Buffer(fragment);
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
if (header->common.ptype == PTYPE_RESPONSE)
|
|
Packit Service |
fa4841 |
{
|
|
Packit Service |
fa4841 |
rpc->VirtualConnection->DefaultOutChannel->BytesReceived += header->common.frag_length;
|
|
Packit Service |
fa4841 |
rpc->VirtualConnection->DefaultOutChannel->ReceiverAvailableWindow -=
|
|
Packit Service |
fa4841 |
header->common.frag_length;
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
if (rpc->VirtualConnection->DefaultOutChannel->ReceiverAvailableWindow <
|
|
Packit Service |
fa4841 |
(rpc->ReceiveWindow / 2))
|
|
Packit Service |
fa4841 |
{
|
|
Packit Service |
fa4841 |
if (rts_send_flow_control_ack_pdu(rpc) < 0)
|
|
Packit Service |
fa4841 |
return -1;
|
|
Packit Service |
fa4841 |
}
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
if (!rpc_get_stub_data_info(rpc, buffer, &StubOffset, &StubLength))
|
|
Packit Service |
fa4841 |
{
|
|
Packit Service |
fa4841 |
WLog_ERR(TAG, "expected stub");
|
|
Packit Service |
fa4841 |
return -1;
|
|
Packit Service |
fa4841 |
}
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
if (StubLength == 4)
|
|
Packit Service |
fa4841 |
{
|
|
Packit Service |
fa4841 |
if ((header->common.call_id == rpc->PipeCallId) &&
|
|
Packit Service |
fa4841 |
(header->common.pfc_flags & PFC_LAST_FRAG))
|
|
Packit Service |
fa4841 |
{
|
|
Packit Service |
fa4841 |
/* End of TsProxySetupReceivePipe */
|
|
Packit Service |
fa4841 |
TerminateEventArgs e;
|
|
Packit Service |
fa4841 |
rpc->result = *((UINT32*)&buffer[StubOffset]);
|
|
Packit Service |
fa4841 |
freerdp_abort_connect(rpc->context->instance);
|
|
Packit Service |
fa4841 |
tsg_set_state(rpc->transport->tsg, TSG_STATE_TUNNEL_CLOSE_PENDING);
|
|
Packit Service |
fa4841 |
EventArgsInit(&e, "freerdp");
|
|
Packit Service |
fa4841 |
e.code = 0;
|
|
Packit Service |
fa4841 |
PubSub_OnTerminate(rpc->context->pubSub, rpc->context, &e);
|
|
Packit Service |
fa4841 |
return 0;
|
|
Packit Service |
fa4841 |
}
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
if (header->common.call_id != rpc->PipeCallId)
|
|
Packit Service |
fa4841 |
{
|
|
Packit Service |
fa4841 |
/* Ignoring non-TsProxySetupReceivePipe Response */
|
|
Packit Service |
fa4841 |
return 0;
|
|
Packit Service |
fa4841 |
}
|
|
Packit Service |
fa4841 |
}
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
if (rpc->StubFragCount == 0)
|
|
Packit Service |
fa4841 |
rpc->StubCallId = header->common.call_id;
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
if (rpc->StubCallId != header->common.call_id)
|
|
Packit Service |
fa4841 |
{
|
|
Packit Service |
fa4841 |
WLog_ERR(TAG,
|
|
Packit Service |
fa4841 |
"invalid call_id: actual: %" PRIu32 ", expected: %" PRIu32
|
|
Packit Service |
fa4841 |
", frag_count: %" PRIu32 "",
|
|
Packit Service |
fa4841 |
rpc->StubCallId, header->common.call_id, rpc->StubFragCount);
|
|
Packit Service |
fa4841 |
}
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
call = rpc_client_call_find_by_id(rpc->client, rpc->StubCallId);
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
if (!call)
|
|
Packit Service |
fa4841 |
return -1;
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
if (call->OpNum != TsProxySetupReceivePipeOpnum)
|
|
Packit Service |
fa4841 |
{
|
|
Packit Service |
fa4841 |
if (!Stream_EnsureCapacity(pdu->s, header->response.alloc_hint))
|
|
Packit Service |
fa4841 |
return -1;
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
Stream_Write(pdu->s, &buffer[StubOffset], StubLength);
|
|
Packit Service |
fa4841 |
rpc->StubFragCount++;
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
if (header->response.alloc_hint == StubLength)
|
|
Packit Service |
fa4841 |
{
|
|
Packit Service |
fa4841 |
pdu->Flags = RPC_PDU_FLAG_STUB;
|
|
Packit Service |
fa4841 |
pdu->Type = PTYPE_RESPONSE;
|
|
Packit Service |
fa4841 |
pdu->CallId = rpc->StubCallId;
|
|
Packit Service |
fa4841 |
Stream_SealLength(pdu->s);
|
|
Packit Service |
fa4841 |
rpc_client_recv_pdu(rpc, pdu);
|
|
Packit Service |
fa4841 |
rpc_pdu_reset(pdu);
|
|
Packit Service |
fa4841 |
rpc->StubFragCount = 0;
|
|
Packit Service |
fa4841 |
rpc->StubCallId = 0;
|
|
Packit Service |
fa4841 |
}
|
|
Packit Service |
fa4841 |
}
|
|
Packit Service |
fa4841 |
else
|
|
Packit Service |
fa4841 |
{
|
|
Packit Service |
fa4841 |
rpc_client_receive_pipe_write(rpc->client, &buffer[StubOffset], (size_t)StubLength);
|
|
Packit Service |
fa4841 |
rpc->StubFragCount++;
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
if (header->response.alloc_hint == StubLength)
|
|
Packit Service |
fa4841 |
{
|
|
Packit Service |
fa4841 |
rpc->StubFragCount = 0;
|
|
Packit Service |
fa4841 |
rpc->StubCallId = 0;
|
|
Packit Service |
fa4841 |
}
|
|
Packit Service |
fa4841 |
}
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
return 1;
|
|
Packit Service |
fa4841 |
}
|
|
Packit Service |
fa4841 |
else if (header->common.ptype == PTYPE_RTS)
|
|
Packit Service |
fa4841 |
{
|
|
Packit Service |
fa4841 |
if (rpc->State < RPC_CLIENT_STATE_CONTEXT_NEGOTIATED)
|
|
Packit Service |
fa4841 |
{
|
|
Packit Service |
fa4841 |
pdu->Flags = 0;
|
|
Packit Service |
fa4841 |
pdu->Type = header->common.ptype;
|
|
Packit Service |
fa4841 |
pdu->CallId = header->common.call_id;
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
if (!Stream_EnsureCapacity(pdu->s, Stream_Length(fragment)))
|
|
Packit Service |
fa4841 |
return -1;
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
Stream_Write(pdu->s, buffer, Stream_Length(fragment));
|
|
Packit Service |
fa4841 |
Stream_SealLength(pdu->s);
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
if (rpc_client_recv_pdu(rpc, pdu) < 0)
|
|
Packit Service |
fa4841 |
return -1;
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
rpc_pdu_reset(pdu);
|
|
Packit Service |
fa4841 |
}
|
|
Packit Service |
fa4841 |
else
|
|
Packit Service |
fa4841 |
{
|
|
Packit Service |
fa4841 |
if (rts_recv_out_of_sequence_pdu(rpc, buffer, header->common.frag_length) < 0)
|
|
Packit Service |
fa4841 |
return -1;
|
|
Packit Service |
fa4841 |
}
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
return 1;
|
|
Packit Service |
fa4841 |
}
|
|
Packit Service |
fa4841 |
else if (header->common.ptype == PTYPE_BIND_ACK)
|
|
Packit Service |
fa4841 |
{
|
|
Packit Service |
fa4841 |
pdu->Flags = 0;
|
|
Packit Service |
fa4841 |
pdu->Type = header->common.ptype;
|
|
Packit Service |
fa4841 |
pdu->CallId = header->common.call_id;
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
if (!Stream_EnsureCapacity(pdu->s, Stream_Length(fragment)))
|
|
Packit Service |
fa4841 |
return -1;
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
Stream_Write(pdu->s, buffer, Stream_Length(fragment));
|
|
Packit Service |
fa4841 |
Stream_SealLength(pdu->s);
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
if (rpc_client_recv_pdu(rpc, pdu) < 0)
|
|
Packit Service |
fa4841 |
return -1;
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
rpc_pdu_reset(pdu);
|
|
Packit Service |
fa4841 |
return 1;
|
|
Packit Service |
fa4841 |
}
|
|
Packit Service |
fa4841 |
else if (header->common.ptype == PTYPE_FAULT)
|
|
Packit Service |
fa4841 |
{
|
|
Packit Service |
fa4841 |
rpc_recv_fault_pdu(header->fault.status);
|
|
Packit Service |
fa4841 |
return -1;
|
|
Packit Service |
fa4841 |
}
|
|
Packit Service |
fa4841 |
else
|
|
Packit Service |
fa4841 |
{
|
|
Packit Service |
fa4841 |
WLog_ERR(TAG, "unexpected RPC PDU type 0x%02" PRIX8 "", header->common.ptype);
|
|
Packit Service |
fa4841 |
return -1;
|
|
Packit Service |
fa4841 |
}
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
return 1;
|
|
Packit Service |
fa4841 |
}
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
static int rpc_client_default_out_channel_recv(rdpRpc* rpc)
|
|
Packit Service |
fa4841 |
{
|
|
Packit Service |
fa4841 |
int status = -1;
|
|
Packit Service |
fa4841 |
UINT32 statusCode;
|
|
Packit Service |
fa4841 |
HttpResponse* response;
|
|
Packit Service |
fa4841 |
RpcInChannel* inChannel;
|
|
Packit Service |
fa4841 |
RpcOutChannel* outChannel;
|
|
Packit Service |
fa4841 |
HANDLE outChannelEvent = NULL;
|
|
Packit Service |
fa4841 |
RpcVirtualConnection* connection = rpc->VirtualConnection;
|
|
Packit Service |
fa4841 |
inChannel = connection->DefaultInChannel;
|
|
Packit Service |
fa4841 |
outChannel = connection->DefaultOutChannel;
|
|
Packit Service |
fa4841 |
BIO_get_event(outChannel->common.tls->bio, &outChannelEvent);
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
if (outChannel->State < CLIENT_OUT_CHANNEL_STATE_OPENED)
|
|
Packit Service |
fa4841 |
{
|
|
Packit Service |
fa4841 |
if (WaitForSingleObject(outChannelEvent, 0) != WAIT_OBJECT_0)
|
|
Packit Service |
fa4841 |
return 1;
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
response = http_response_recv(outChannel->common.tls, TRUE);
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
if (!response)
|
|
Packit Service |
fa4841 |
return -1;
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
if (outChannel->State == CLIENT_OUT_CHANNEL_STATE_SECURITY)
|
|
Packit Service |
fa4841 |
{
|
|
Packit Service |
fa4841 |
/* Receive OUT Channel Response */
|
|
Packit Service |
fa4841 |
if (!rpc_ncacn_http_recv_out_channel_response(&outChannel->common, response))
|
|
Packit Service |
fa4841 |
{
|
|
Packit Service |
fa4841 |
http_response_free(response);
|
|
Packit Service |
fa4841 |
WLog_ERR(TAG, "rpc_ncacn_http_recv_out_channel_response failure");
|
|
Packit Service |
fa4841 |
return -1;
|
|
Packit Service |
fa4841 |
}
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
/* Send OUT Channel Request */
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
if (!rpc_ncacn_http_send_out_channel_request(&outChannel->common, FALSE))
|
|
Packit Service |
fa4841 |
{
|
|
Packit Service |
fa4841 |
http_response_free(response);
|
|
Packit Service |
fa4841 |
WLog_ERR(TAG, "rpc_ncacn_http_send_out_channel_request failure");
|
|
Packit Service |
fa4841 |
return -1;
|
|
Packit Service |
fa4841 |
}
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
rpc_ncacn_http_ntlm_uninit(&outChannel->common);
|
|
Packit Service |
fa4841 |
rpc_out_channel_transition_to_state(outChannel, CLIENT_OUT_CHANNEL_STATE_NEGOTIATED);
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
/* Send CONN/A1 PDU over OUT channel */
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
if (rts_send_CONN_A1_pdu(rpc) < 0)
|
|
Packit Service |
fa4841 |
{
|
|
Packit Service |
fa4841 |
http_response_free(response);
|
|
Packit Service |
fa4841 |
WLog_ERR(TAG, "rpc_send_CONN_A1_pdu error!");
|
|
Packit Service |
fa4841 |
return -1;
|
|
Packit Service |
fa4841 |
}
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
rpc_out_channel_transition_to_state(outChannel, CLIENT_OUT_CHANNEL_STATE_OPENED);
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
if (inChannel->State == CLIENT_IN_CHANNEL_STATE_OPENED)
|
|
Packit Service |
fa4841 |
{
|
|
Packit Service |
fa4841 |
rpc_virtual_connection_transition_to_state(
|
|
Packit Service |
fa4841 |
rpc, connection, VIRTUAL_CONNECTION_STATE_OUT_CHANNEL_WAIT);
|
|
Packit Service |
fa4841 |
}
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
status = 1;
|
|
Packit Service |
fa4841 |
}
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
http_response_free(response);
|
|
Packit Service |
fa4841 |
}
|
|
Packit Service |
fa4841 |
else if (connection->State == VIRTUAL_CONNECTION_STATE_OUT_CHANNEL_WAIT)
|
|
Packit Service |
fa4841 |
{
|
|
Packit Service |
fa4841 |
/* Receive OUT channel response */
|
|
Packit Service |
fa4841 |
if (WaitForSingleObject(outChannelEvent, 0) != WAIT_OBJECT_0)
|
|
Packit Service |
fa4841 |
return 1;
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
response = http_response_recv(outChannel->common.tls, FALSE);
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
if (!response)
|
|
Packit Service |
fa4841 |
return -1;
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
statusCode = http_response_get_status_code(response);
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
if (statusCode != HTTP_STATUS_OK)
|
|
Packit Service |
fa4841 |
{
|
|
Packit Service |
fa4841 |
WLog_ERR(TAG, "error! Status Code: %" PRIu32 "", statusCode);
|
|
Packit Service |
fa4841 |
http_response_print(response);
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
if (statusCode == HTTP_STATUS_DENIED)
|
|
Packit Service |
fa4841 |
{
|
|
Packit Service |
fa4841 |
freerdp_set_last_error_if_not(rpc->context, FREERDP_ERROR_AUTHENTICATION_FAILED);
|
|
Packit Service |
fa4841 |
}
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
http_response_free(response);
|
|
Packit Service |
fa4841 |
return -1;
|
|
Packit Service |
fa4841 |
}
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
http_response_free(response);
|
|
Packit Service |
fa4841 |
rpc_virtual_connection_transition_to_state(rpc, rpc->VirtualConnection,
|
|
Packit Service |
fa4841 |
VIRTUAL_CONNECTION_STATE_WAIT_A3W);
|
|
Packit Service |
fa4841 |
status = 1;
|
|
Packit Service |
fa4841 |
}
|
|
Packit Service |
fa4841 |
else
|
|
Packit Service |
fa4841 |
{
|
|
Packit Service |
fa4841 |
wStream* fragment;
|
|
Packit Service |
fa4841 |
rpcconn_common_hdr_t* header;
|
|
Packit Service |
fa4841 |
fragment = rpc->client->ReceiveFragment;
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
while (1)
|
|
Packit Service |
fa4841 |
{
|
|
Packit Service |
fa4841 |
while (Stream_GetPosition(fragment) < RPC_COMMON_FIELDS_LENGTH)
|
|
Packit Service |
fa4841 |
{
|
|
Packit Service |
fa4841 |
status = rpc_channel_read(&outChannel->common, fragment,
|
|
Packit Service |
fa4841 |
RPC_COMMON_FIELDS_LENGTH - Stream_GetPosition(fragment));
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
if (status < 0)
|
|
Packit Service |
fa4841 |
return -1;
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
if (Stream_GetPosition(fragment) < RPC_COMMON_FIELDS_LENGTH)
|
|
Packit Service |
fa4841 |
return 0;
|
|
Packit Service |
fa4841 |
}
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
header = (rpcconn_common_hdr_t*)Stream_Buffer(fragment);
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
if (header->frag_length > rpc->max_recv_frag)
|
|
Packit Service |
fa4841 |
{
|
|
Packit Service |
fa4841 |
WLog_ERR(TAG,
|
|
Packit Service |
fa4841 |
"rpc_client_recv: invalid fragment size: %" PRIu16 " (max: %" PRIu16 ")",
|
|
Packit Service |
fa4841 |
header->frag_length, rpc->max_recv_frag);
|
|
Packit Service |
fa4841 |
winpr_HexDump(TAG, WLOG_ERROR, Stream_Buffer(fragment),
|
|
Packit Service |
fa4841 |
Stream_GetPosition(fragment));
|
|
Packit Service |
fa4841 |
return -1;
|
|
Packit Service |
fa4841 |
}
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
while (Stream_GetPosition(fragment) < header->frag_length)
|
|
Packit Service |
fa4841 |
{
|
|
Packit Service |
fa4841 |
status = rpc_channel_read(&outChannel->common, fragment,
|
|
Packit Service |
fa4841 |
header->frag_length - Stream_GetPosition(fragment));
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
if (status < 0)
|
|
Packit Service |
fa4841 |
{
|
|
Packit Service |
fa4841 |
WLog_ERR(TAG, "error reading fragment body");
|
|
Packit Service |
fa4841 |
return -1;
|
|
Packit Service |
fa4841 |
}
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
if (Stream_GetPosition(fragment) < header->frag_length)
|
|
Packit Service |
fa4841 |
return 0;
|
|
Packit Service |
fa4841 |
}
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
{
|
|
Packit Service |
fa4841 |
/* complete fragment received */
|
|
Packit Service |
fa4841 |
Stream_SealLength(fragment);
|
|
Packit Service |
fa4841 |
Stream_SetPosition(fragment, 0);
|
|
Packit Service |
fa4841 |
status = rpc_client_recv_fragment(rpc, fragment);
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
if (status < 0)
|
|
Packit Service |
fa4841 |
return status;
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
/* channel recycling may update channel pointers */
|
|
Packit Service |
fa4841 |
if (outChannel->State == CLIENT_OUT_CHANNEL_STATE_RECYCLED &&
|
|
Packit Service |
fa4841 |
connection->NonDefaultOutChannel)
|
|
Packit Service |
fa4841 |
{
|
|
Packit Service |
fa4841 |
rpc_channel_free(&connection->DefaultOutChannel->common);
|
|
Packit Service |
fa4841 |
connection->DefaultOutChannel = connection->NonDefaultOutChannel;
|
|
Packit Service |
fa4841 |
connection->NonDefaultOutChannel = NULL;
|
|
Packit Service |
fa4841 |
rpc_out_channel_transition_to_state(connection->DefaultOutChannel,
|
|
Packit Service |
fa4841 |
CLIENT_OUT_CHANNEL_STATE_OPENED);
|
|
Packit Service |
fa4841 |
rpc_virtual_connection_transition_to_state(
|
|
Packit Service |
fa4841 |
rpc, connection, VIRTUAL_CONNECTION_STATE_OUT_CHANNEL_WAIT);
|
|
Packit Service |
fa4841 |
return 0;
|
|
Packit Service |
fa4841 |
}
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
Stream_SetPosition(fragment, 0);
|
|
Packit Service |
fa4841 |
}
|
|
Packit Service |
fa4841 |
}
|
|
Packit Service |
fa4841 |
}
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
return status;
|
|
Packit Service |
fa4841 |
}
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
static int rpc_client_nondefault_out_channel_recv(rdpRpc* rpc)
|
|
Packit Service |
fa4841 |
{
|
|
Packit Service |
fa4841 |
int status = -1;
|
|
Packit Service |
fa4841 |
HttpResponse* response;
|
|
Packit Service |
fa4841 |
RpcOutChannel* nextOutChannel;
|
|
Packit Service |
fa4841 |
HANDLE nextOutChannelEvent = NULL;
|
|
Packit Service |
fa4841 |
nextOutChannel = rpc->VirtualConnection->NonDefaultOutChannel;
|
|
Packit Service |
fa4841 |
BIO_get_event(nextOutChannel->common.tls->bio, &nextOutChannelEvent);
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
if (WaitForSingleObject(nextOutChannelEvent, 0) != WAIT_OBJECT_0)
|
|
Packit Service |
fa4841 |
return 1;
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
response = http_response_recv(nextOutChannel->common.tls, TRUE);
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
if (response)
|
|
Packit Service |
fa4841 |
{
|
|
Packit Service |
fa4841 |
switch (nextOutChannel->State)
|
|
Packit Service |
fa4841 |
{
|
|
Packit Service |
fa4841 |
case CLIENT_OUT_CHANNEL_STATE_SECURITY:
|
|
Packit Service |
fa4841 |
if (rpc_ncacn_http_recv_out_channel_response(&nextOutChannel->common, response))
|
|
Packit Service |
fa4841 |
{
|
|
Packit Service |
fa4841 |
if (rpc_ncacn_http_send_out_channel_request(&nextOutChannel->common, TRUE))
|
|
Packit Service |
fa4841 |
{
|
|
Packit Service |
fa4841 |
rpc_ncacn_http_ntlm_uninit(&nextOutChannel->common);
|
|
Packit Service |
fa4841 |
status = rts_send_OUT_R1_A3_pdu(rpc);
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
if (status >= 0)
|
|
Packit Service |
fa4841 |
{
|
|
Packit Service |
fa4841 |
rpc_out_channel_transition_to_state(
|
|
Packit Service |
fa4841 |
nextOutChannel, CLIENT_OUT_CHANNEL_STATE_OPENED_A6W);
|
|
Packit Service |
fa4841 |
}
|
|
Packit Service |
fa4841 |
else
|
|
Packit Service |
fa4841 |
{
|
|
Packit Service |
fa4841 |
WLog_ERR(TAG, "rts_send_OUT_R1/A3_pdu failure");
|
|
Packit Service |
fa4841 |
}
|
|
Packit Service |
fa4841 |
}
|
|
Packit Service |
fa4841 |
else
|
|
Packit Service |
fa4841 |
{
|
|
Packit Service |
fa4841 |
WLog_ERR(TAG, "rpc_ncacn_http_send_out_channel_request failure");
|
|
Packit Service |
fa4841 |
}
|
|
Packit Service |
fa4841 |
}
|
|
Packit Service |
fa4841 |
else
|
|
Packit Service |
fa4841 |
{
|
|
Packit Service |
fa4841 |
WLog_ERR(TAG, "rpc_ncacn_http_recv_out_channel_response failure");
|
|
Packit Service |
fa4841 |
}
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
break;
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
default:
|
|
Packit Service |
fa4841 |
WLog_ERR(TAG,
|
|
Packit Service |
fa4841 |
"rpc_client_nondefault_out_channel_recv: Unexpected message %08" PRIx32,
|
|
Packit Service |
fa4841 |
nextOutChannel->State);
|
|
Packit Service |
fa4841 |
return -1;
|
|
Packit Service |
fa4841 |
}
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
http_response_free(response);
|
|
Packit Service |
fa4841 |
}
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
return status;
|
|
Packit Service |
fa4841 |
}
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
int rpc_client_out_channel_recv(rdpRpc* rpc)
|
|
Packit Service |
fa4841 |
{
|
|
Packit Service |
fa4841 |
int status;
|
|
Packit Service |
fa4841 |
RpcVirtualConnection* connection = rpc->VirtualConnection;
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
if (connection->DefaultOutChannel)
|
|
Packit Service |
fa4841 |
{
|
|
Packit Service |
fa4841 |
status = rpc_client_default_out_channel_recv(rpc);
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
if (status < 0)
|
|
Packit Service |
fa4841 |
return -1;
|
|
Packit Service |
fa4841 |
}
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
if (connection->NonDefaultOutChannel)
|
|
Packit Service |
fa4841 |
{
|
|
Packit Service |
fa4841 |
status = rpc_client_nondefault_out_channel_recv(rpc);
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
if (status < 0)
|
|
Packit Service |
fa4841 |
return -1;
|
|
Packit Service |
fa4841 |
}
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
return 1;
|
|
Packit Service |
fa4841 |
}
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
int rpc_client_in_channel_recv(rdpRpc* rpc)
|
|
Packit Service |
fa4841 |
{
|
|
Packit Service |
fa4841 |
int status = 1;
|
|
Packit Service |
fa4841 |
HttpResponse* response;
|
|
Packit Service |
fa4841 |
RpcInChannel* inChannel;
|
|
Packit Service |
fa4841 |
RpcOutChannel* outChannel;
|
|
Packit Service |
fa4841 |
HANDLE InChannelEvent = NULL;
|
|
Packit Service |
fa4841 |
RpcVirtualConnection* connection = rpc->VirtualConnection;
|
|
Packit Service |
fa4841 |
inChannel = connection->DefaultInChannel;
|
|
Packit Service |
fa4841 |
outChannel = connection->DefaultOutChannel;
|
|
Packit Service |
fa4841 |
BIO_get_event(inChannel->common.tls->bio, &InChannelEvent);
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
if (WaitForSingleObject(InChannelEvent, 0) != WAIT_OBJECT_0)
|
|
Packit Service |
fa4841 |
return 1;
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
if (inChannel->State < CLIENT_IN_CHANNEL_STATE_OPENED)
|
|
Packit Service |
fa4841 |
{
|
|
Packit Service |
fa4841 |
response = http_response_recv(inChannel->common.tls, TRUE);
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
if (!response)
|
|
Packit Service |
fa4841 |
return -1;
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
if (inChannel->State == CLIENT_IN_CHANNEL_STATE_SECURITY)
|
|
Packit Service |
fa4841 |
{
|
|
Packit Service |
fa4841 |
if (!rpc_ncacn_http_recv_in_channel_response(&inChannel->common, response))
|
|
Packit Service |
fa4841 |
{
|
|
Packit Service |
fa4841 |
WLog_ERR(TAG, "rpc_ncacn_http_recv_in_channel_response failure");
|
|
Packit Service |
fa4841 |
http_response_free(response);
|
|
Packit Service |
fa4841 |
return -1;
|
|
Packit Service |
fa4841 |
}
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
/* Send IN Channel Request */
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
if (!rpc_ncacn_http_send_in_channel_request(&inChannel->common))
|
|
Packit Service |
fa4841 |
{
|
|
Packit Service |
fa4841 |
WLog_ERR(TAG, "rpc_ncacn_http_send_in_channel_request failure");
|
|
Packit Service |
fa4841 |
http_response_free(response);
|
|
Packit Service |
fa4841 |
return -1;
|
|
Packit Service |
fa4841 |
}
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
rpc_ncacn_http_ntlm_uninit(&inChannel->common);
|
|
Packit Service |
fa4841 |
rpc_in_channel_transition_to_state(inChannel, CLIENT_IN_CHANNEL_STATE_NEGOTIATED);
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
/* Send CONN/B1 PDU over IN channel */
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
if (rts_send_CONN_B1_pdu(rpc) < 0)
|
|
Packit Service |
fa4841 |
{
|
|
Packit Service |
fa4841 |
WLog_ERR(TAG, "rpc_send_CONN_B1_pdu error!");
|
|
Packit Service |
fa4841 |
http_response_free(response);
|
|
Packit Service |
fa4841 |
return -1;
|
|
Packit Service |
fa4841 |
}
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
rpc_in_channel_transition_to_state(inChannel, CLIENT_IN_CHANNEL_STATE_OPENED);
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
if (outChannel->State == CLIENT_OUT_CHANNEL_STATE_OPENED)
|
|
Packit Service |
fa4841 |
{
|
|
Packit Service |
fa4841 |
rpc_virtual_connection_transition_to_state(
|
|
Packit Service |
fa4841 |
rpc, connection, VIRTUAL_CONNECTION_STATE_OUT_CHANNEL_WAIT);
|
|
Packit Service |
fa4841 |
}
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
status = 1;
|
|
Packit Service |
fa4841 |
}
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
http_response_free(response);
|
|
Packit Service |
fa4841 |
}
|
|
Packit Service |
fa4841 |
else
|
|
Packit Service |
fa4841 |
{
|
|
Packit Service |
fa4841 |
response = http_response_recv(inChannel->common.tls, TRUE);
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
if (!response)
|
|
Packit Service |
fa4841 |
return -1;
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
/* We can receive an unauthorized HTTP response on the IN channel */
|
|
Packit Service |
fa4841 |
http_response_free(response);
|
|
Packit Service |
fa4841 |
}
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
return status;
|
|
Packit Service |
fa4841 |
}
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
/**
|
|
Packit Service |
fa4841 |
* [MS-RPCE] Client Call:
|
|
Packit Service |
fa4841 |
* http://msdn.microsoft.com/en-us/library/gg593159/
|
|
Packit Service |
fa4841 |
*/
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
RpcClientCall* rpc_client_call_find_by_id(RpcClient* client, UINT32 CallId)
|
|
Packit Service |
fa4841 |
{
|
|
Packit Service |
fa4841 |
int index;
|
|
Packit Service |
fa4841 |
int count;
|
|
Packit Service |
fa4841 |
RpcClientCall* clientCall = NULL;
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
if (!client)
|
|
Packit Service |
fa4841 |
return NULL;
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
ArrayList_Lock(client->ClientCallList);
|
|
Packit Service |
fa4841 |
count = ArrayList_Count(client->ClientCallList);
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
for (index = 0; index < count; index++)
|
|
Packit Service |
fa4841 |
{
|
|
Packit Service |
fa4841 |
clientCall = (RpcClientCall*)ArrayList_GetItem(client->ClientCallList, index);
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
if (clientCall->CallId == CallId)
|
|
Packit Service |
fa4841 |
break;
|
|
Packit Service |
fa4841 |
}
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
ArrayList_Unlock(client->ClientCallList);
|
|
Packit Service |
fa4841 |
return clientCall;
|
|
Packit Service |
fa4841 |
}
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
RpcClientCall* rpc_client_call_new(UINT32 CallId, UINT32 OpNum)
|
|
Packit Service |
fa4841 |
{
|
|
Packit Service |
fa4841 |
RpcClientCall* clientCall;
|
|
Packit Service |
fa4841 |
clientCall = (RpcClientCall*)calloc(1, sizeof(RpcClientCall));
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
if (!clientCall)
|
|
Packit Service |
fa4841 |
return NULL;
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
clientCall->CallId = CallId;
|
|
Packit Service |
fa4841 |
clientCall->OpNum = OpNum;
|
|
Packit Service |
fa4841 |
clientCall->State = RPC_CLIENT_CALL_STATE_SEND_PDUS;
|
|
Packit Service |
fa4841 |
return clientCall;
|
|
Packit Service |
fa4841 |
}
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
void rpc_client_call_free(RpcClientCall* clientCall)
|
|
Packit Service |
fa4841 |
{
|
|
Packit Service |
fa4841 |
free(clientCall);
|
|
Packit Service |
fa4841 |
}
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
static void rpc_array_client_call_free(void* call)
|
|
Packit Service |
fa4841 |
{
|
|
Packit Service |
fa4841 |
rpc_client_call_free((RpcClientCall*)call);
|
|
Packit Service |
fa4841 |
}
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
int rpc_in_channel_send_pdu(RpcInChannel* inChannel, BYTE* buffer, UINT32 length)
|
|
Packit Service |
fa4841 |
{
|
|
Packit Service |
fa4841 |
int status;
|
|
Packit Service |
fa4841 |
RpcClientCall* clientCall;
|
|
Packit Service |
fa4841 |
rpcconn_common_hdr_t* header;
|
|
Packit Service |
fa4841 |
status = rpc_channel_write(&inChannel->common, buffer, length);
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
if (status <= 0)
|
|
Packit Service |
fa4841 |
return -1;
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
header = (rpcconn_common_hdr_t*)buffer;
|
|
Packit Service |
fa4841 |
clientCall = rpc_client_call_find_by_id(inChannel->common.client, header->call_id);
|
|
Packit Service |
fa4841 |
clientCall->State = RPC_CLIENT_CALL_STATE_DISPATCHED;
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
/*
|
|
Packit Service |
fa4841 |
* This protocol specifies that only RPC PDUs are subject to the flow control abstract
|
|
Packit Service |
fa4841 |
* data model. RTS PDUs and the HTTP request and response headers are not subject to flow
|
|
Packit Service |
fa4841 |
* control. Implementations of this protocol MUST NOT include them when computing any of the
|
|
Packit Service |
fa4841 |
* variables specified by this abstract data model.
|
|
Packit Service |
fa4841 |
*/
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
if (header->ptype == PTYPE_REQUEST)
|
|
Packit Service |
fa4841 |
{
|
|
Packit Service |
fa4841 |
inChannel->BytesSent += status;
|
|
Packit Service |
fa4841 |
inChannel->SenderAvailableWindow -= status;
|
|
Packit Service |
fa4841 |
}
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
return status;
|
|
Packit Service |
fa4841 |
}
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
BOOL rpc_client_write_call(rdpRpc* rpc, wStream* s, UINT16 opnum)
|
|
Packit Service |
fa4841 |
{
|
|
Packit Service |
fa4841 |
UINT32 offset;
|
|
Packit Service |
fa4841 |
BYTE* buffer = NULL;
|
|
Packit Service |
fa4841 |
UINT32 stub_data_pad;
|
|
Packit Service |
fa4841 |
SecBuffer Buffers[2] = { 0 };
|
|
Packit Service |
fa4841 |
SecBufferDesc Message;
|
|
Packit Service |
fa4841 |
RpcClientCall* clientCall = NULL;
|
|
Packit Service |
fa4841 |
rdpNtlm* ntlm;
|
|
Packit Service |
fa4841 |
rpcconn_request_hdr_t request_pdu = { 0 };
|
|
Packit Service |
fa4841 |
RpcVirtualConnection* connection;
|
|
Packit Service |
fa4841 |
RpcInChannel* inChannel;
|
|
Packit Service |
fa4841 |
size_t length;
|
|
Packit Service |
fa4841 |
SSIZE_T size;
|
|
Packit Service |
fa4841 |
BOOL rc = FALSE;
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
if (!s)
|
|
Packit Service |
fa4841 |
return FALSE;
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
if (!rpc)
|
|
Packit Service |
fa4841 |
goto fail;
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
ntlm = rpc->ntlm;
|
|
Packit Service |
fa4841 |
connection = rpc->VirtualConnection;
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
if (!ntlm)
|
|
Packit Service |
fa4841 |
{
|
|
Packit Service |
fa4841 |
WLog_ERR(TAG, "invalid ntlm context");
|
|
Packit Service |
fa4841 |
goto fail;
|
|
Packit Service |
fa4841 |
}
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
if (!connection)
|
|
Packit Service |
fa4841 |
goto fail;
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
inChannel = connection->DefaultInChannel;
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
if (!inChannel)
|
|
Packit Service |
fa4841 |
goto fail;
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
Stream_SealLength(s);
|
|
Packit Service |
fa4841 |
length = Stream_Length(s);
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
if (ntlm_client_query_auth_size(ntlm) < 0)
|
|
Packit Service |
fa4841 |
goto fail;
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
size = ntlm_client_get_context_max_size(ntlm);
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
if (size < 0)
|
|
Packit Service |
fa4841 |
goto fail;
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
rpc_pdu_header_init(rpc, &request_pdu.header);
|
|
Packit Service |
fa4841 |
request_pdu.header.ptype = PTYPE_REQUEST;
|
|
Packit Service |
fa4841 |
request_pdu.header.pfc_flags = PFC_FIRST_FRAG | PFC_LAST_FRAG;
|
|
Packit Service |
fa4841 |
request_pdu.header.auth_length = (UINT16)size;
|
|
Packit Service |
fa4841 |
request_pdu.header.call_id = rpc->CallId++;
|
|
Packit Service |
fa4841 |
request_pdu.alloc_hint = length;
|
|
Packit Service |
fa4841 |
request_pdu.p_cont_id = 0x0000;
|
|
Packit Service |
fa4841 |
request_pdu.opnum = opnum;
|
|
Packit Service |
fa4841 |
clientCall = rpc_client_call_new(request_pdu.header.call_id, request_pdu.opnum);
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
if (!clientCall)
|
|
Packit Service |
fa4841 |
goto fail;
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
if (ArrayList_Add(rpc->client->ClientCallList, clientCall) < 0)
|
|
Packit Service |
fa4841 |
{
|
|
Packit Service |
fa4841 |
rpc_client_call_free(clientCall);
|
|
Packit Service |
fa4841 |
goto fail;
|
|
Packit Service |
fa4841 |
}
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
if (request_pdu.opnum == TsProxySetupReceivePipeOpnum)
|
|
Packit Service |
fa4841 |
rpc->PipeCallId = request_pdu.header.call_id;
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
request_pdu.stub_data = Stream_Buffer(s);
|
|
Packit Service |
fa4841 |
offset = 24;
|
|
Packit Service |
fa4841 |
stub_data_pad = rpc_offset_align(&offset, 8);
|
|
Packit Service |
fa4841 |
offset += length;
|
|
Packit Service |
fa4841 |
request_pdu.auth_verifier.auth_pad_length = rpc_offset_align(&offset, 4);
|
|
Packit Service |
fa4841 |
request_pdu.auth_verifier.auth_type = RPC_C_AUTHN_WINNT;
|
|
Packit Service |
fa4841 |
request_pdu.auth_verifier.auth_level = RPC_C_AUTHN_LEVEL_PKT_INTEGRITY;
|
|
Packit Service |
fa4841 |
request_pdu.auth_verifier.auth_reserved = 0x00;
|
|
Packit Service |
fa4841 |
request_pdu.auth_verifier.auth_context_id = 0x00000000;
|
|
Packit Service |
fa4841 |
offset += (8 + request_pdu.header.auth_length);
|
|
Packit Service |
fa4841 |
request_pdu.header.frag_length = offset;
|
|
Packit Service |
fa4841 |
buffer = (BYTE*)calloc(1, request_pdu.header.frag_length);
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
if (!buffer)
|
|
Packit Service |
fa4841 |
goto fail;
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
CopyMemory(buffer, &request_pdu, 24);
|
|
Packit Service |
fa4841 |
offset = 24;
|
|
Packit Service |
fa4841 |
rpc_offset_pad(&offset, stub_data_pad);
|
|
Packit Service |
fa4841 |
CopyMemory(&buffer[offset], request_pdu.stub_data, length);
|
|
Packit Service |
fa4841 |
offset += length;
|
|
Packit Service |
fa4841 |
rpc_offset_pad(&offset, request_pdu.auth_verifier.auth_pad_length);
|
|
Packit Service |
fa4841 |
CopyMemory(&buffer[offset], &request_pdu.auth_verifier.auth_type, 8);
|
|
Packit Service |
fa4841 |
offset += 8;
|
|
Packit Service |
fa4841 |
Buffers[0].BufferType = SECBUFFER_DATA | SECBUFFER_READONLY; /* auth_data */
|
|
Packit Service |
fa4841 |
Buffers[1].BufferType = SECBUFFER_TOKEN; /* signature */
|
|
Packit Service |
fa4841 |
Buffers[0].pvBuffer = buffer;
|
|
Packit Service |
fa4841 |
Buffers[0].cbBuffer = offset;
|
|
Packit Service |
fa4841 |
Buffers[1].cbBuffer = size;
|
|
Packit Service |
fa4841 |
Buffers[1].pvBuffer = calloc(1, Buffers[1].cbBuffer);
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
if (!Buffers[1].pvBuffer)
|
|
Packit Service |
fa4841 |
goto fail;
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
Message.cBuffers = 2;
|
|
Packit Service |
fa4841 |
Message.ulVersion = SECBUFFER_VERSION;
|
|
Packit Service |
fa4841 |
Message.pBuffers = (PSecBuffer)&Buffers;
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
if (!ntlm_client_encrypt(ntlm, 0, &Message, rpc->SendSeqNum++))
|
|
Packit Service |
fa4841 |
goto fail;
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
CopyMemory(&buffer[offset], Buffers[1].pvBuffer, Buffers[1].cbBuffer);
|
|
Packit Service |
fa4841 |
offset += Buffers[1].cbBuffer;
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
if (rpc_in_channel_send_pdu(inChannel, buffer, request_pdu.header.frag_length) < 0)
|
|
Packit Service |
fa4841 |
goto fail;
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
rc = TRUE;
|
|
Packit Service |
fa4841 |
fail:
|
|
Packit Service |
fa4841 |
free(buffer);
|
|
Packit Service |
fa4841 |
free(Buffers[1].pvBuffer);
|
|
Packit Service |
fa4841 |
Stream_Free(s, TRUE);
|
|
Packit Service |
fa4841 |
return rc;
|
|
Packit Service |
fa4841 |
}
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
static BOOL rpc_client_resolve_gateway(rdpSettings* settings, char** host, UINT16* port,
|
|
Packit Service |
fa4841 |
BOOL* isProxy)
|
|
Packit Service |
fa4841 |
{
|
|
Packit Service |
fa4841 |
struct addrinfo* result;
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
if (!settings || !host || !port || !isProxy)
|
|
Packit Service |
fa4841 |
return FALSE;
|
|
Packit Service |
fa4841 |
else
|
|
Packit Service |
fa4841 |
{
|
|
Packit Service |
fa4841 |
const char* peerHostname = settings->GatewayHostname;
|
|
Packit Service |
fa4841 |
const char* proxyUsername = settings->ProxyUsername;
|
|
Packit Service |
fa4841 |
const char* proxyPassword = settings->ProxyPassword;
|
|
Packit Service |
fa4841 |
*port = settings->GatewayPort;
|
|
Packit Service |
fa4841 |
*isProxy = proxy_prepare(settings, &peerHostname, port, &proxyUsername, &proxyPassword);
|
|
Packit Service |
fa4841 |
result = freerdp_tcp_resolve_host(peerHostname, *port, 0);
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
if (!result)
|
|
Packit Service |
fa4841 |
return FALSE;
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
*host =
|
|
Packit Service |
fa4841 |
freerdp_tcp_address_to_string((const struct sockaddr_storage*)result->ai_addr, NULL);
|
|
Packit Service |
fa4841 |
freeaddrinfo(result);
|
|
Packit Service |
fa4841 |
return TRUE;
|
|
Packit Service |
fa4841 |
}
|
|
Packit Service |
fa4841 |
}
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
RpcClient* rpc_client_new(rdpContext* context, UINT32 max_recv_frag)
|
|
Packit Service |
fa4841 |
{
|
|
Packit Service |
fa4841 |
RpcClient* client = (RpcClient*)calloc(1, sizeof(RpcClient));
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
if (!client)
|
|
Packit Service |
fa4841 |
return NULL;
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
if (!rpc_client_resolve_gateway(context->settings, &client->host, &client->port,
|
|
Packit Service |
fa4841 |
&client->isProxy))
|
|
Packit Service |
fa4841 |
goto fail;
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
client->context = context;
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
if (!client->context)
|
|
Packit Service |
fa4841 |
goto fail;
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
client->pdu = rpc_pdu_new();
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
if (!client->pdu)
|
|
Packit Service |
fa4841 |
goto fail;
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
client->ReceiveFragment = Stream_New(NULL, max_recv_frag);
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
if (!client->ReceiveFragment)
|
|
Packit Service |
fa4841 |
goto fail;
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
client->PipeEvent = CreateEvent(NULL, TRUE, FALSE, NULL);
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
if (!client->PipeEvent)
|
|
Packit Service |
fa4841 |
goto fail;
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
if (!ringbuffer_init(&(client->ReceivePipe), 4096))
|
|
Packit Service |
fa4841 |
goto fail;
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
if (!InitializeCriticalSectionAndSpinCount(&(client->PipeLock), 4000))
|
|
Packit Service |
fa4841 |
goto fail;
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
client->ClientCallList = ArrayList_New(TRUE);
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
if (!client->ClientCallList)
|
|
Packit Service |
fa4841 |
goto fail;
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
ArrayList_Object(client->ClientCallList)->fnObjectFree = rpc_array_client_call_free;
|
|
Packit Service |
fa4841 |
return client;
|
|
Packit Service |
fa4841 |
fail:
|
|
Packit Service |
fa4841 |
rpc_client_free(client);
|
|
Packit Service |
fa4841 |
return NULL;
|
|
Packit Service |
fa4841 |
}
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
void rpc_client_free(RpcClient* client)
|
|
Packit Service |
fa4841 |
{
|
|
Packit Service |
fa4841 |
if (!client)
|
|
Packit Service |
fa4841 |
return;
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
free(client->host);
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
if (client->ReceiveFragment)
|
|
Packit Service |
fa4841 |
Stream_Free(client->ReceiveFragment, TRUE);
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
if (client->PipeEvent)
|
|
Packit Service |
fa4841 |
CloseHandle(client->PipeEvent);
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
ringbuffer_destroy(&(client->ReceivePipe));
|
|
Packit Service |
fa4841 |
DeleteCriticalSection(&(client->PipeLock));
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
if (client->pdu)
|
|
Packit Service |
fa4841 |
rpc_pdu_free(client->pdu);
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
if (client->ClientCallList)
|
|
Packit Service |
fa4841 |
ArrayList_Free(client->ClientCallList);
|
|
Packit Service |
fa4841 |
|
|
Packit Service |
fa4841 |
free(client);
|
|
Packit Service |
fa4841 |
}
|