252 lines
7.9 KiB
C
252 lines
7.9 KiB
C
|
/**
|
||
|
|
* FreeRDP: A Remote Desktop Protocol Implementation
|
||
|
|
* RDP Core
|
||
|
|
*
|
||
|
|
* Copyright 2011 Marc-Andre Moreau <marcandre.moreau@gmail.com>
|
||
|
|
* Copyright 2014 DI (FH) Martin Haimberger <martin.haimberger@thincast.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.
|
||
|
|
*/
|
||
|
|
|
||
|
|
#ifndef FREERDP_LIB_CORE_RDP_H
|
||
|
|
#define FREERDP_LIB_CORE_RDP_H
|
||
|
|
|
||
|
|
#ifdef HAVE_CONFIG_H
|
||
|
|
#include "config.h"
|
||
|
|
#endif
|
||
|
|
|
||
|
|
#include "nla.h"
|
||
|
|
#include "mcs.h"
|
||
|
|
#include "tpkt.h"
|
||
|
|
#include "bulk.h"
|
||
|
|
#include "fastpath.h"
|
||
|
|
#include "tpdu.h"
|
||
|
|
#include "nego.h"
|
||
|
|
#include "input.h"
|
||
|
|
#include "update.h"
|
||
|
|
#include "license.h"
|
||
|
|
#include "errinfo.h"
|
||
|
|
#include "autodetect.h"
|
||
|
|
#include "heartbeat.h"
|
||
|
|
#include "multitransport.h"
|
||
|
|
#include "security.h"
|
||
|
|
#include "transport.h"
|
||
|
|
#include "connection.h"
|
||
|
|
#include "redirection.h"
|
||
|
|
#include "capabilities.h"
|
||
|
|
#include "channels.h"
|
||
|
|
|
||
|
|
#include <freerdp/freerdp.h>
|
||
|
|
#include <freerdp/settings.h>
|
||
|
|
#include <freerdp/log.h>
|
||
|
|
#include <freerdp/api.h>
|
||
|
|
|
||
|
|
#include <winpr/stream.h>
|
||
|
|
#include <winpr/crypto.h>
|
||
|
|
|
||
|
|
/* Security Header Flags */
|
||
|
|
#define SEC_EXCHANGE_PKT 0x0001
|
||
|
|
#define SEC_TRANSPORT_REQ 0x0002
|
||
|
|
#define SEC_TRANSPORT_RSP 0x0004
|
||
|
|
#define SEC_ENCRYPT 0x0008
|
||
|
|
#define SEC_RESET_SEQNO 0x0010
|
||
|
|
#define SEC_IGNORE_SEQNO 0x0020
|
||
|
|
#define SEC_INFO_PKT 0x0040
|
||
|
|
#define SEC_LICENSE_PKT 0x0080
|
||
|
|
#define SEC_LICENSE_ENCRYPT_CS 0x0200
|
||
|
|
#define SEC_LICENSE_ENCRYPT_SC 0x0200
|
||
|
|
#define SEC_REDIRECTION_PKT 0x0400
|
||
|
|
#define SEC_SECURE_CHECKSUM 0x0800
|
||
|
|
#define SEC_AUTODETECT_REQ 0x1000
|
||
|
|
#define SEC_AUTODETECT_RSP 0x2000
|
||
|
|
#define SEC_HEARTBEAT 0x4000
|
||
|
|
#define SEC_FLAGSHI_VALID 0x8000
|
||
|
|
|
||
|
|
#define SEC_PKT_CS_MASK (SEC_EXCHANGE_PKT | SEC_INFO_PKT)
|
||
|
|
#define SEC_PKT_SC_MASK (SEC_LICENSE_PKT | SEC_REDIRECTION_PKT)
|
||
|
|
#define SEC_PKT_MASK (SEC_PKT_CS_MASK | SEC_PKT_SC_MASK)
|
||
|
|
|
||
|
|
#define RDP_SECURITY_HEADER_LENGTH 4
|
||
|
|
#define RDP_SHARE_CONTROL_HEADER_LENGTH 6
|
||
|
|
#define RDP_SHARE_DATA_HEADER_LENGTH 12
|
||
|
|
#define RDP_PACKET_HEADER_MAX_LENGTH (TPDU_DATA_LENGTH + MCS_SEND_DATA_HEADER_MAX_LENGTH)
|
||
|
|
|
||
|
|
#define PDU_TYPE_DEMAND_ACTIVE 0x1
|
||
|
|
#define PDU_TYPE_CONFIRM_ACTIVE 0x3
|
||
|
|
#define PDU_TYPE_DEACTIVATE_ALL 0x6
|
||
|
|
#define PDU_TYPE_DATA 0x7
|
||
|
|
#define PDU_TYPE_SERVER_REDIRECTION 0xA
|
||
|
|
|
||
|
|
#define PDU_TYPE_FLOW_TEST 0x41
|
||
|
|
#define PDU_TYPE_FLOW_RESPONSE 0x42
|
||
|
|
#define PDU_TYPE_FLOW_STOP 0x43
|
||
|
|
|
||
|
|
#define FINALIZE_SC_SYNCHRONIZE_PDU 0x01
|
||
|
|
#define FINALIZE_SC_CONTROL_COOPERATE_PDU 0x02
|
||
|
|
#define FINALIZE_SC_CONTROL_GRANTED_PDU 0x04
|
||
|
|
#define FINALIZE_SC_FONT_MAP_PDU 0x08
|
||
|
|
#define FINALIZE_SC_COMPLETE 0x0F
|
||
|
|
|
||
|
|
/* Data PDU Types */
|
||
|
|
#define DATA_PDU_TYPE_UPDATE 0x02
|
||
|
|
#define DATA_PDU_TYPE_CONTROL 0x14
|
||
|
|
#define DATA_PDU_TYPE_POINTER 0x1B
|
||
|
|
#define DATA_PDU_TYPE_INPUT 0x1C
|
||
|
|
#define DATA_PDU_TYPE_SYNCHRONIZE 0x1F
|
||
|
|
#define DATA_PDU_TYPE_REFRESH_RECT 0x21
|
||
|
|
#define DATA_PDU_TYPE_PLAY_SOUND 0x22
|
||
|
|
#define DATA_PDU_TYPE_SUPPRESS_OUTPUT 0x23
|
||
|
|
#define DATA_PDU_TYPE_SHUTDOWN_REQUEST 0x24
|
||
|
|
#define DATA_PDU_TYPE_SHUTDOWN_DENIED 0x25
|
||
|
|
#define DATA_PDU_TYPE_SAVE_SESSION_INFO 0x26
|
||
|
|
#define DATA_PDU_TYPE_FONT_LIST 0x27
|
||
|
|
#define DATA_PDU_TYPE_FONT_MAP 0x28
|
||
|
|
#define DATA_PDU_TYPE_SET_KEYBOARD_INDICATORS 0x29
|
||
|
|
#define DATA_PDU_TYPE_BITMAP_CACHE_PERSISTENT_LIST 0x2B
|
||
|
|
#define DATA_PDU_TYPE_BITMAP_CACHE_ERROR 0x2C
|
||
|
|
#define DATA_PDU_TYPE_SET_KEYBOARD_IME_STATUS 0x2D
|
||
|
|
#define DATA_PDU_TYPE_OFFSCREEN_CACHE_ERROR 0x2E
|
||
|
|
#define DATA_PDU_TYPE_SET_ERROR_INFO 0x2F
|
||
|
|
#define DATA_PDU_TYPE_DRAW_NINEGRID_ERROR 0x30
|
||
|
|
#define DATA_PDU_TYPE_DRAW_GDIPLUS_ERROR 0x31
|
||
|
|
#define DATA_PDU_TYPE_ARC_STATUS 0x32
|
||
|
|
#define DATA_PDU_TYPE_STATUS_INFO 0x36
|
||
|
|
#define DATA_PDU_TYPE_MONITOR_LAYOUT 0x37
|
||
|
|
#define DATA_PDU_TYPE_FRAME_ACKNOWLEDGE 0x38
|
||
|
|
|
||
|
|
/* Stream Identifiers */
|
||
|
|
#define STREAM_UNDEFINED 0x00
|
||
|
|
#define STREAM_LOW 0x01
|
||
|
|
#define STREAM_MED 0x02
|
||
|
|
#define STREAM_HI 0x04
|
||
|
|
|
||
|
|
struct rdp_rdp
|
||
|
|
{
|
||
|
|
CONNECTION_STATE state;
|
||
|
|
freerdp* instance;
|
||
|
|
rdpContext* context;
|
||
|
|
rdpNla* nla;
|
||
|
|
rdpMcs* mcs;
|
||
|
|
rdpNego* nego;
|
||
|
|
rdpBulk* bulk;
|
||
|
|
rdpInput* input;
|
||
|
|
rdpUpdate* update;
|
||
|
|
rdpFastPath* fastpath;
|
||
|
|
rdpLicense* license;
|
||
|
|
rdpRedirection* redirection;
|
||
|
|
rdpSettings* settings;
|
||
|
|
rdpTransport* transport;
|
||
|
|
rdpAutoDetect* autodetect;
|
||
|
|
rdpHeartbeat* heartbeat;
|
||
|
|
rdpMultitransport* multitransport;
|
||
|
|
WINPR_RC4_CTX* rc4_decrypt_key;
|
||
|
|
int decrypt_use_count;
|
||
|
|
int decrypt_checksum_use_count;
|
||
|
|
WINPR_RC4_CTX* rc4_encrypt_key;
|
||
|
|
int encrypt_use_count;
|
||
|
|
int encrypt_checksum_use_count;
|
||
|
|
WINPR_CIPHER_CTX* fips_encrypt;
|
||
|
|
WINPR_CIPHER_CTX* fips_decrypt;
|
||
|
|
UINT32 sec_flags;
|
||
|
|
BOOL do_crypt;
|
||
|
|
BOOL do_crypt_license;
|
||
|
|
BOOL do_secure_checksum;
|
||
|
|
BYTE sign_key[16];
|
||
|
|
BYTE decrypt_key[16];
|
||
|
|
BYTE encrypt_key[16];
|
||
|
|
BYTE decrypt_update_key[16];
|
||
|
|
BYTE encrypt_update_key[16];
|
||
|
|
int rc4_key_len;
|
||
|
|
BYTE fips_sign_key[20];
|
||
|
|
BYTE fips_encrypt_key[24];
|
||
|
|
BYTE fips_decrypt_key[24];
|
||
|
|
UINT32 errorInfo;
|
||
|
|
UINT32 finalize_sc_pdus;
|
||
|
|
BOOL resendFocus;
|
||
|
|
BOOL deactivation_reactivation;
|
||
|
|
BOOL AwaitCapabilities;
|
||
|
|
UINT64 inBytes;
|
||
|
|
UINT64 inPackets;
|
||
|
|
UINT64 outBytes;
|
||
|
|
UINT64 outPackets;
|
||
|
|
CRITICAL_SECTION critical;
|
||
|
|
};
|
||
|
|
|
||
|
|
FREERDP_LOCAL BOOL rdp_read_security_header(wStream* s, UINT16* flags, UINT16* length);
|
||
|
|
FREERDP_LOCAL void rdp_write_security_header(wStream* s, UINT16 flags);
|
||
|
|
|
||
|
|
FREERDP_LOCAL BOOL rdp_read_share_control_header(wStream* s, UINT16* tpktLength,
|
||
|
|
UINT16* remainingLength, UINT16* type,
|
||
|
|
UINT16* channel_id);
|
||
|
|
|
||
|
|
FREERDP_LOCAL BOOL rdp_read_share_data_header(wStream* s, UINT16* length, BYTE* type,
|
||
|
|
UINT32* share_id, BYTE* compressed_type,
|
||
|
|
UINT16* compressed_len);
|
||
|
|
|
||
|
|
FREERDP_LOCAL wStream* rdp_send_stream_init(rdpRdp* rdp);
|
||
|
|
FREERDP_LOCAL wStream* rdp_send_stream_pdu_init(rdpRdp* rdp);
|
||
|
|
|
||
|
|
FREERDP_LOCAL BOOL rdp_read_header(rdpRdp* rdp, wStream* s, UINT16* length, UINT16* channel_id);
|
||
|
|
FREERDP_LOCAL void rdp_write_header(rdpRdp* rdp, wStream* s, UINT16 length, UINT16 channel_id);
|
||
|
|
|
||
|
|
FREERDP_LOCAL BOOL rdp_send_pdu(rdpRdp* rdp, wStream* s, UINT16 type, UINT16 channel_id);
|
||
|
|
|
||
|
|
FREERDP_LOCAL wStream* rdp_data_pdu_init(rdpRdp* rdp);
|
||
|
|
FREERDP_LOCAL BOOL rdp_send_data_pdu(rdpRdp* rdp, wStream* s, BYTE type, UINT16 channel_id);
|
||
|
|
FREERDP_LOCAL int rdp_recv_data_pdu(rdpRdp* rdp, wStream* s);
|
||
|
|
|
||
|
|
FREERDP_LOCAL BOOL rdp_send(rdpRdp* rdp, wStream* s, UINT16 channelId);
|
||
|
|
|
||
|
|
FREERDP_LOCAL BOOL rdp_send_channel_data(rdpRdp* rdp, UINT16 channelId, const BYTE* data,
|
||
|
|
size_t size);
|
||
|
|
|
||
|
|
FREERDP_LOCAL wStream* rdp_message_channel_pdu_init(rdpRdp* rdp);
|
||
|
|
FREERDP_LOCAL BOOL rdp_send_message_channel_pdu(rdpRdp* rdp, wStream* s, UINT16 sec_flags);
|
||
|
|
FREERDP_LOCAL int rdp_recv_message_channel_pdu(rdpRdp* rdp, wStream* s, UINT16 securityFlags);
|
||
|
|
|
||
|
|
FREERDP_LOCAL int rdp_recv_out_of_sequence_pdu(rdpRdp* rdp, wStream* s);
|
||
|
|
|
||
|
|
FREERDP_LOCAL int rdp_recv_callback(rdpTransport* transport, wStream* s, void* extra);
|
||
|
|
|
||
|
|
FREERDP_LOCAL int rdp_check_fds(rdpRdp* rdp);
|
||
|
|
|
||
|
|
FREERDP_LOCAL rdpRdp* rdp_new(rdpContext* context);
|
||
|
|
FREERDP_LOCAL void rdp_reset(rdpRdp* rdp);
|
||
|
|
FREERDP_LOCAL void rdp_free(rdpRdp* rdp);
|
||
|
|
|
||
|
|
#define RDP_TAG FREERDP_TAG("core.rdp")
|
||
|
|
#ifdef WITH_DEBUG_RDP
|
||
|
|
#define DEBUG_RDP(...) WLog_DBG(RDP_TAG, __VA_ARGS__)
|
||
|
|
#else
|
||
|
|
#define DEBUG_RDP(...) \
|
||
|
|
do \
|
||
|
|
{ \
|
||
|
|
} while (0)
|
||
|
|
#endif
|
||
|
|
|
||
|
|
const char* data_pdu_type_to_string(UINT8 type);
|
||
|
|
const char* pdu_type_to_str(UINT16 pduType);
|
||
|
|
|
||
|
|
BOOL rdp_decrypt(rdpRdp* rdp, wStream* s, UINT16* pLength, UINT16 securityFlags);
|
||
|
|
|
||
|
|
BOOL rdp_set_error_info(rdpRdp* rdp, UINT32 errorInfo);
|
||
|
|
BOOL rdp_send_error_info(rdpRdp* rdp);
|
||
|
|
|
||
|
|
void rdp_free_rc4_encrypt_keys(rdpRdp* rdp);
|
||
|
|
BOOL rdp_reset_rc4_encrypt_keys(rdpRdp* rdp);
|
||
|
|
|
||
|
|
void rdp_free_rc4_decrypt_keys(rdpRdp* rdp);
|
||
|
|
BOOL rdp_reset_rc4_decrypt_keys(rdpRdp* rdp);
|
||
|
|
|
||
|
|
#endif /* FREERDP_LIB_CORE_RDP_H */
|