/**
 * FreeRDP: A Remote Desktop Protocol Implementation
 * RDP Protocol Security Negotiation
 *
 * Copyright 2011-2012 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.
 */

#ifndef FREERDP_LIB_CORE_NEGO_H
#define FREERDP_LIB_CORE_NEGO_H

#include "transport.h"

#include <freerdp/types.h>
#include <freerdp/settings.h>
#include <freerdp/log.h>
#include <freerdp/api.h>

#include <winpr/stream.h>

/* Protocol Security Negotiation Protocols
 * [MS-RDPBCGR] 2.2.1.1.1 RDP Negotiation Request (RDP_NEG_REQ)
 */
#define PROTOCOL_RDP 0x00000000
#define PROTOCOL_SSL 0x00000001
#define PROTOCOL_HYBRID 0x00000002
#define PROTOCOL_RDSTLS 0x00000004
#define PROTOCOL_HYBRID_EX 0x00000008
#define PROTOCOL_RDSAAD 0x00000010

#define PROTOCOL_FAILED_NEGO 0x80000000 /* only used internally, not on the wire */

/* Protocol Security Negotiation Failure Codes */
enum RDP_NEG_FAILURE_FAILURECODES
{
	SSL_REQUIRED_BY_SERVER = 0x00000001,
	SSL_NOT_ALLOWED_BY_SERVER = 0x00000002,
	SSL_CERT_NOT_ON_SERVER = 0x00000003,
	INCONSISTENT_FLAGS = 0x00000004,
	HYBRID_REQUIRED_BY_SERVER = 0x00000005,
	SSL_WITH_USER_AUTH_REQUIRED_BY_SERVER = 0x00000006
};

typedef enum
{
	NEGO_STATE_INITIAL,
	NEGO_STATE_RDSTLS, /* RDSTLS (TLS implicit) */
	NEGO_STATE_AAD,    /* Azure AD Authentication (TLS implicit) */
	NEGO_STATE_EXT,    /* Extended NLA (NLA + TLS implicit) */
	NEGO_STATE_NLA,    /* Network Level Authentication (TLS implicit) */
	NEGO_STATE_TLS,    /* TLS Encryption without NLA */
	NEGO_STATE_RDP,    /* Standard Legacy RDP Encryption */
	NEGO_STATE_FAIL,   /* Negotiation failure */
	NEGO_STATE_FINAL
} NEGO_STATE;

/* RDP Negotiation Messages */
enum RDP_NEG_MSG
{
	/* X224_TPDU_CONNECTION_REQUEST */
	TYPE_RDP_NEG_REQ = 0x1,
	/* X224_TPDU_CONNECTION_CONFIRM */
	TYPE_RDP_NEG_RSP = 0x2,
	TYPE_RDP_NEG_FAILURE = 0x3,
	TYPE_RDP_CORRELATION_INFO = 0x6
};

typedef enum
{
	EXTENDED_CLIENT_DATA_SUPPORTED = 0x01,
	DYNVC_GFX_PROTOCOL_SUPPORTED = 0x02,
	RDP_NEGRSP_RESERVED = 0x04,
	RESTRICTED_ADMIN_MODE_SUPPORTED = 0x08,
	REDIRECTED_AUTHENTICATION_MODE_SUPPORTED = 0x10
} RdpNegRespFlags;

#define PRECONNECTION_PDU_V1_SIZE 16
#define PRECONNECTION_PDU_V2_MIN_SIZE (PRECONNECTION_PDU_V1_SIZE + 2)

#define PRECONNECTION_PDU_V1 1
#define PRECONNECTION_PDU_V2 2

#define RESTRICTED_ADMIN_MODE_REQUIRED 0x01
#define REDIRECTED_AUTHENTICATION_MODE_REQUIRED 0x02
#define CORRELATION_INFO_PRESENT 0x08

typedef struct rdp_nego rdpNego;

WINPR_ATTR_NODISCARD
FREERDP_LOCAL BOOL nego_connect(rdpNego* nego);

WINPR_ATTR_NODISCARD
FREERDP_LOCAL BOOL nego_disconnect(rdpNego* nego);

WINPR_ATTR_NODISCARD
FREERDP_LOCAL int nego_recv(rdpTransport* transport, wStream* s, void* extra);

WINPR_ATTR_NODISCARD
FREERDP_LOCAL BOOL nego_read_request(rdpNego* nego, wStream* s);

WINPR_ATTR_NODISCARD
FREERDP_LOCAL BOOL nego_send_negotiation_request(rdpNego* nego);

WINPR_ATTR_NODISCARD
FREERDP_LOCAL BOOL nego_send_negotiation_response(rdpNego* nego);

FREERDP_LOCAL void nego_free(rdpNego* nego);

WINPR_ATTR_MALLOC(nego_free, 1)
WINPR_ATTR_NODISCARD
FREERDP_LOCAL rdpNego* nego_new(rdpTransport* transport);

FREERDP_LOCAL void nego_init(rdpNego* nego);

FREERDP_LOCAL BOOL nego_set_target(rdpNego* nego, const char* hostname, UINT16 port);

FREERDP_LOCAL void nego_set_negotiation_enabled(rdpNego* nego, BOOL NegotiateSecurityLayer);
FREERDP_LOCAL void nego_set_restricted_admin_mode_required(rdpNego* nego,
                                                           BOOL RestrictedAdminModeRequired);
FREERDP_LOCAL void nego_set_restricted_admin_mode_supported(rdpNego* nego, BOOL enabled);
FREERDP_LOCAL void nego_set_RCG_required(rdpNego* nego, BOOL enabled);
FREERDP_LOCAL void nego_set_RCG_supported(rdpNego* nego, BOOL enabled);

WINPR_ATTR_NODISCARD
FREERDP_LOCAL BOOL nego_get_remoteCredentialGuard(const rdpNego* nego);

FREERDP_LOCAL void nego_set_childsession_enabled(rdpNego* nego, BOOL ChildSessionEnabled);
FREERDP_LOCAL void nego_set_gateway_enabled(rdpNego* nego, BOOL GatewayEnabled);
FREERDP_LOCAL void nego_set_gateway_bypass_local(rdpNego* nego, BOOL GatewayBypassLocal);
FREERDP_LOCAL void nego_enable_rdp(rdpNego* nego, BOOL enable_rdp);
FREERDP_LOCAL void nego_enable_tls(rdpNego* nego, BOOL enable_tls);
FREERDP_LOCAL void nego_enable_nla(rdpNego* nego, BOOL enable_nla);
FREERDP_LOCAL void nego_enable_rdstls(rdpNego* nego, BOOL enable_rdstls);
FREERDP_LOCAL void nego_enable_aad(rdpNego* nego, BOOL enable_aad);
FREERDP_LOCAL void nego_enable_ext(rdpNego* nego, BOOL enable_ext);

WINPR_ATTR_NODISCARD
FREERDP_LOCAL const BYTE* nego_get_routing_token(const rdpNego* nego, DWORD* RoutingTokenLength);

WINPR_ATTR_NODISCARD
FREERDP_LOCAL BOOL nego_set_routing_token(rdpNego* nego, const void* RoutingToken,
                                          DWORD RoutingTokenLength);

WINPR_ATTR_NODISCARD
FREERDP_LOCAL BOOL nego_set_cookie(rdpNego* nego, const char* cookie);

FREERDP_LOCAL void nego_set_cookie_max_length(rdpNego* nego, UINT32 CookieMaxLength);
FREERDP_LOCAL void nego_set_send_preconnection_pdu(rdpNego* nego, BOOL SendPreconnectionPdu);
FREERDP_LOCAL void nego_set_preconnection_id(rdpNego* nego, UINT32 PreconnectionId);
FREERDP_LOCAL void nego_set_preconnection_blob(rdpNego* nego, const char* PreconnectionBlob);

WINPR_ATTR_NODISCARD
FREERDP_LOCAL UINT32 nego_get_selected_protocol(const rdpNego* nego);

WINPR_ATTR_NODISCARD
FREERDP_LOCAL BOOL nego_set_selected_protocol(rdpNego* nego, UINT32 SelectedProtocol);

WINPR_ATTR_NODISCARD
FREERDP_LOCAL UINT32 nego_get_requested_protocols(const rdpNego* nego);

WINPR_ATTR_NODISCARD
FREERDP_LOCAL BOOL nego_set_requested_protocols(rdpNego* nego, UINT32 RequestedProtocols);

WINPR_ATTR_NODISCARD
FREERDP_LOCAL BOOL nego_update_settings_from_state(rdpNego* nego, rdpSettings* settings);

FREERDP_LOCAL BOOL nego_set_state(rdpNego* nego, NEGO_STATE state);

WINPR_ATTR_NODISCARD
FREERDP_LOCAL NEGO_STATE nego_get_state(const rdpNego* nego);

WINPR_ATTR_NODISCARD
FREERDP_LOCAL SEC_WINNT_AUTH_IDENTITY* nego_get_identity(rdpNego* nego);

FREERDP_LOCAL void nego_free_nla(rdpNego* nego);

WINPR_ATTR_NODISCARD
FREERDP_LOCAL const char* nego_protocol_to_str(UINT32 protocol, char* buffer, size_t size);

#endif /* FREERDP_LIB_CORE_NEGO_H */
