/**

 * 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

{

	int 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__)

extern const char* DATA_PDU_TYPE_STRINGS[80];

#else

#define DEBUG_RDP(...) \

	do                 \

	{                  \

	} while (0)

#endif

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);

#endif /* FREERDP_LIB_CORE_RDP_H */