186 lines
4.0 KiB
C
186 lines
4.0 KiB
C
/**
|
|
* FreeRDP: A Remote Desktop Protocol Client
|
|
* RDP Core
|
|
*
|
|
* Copyright 2011 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.
|
|
*/
|
|
|
|
#include "rdp.h"
|
|
|
|
/**
|
|
* Read RDP Security Header.\n
|
|
* @msdn{cc240579}
|
|
* @param s stream
|
|
* @param flags security flags
|
|
*/
|
|
|
|
void rdp_read_security_header(STREAM* s, uint16* flags)
|
|
{
|
|
/* Basic Security Header */
|
|
stream_read_uint16(s, *flags); /* flags */
|
|
stream_seek(s, 2); /* flagsHi (unused) */
|
|
}
|
|
|
|
/**
|
|
* Write RDP Security Header.\n
|
|
* @msdn{cc240579}
|
|
* @param s stream
|
|
* @param flags security flags
|
|
*/
|
|
|
|
void rdp_write_security_header(STREAM* s, uint16 flags)
|
|
{
|
|
/* Basic Security Header */
|
|
stream_write_uint16(s, flags); /* flags */
|
|
stream_write_uint16(s, 0); /* flagsHi (unused) */
|
|
}
|
|
|
|
/**
|
|
* Initialize an RDP packet stream.\n
|
|
* @param rdp rdp module
|
|
* @return
|
|
*/
|
|
|
|
STREAM* rdp_send_stream_init(rdpRdp* rdp)
|
|
{
|
|
STREAM* s;
|
|
s = transport_send_stream_init(rdp->transport, 1024);
|
|
stream_seek(s, RDP_PACKET_HEADER_LENGTH);
|
|
return s;
|
|
}
|
|
|
|
/**
|
|
* Write an RDP packet header.\n
|
|
* @param rdp rdp module
|
|
* @param s stream
|
|
* @param length RDP packet length
|
|
*/
|
|
|
|
void rdp_write_header(rdpRdp* rdp, STREAM* s, int length)
|
|
{
|
|
mcs_write_domain_mcspdu_header(s, DomainMCSPDU_SendDataRequest, length);
|
|
per_write_integer16(s, rdp->mcs->user_id, MCS_BASE_CHANNEL_ID); /* initiator */
|
|
per_write_integer16(s, MCS_GLOBAL_CHANNEL_ID, 0); /* channelId */
|
|
stream_write_uint8(s, 0x70); /* dataPriority + segmentation */
|
|
per_write_length(s, length - RDP_PACKET_HEADER_LENGTH); /* userData (OCTET_STRING) */
|
|
}
|
|
|
|
/**
|
|
* Send an RDP packet.\n
|
|
* @param rdp RDP module
|
|
* @param s stream
|
|
*/
|
|
|
|
void rdp_send(rdpRdp* rdp, STREAM* s)
|
|
{
|
|
int length;
|
|
|
|
length = stream_get_length(s);
|
|
stream_set_pos(s, 0);
|
|
|
|
rdp_write_header(rdp, s, length);
|
|
|
|
stream_set_pos(s, length);
|
|
transport_write(rdp->transport, s);
|
|
}
|
|
|
|
/**
|
|
* Receive an RDP packet.\n
|
|
* @param rdp RDP module
|
|
*/
|
|
|
|
void rdp_recv(rdpRdp* rdp)
|
|
{
|
|
STREAM* s;
|
|
int length;
|
|
int pduLength;
|
|
uint16 initiator;
|
|
uint16 channelId;
|
|
uint16 sec_flags;
|
|
enum DomainMCSPDU MCSPDU;
|
|
|
|
s = transport_recv_stream_init(rdp->transport, 4096);
|
|
transport_read(rdp->transport, s);
|
|
|
|
MCSPDU = DomainMCSPDU_SendDataIndication;
|
|
mcs_read_domain_mcspdu_header(s, &MCSPDU, &length);
|
|
|
|
per_read_integer16(s, &initiator, MCS_BASE_CHANNEL_ID); /* initiator (UserId) */
|
|
per_read_integer16(s, &channelId, 0); /* channelId */
|
|
stream_seek(s, 1); /* dataPriority + Segmentation (0x70) */
|
|
per_read_length(s, &pduLength); /* userData (OCTET_STRING) */
|
|
|
|
rdp_read_security_header(s, &sec_flags);
|
|
|
|
if (sec_flags & SEC_PKT_MASK)
|
|
{
|
|
switch (sec_flags & SEC_PKT_MASK)
|
|
{
|
|
case SEC_LICENSE_PKT:
|
|
license_recv(rdp->license, s);
|
|
break;
|
|
|
|
case SEC_REDIRECTION_PKT:
|
|
rdp_read_redirection_packet(rdp, s);
|
|
break;
|
|
|
|
default:
|
|
printf("incorrect security flags: 0x%02X\n", sec_flags);
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Instantiate new RDP module.
|
|
* @return new RDP module
|
|
*/
|
|
|
|
rdpRdp* rdp_new()
|
|
{
|
|
rdpRdp* rdp;
|
|
|
|
rdp = (rdpRdp*) xzalloc(sizeof(rdpRdp));
|
|
|
|
if (rdp != NULL)
|
|
{
|
|
rdp->settings = settings_new();
|
|
rdp->transport = transport_new(rdp->settings);
|
|
rdp->license = license_new(rdp);
|
|
rdp->nego = nego_new(rdp->transport);
|
|
rdp->mcs = mcs_new(rdp->transport);
|
|
}
|
|
|
|
return rdp;
|
|
}
|
|
|
|
/**
|
|
* Free RDP module.
|
|
* @param rdp RDP module to be freed
|
|
*/
|
|
|
|
void rdp_free(rdpRdp* rdp)
|
|
{
|
|
if (rdp != NULL)
|
|
{
|
|
settings_free(rdp->settings);
|
|
transport_free(rdp->transport);
|
|
license_free(rdp->license);
|
|
mcs_free(rdp->mcs);
|
|
xfree(rdp);
|
|
}
|
|
}
|
|
|