morgan/FreeRDP
1
0
Fork 0
mirror of https://github.com/morgan9e/FreeRDP synced 2026-04-15 00:44:19 +09:00
Code Issues Packages Projects Releases Wiki Activity

libfreerdp-core: added TSG state machine

Browse Source
This commit is contained in:
Marc-André Moreau
2012-11-14 22:34:45 -05:00
parent b70bfc1579
commit 93cf614ebf
2 changed files with 45 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

42
libfreerdp/core/tsg.c
View File

@@ -966,6 +966,8 @@ BOOL tsg_connect(rdpTsg* tsg, const char* hostname, UINT16 port)
DEBUG_TSG("rpc_connect success");
tsg->state = TSG_STATE_INITIAL;
/*
* Sequential processing rules for connection process:
*
@@ -1009,7 +1011,12 @@ BOOL tsg_connect(rdpTsg* tsg, const char* hostname, UINT16 port)
*/
if (!TsProxyCreateTunnel(tsg, NULL, NULL, NULL, NULL))
{
tsg->state = TSG_STATE_FINAL;
return FALSE;
}
tsg->state = TSG_STATE_CONNECTED;
/**
* Sequential processing rules for connection process (continued):
@@ -1042,7 +1049,12 @@ BOOL tsg_connect(rdpTsg* tsg, const char* hostname, UINT16 port)
*/
if (!TsProxyAuthorizeTunnel(tsg, NULL, NULL, NULL))
{
tsg->state = TSG_STATE_TUNNEL_CLOSE_PENDING;
return FALSE;
}
tsg->state = TSG_STATE_AUTHORIZED;
/**
* Sequential processing rules for connection process (continued):
@@ -1074,6 +1086,8 @@ BOOL tsg_connect(rdpTsg* tsg, const char* hostname, UINT16 port)
if (!TsProxyCreateChannel(tsg, NULL, NULL, NULL, NULL))
return FALSE;
tsg->state = TSG_STATE_CHANNEL_CREATED;
/**
* Sequential processing rules for data transfer:
*
@@ -1092,6 +1106,8 @@ BOOL tsg_connect(rdpTsg* tsg, const char* hostname, UINT16 port)
if (!TsProxySetupReceivePipe((handle_t) tsg, NULL))
return FALSE;
tsg->state = TSG_STATE_PIPE_CREATED;
return TRUE;
}
@@ -1122,13 +1138,7 @@ int tsg_read(rdpTsg* tsg, BYTE* data, UINT32 length)
else
{
status = rpc_recv_pdu(rpc);
header = (RPC_PDU_HEADER*) rpc->buffer;
if (header->frag_length == 64)
{
printf("Ignoring 64-byte length PDU (probably TsProxySetupReceivePipe return code)\n");
return tsg_read(tsg, data, length);
}
rpcconn_response_hdr_t* header = (rpcconn_response_hdr_t*) rpc->buffer;
if (!rpc_get_stub_data_info(rpc, rpc->buffer, &tsg->StubOffset, &tsg->StubLength))
{
@@ -1136,19 +1146,28 @@ int tsg_read(rdpTsg* tsg, BYTE* data, UINT32 length)
return -1;
}
printf("RPC Stub (offset: %d length: %d):\n", tsg->StubOffset, tsg->StubLength);
freerdp_hexdump(&rpc->buffer[tsg->StubOffset], tsg->StubLength);
if (header->alloc_hint == 4)
{
printf("Ignoring TsProxySetupReceivePipe Response\n");
return tsg_read(tsg, data, length);
}
tsg->PendingPdu = TRUE;
tsg->BytesAvailable = tsg->StubLength;
tsg->BytesRead = 0;
printf("RPC Stub (offset: %d length: %d):\n", tsg->StubOffset, tsg->StubLength);
freerdp_hexdump(&rpc->buffer[tsg->StubOffset], tsg->StubLength);
CopyLength = (tsg->BytesAvailable > length) ? length : tsg->BytesAvailable;
CopyMemory(data, &rpc->buffer[tsg->StubOffset + tsg->BytesRead], CopyLength);
tsg->BytesAvailable -= CopyLength;
tsg->BytesRead += CopyLength;
if (tsg->BytesAvailable < 1)
tsg->PendingPdu = FALSE;
return CopyLength;
}
}
@@ -1162,7 +1181,8 @@ rdpTsg* tsg_new(rdpTransport* transport)
{
rdpTsg* tsg;
tsg = xnew(rdpTsg);
tsg = (rdpTsg*) malloc(sizeof(rdpTsg));
ZeroMemory(tsg, sizeof(rdpTsg));
if (tsg != NULL)
{

14
libfreerdp/core/tsg.h
View File

@@ -39,12 +39,26 @@ typedef struct rdp_tsg rdpTsg;
#include <freerdp/utils/wait_obj.h>
#include <freerdp/utils/debug.h>
enum _TSG_STATE
{
TSG_STATE_INITIAL,
TSG_STATE_CONNECTED,
TSG_STATE_AUTHORIZED,
TSG_STATE_CHANNEL_CREATED,
TSG_STATE_PIPE_CREATED,
TSG_STATE_TUNNEL_CLOSE_PENDING,
TSG_STATE_CHANNEL_CLOSE_PENDING,
TSG_STATE_FINAL
};
typedef enum _TSG_STATE TSG_STATE;
struct rdp_tsg
{
rdpRpc* rpc;
UINT16 Port;
LPWSTR Hostname;
LPWSTR MachineName;
TSG_STATE state;
BOOL PendingPdu;
BOOL BytesRead;
BOOL BytesAvailable;