/**
 * FreeRDP: A Remote Desktop Protocol Implementation
 * FreeRDP Core
 *
 * Copyright 2011 Marc-Andre Moreau <marcandre.moreau@gmail.com>
 * Copyright 2015 Thincast Technologies GmbH
 * Copyright 2015 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.
 */

#include <freerdp/config.h>

#include "settings.h"

#include <string.h>
#include <stdarg.h>
#include <time.h>

#include "rdp.h"
#include "input.h"
#include "update.h"
#include "surface.h"
#include "transport.h"
#include "connection.h"
#include "message.h"
#include <freerdp/buildflags.h>
#include "gateway/rpc_fault.h"

#include <winpr/assert.h>

#include <winpr/crt.h>
#include <winpr/string.h>
#include <winpr/stream.h>
#include <winpr/wtsapi.h>
#include <winpr/ssl.h>
#include <winpr/debug.h>

#include <freerdp/freerdp.h>
#include <freerdp/streamdump.h>
#include <freerdp/error.h>
#include <freerdp/event.h>
#include <freerdp/locale/keyboard.h>
#include <freerdp/locale/locale.h>
#include <freerdp/channels/channels.h>
#include <freerdp/version.h>
#include <freerdp/log.h>
#include <freerdp/utils/signal.h>

#include "../cache/pointer.h"
#include "utils.h"

#define TAG FREERDP_TAG("core")

static void sig_abort_connect(int signum, const char* signame, void* ctx)
{
	rdpContext* context = (rdpContext*)ctx;

	WLog_INFO(TAG, "Signal %s [%d], terminating session %p", signame, signum,
	          WINPR_CXX_COMPAT_CAST(const void*, context));
	if (context)
		freerdp_abort_connect_context(context);
}

/** Creates a new connection based on the settings found in the "instance" parameter
 *  It will use the callbacks registered on the structure to process the pre/post connect operations
 *  that the caller requires.
 *  @see struct rdp_freerdp in freerdp.h
 *
 *  @param instance - pointer to a rdp_freerdp structure that contains base information to establish
 * the connection. On return, this function will be initialized with the new connection's settings.
 *
 *  @return TRUE if successful. FALSE otherwise.
 *
 */
static int freerdp_connect_begin(freerdp* instance)
{
	BOOL rc = 0;
	rdpRdp* rdp = nullptr;
	BOOL status = TRUE;
	rdpSettings* settings = nullptr;

	if (!instance)
		return -1;

	WINPR_ASSERT(instance->context);

	/* We always set the return code to 0 before we start the connect sequence*/
	instance->ConnectionCallbackState = CLIENT_STATE_INITIAL;
	freerdp_set_last_error_log(instance->context, FREERDP_ERROR_SUCCESS);
	clearChannelError(instance->context);
	if (!utils_reset_abort(instance->context->rdp))
		return -1;

	rdp = instance->context->rdp;
	WINPR_ASSERT(rdp);

	settings = instance->context->settings;
	WINPR_ASSERT(settings);

	freerdp_channels_register_instance(instance->context->channels, instance);

	if (!freerdp_settings_set_default_order_support(settings))
		return -1;

	if (!freerdp_add_signal_cleanup_handler(instance->context, sig_abort_connect))
		return -1;

	IFCALLRET(instance->PreConnect, status, instance);
	instance->ConnectionCallbackState = CLIENT_STATE_PRECONNECT_PASSED;

	freerdp_settings_print_warnings(settings);
	if (status)
		status = freerdp_settings_enforce_monitor_exists(settings);

	if (status)
		status = freerdp_settings_enforce_consistency(settings);

	if (status)
		status = freerdp_settings_check_client_after_preconnect(settings);

	if (status)
		status = rdp_set_backup_settings(rdp);
	if (status)
		status = utils_reload_channels(instance->context);

	const UINT32 cp = freerdp_settings_get_uint32(settings, FreeRDP_KeyboardCodePage);
	int64_t KeyboardLayout = freerdp_get_keyboard_default_layout_for_locale(cp);
	if (KeyboardLayout == 0)
		KeyboardLayout = freerdp_settings_get_uint32(settings, FreeRDP_KeyboardLayout);

	switch (KeyboardLayout)
	{
		case KBD_JAPANESE:
		case KBD_JAPANESE_INPUT_SYSTEM_MS_IME2002:
		{
			if (!freerdp_settings_set_uint32(settings, FreeRDP_KeyboardType,
			                                 WINPR_KBD_TYPE_JAPANESE))
				return -1;
			if (!freerdp_settings_set_uint32(settings, FreeRDP_KeyboardSubType, 2))
				return -1;
			if (!freerdp_settings_set_uint32(settings, FreeRDP_KeyboardFunctionKey, 12))
				return -1;
		}
		break;
		default:
			break;
	}

	if (!status)
	{
		rdpContext* context = instance->context;
		WINPR_ASSERT(context);
		freerdp_set_last_error_if_not(context, FREERDP_ERROR_PRE_CONNECT_FAILED);

		WLog_Print(context->log, WLOG_ERROR, "freerdp_pre_connect failed: %s",
		           rdp_client_connection_state_string(instance->ConnectionCallbackState));
		return 0;
	}

	rc = rdp_client_connect(rdp);

	/* --authonly tests the connection without a UI */
	if (freerdp_settings_get_bool(rdp->settings, FreeRDP_AuthenticationOnly))
	{
		rdpContext* context = rdp->context;
		WINPR_ASSERT(context);
		WLog_Print(context->log, WLOG_ERROR, "Authentication only, exit status %" PRId32 "", rc);
		return 0;
	}

	return rc ? 1 : 0;
}

BOOL freerdp_connect(freerdp* instance)
{
	BOOL status = FALSE;
	ConnectionResultEventArgs e = WINPR_C_ARRAY_INIT;
	const int rc = freerdp_connect_begin(instance);
	rdpRdp* rdp = nullptr;
	UINT status2 = ERROR_INTERNAL_ERROR;

	WINPR_ASSERT(instance);
	WINPR_ASSERT(instance->context);

	rdp = instance->context->rdp;
	WINPR_ASSERT(rdp);
	WINPR_ASSERT(rdp->settings);

	if (rc > 0)
	/* Pointers might have changed in between */
	{
		rdp_update_internal* up = update_cast(rdp->update);

		if (freerdp_settings_get_bool(rdp->settings, FreeRDP_DumpRemoteFx))
		{
			up->pcap_rfx = pcap_open(
			    freerdp_settings_get_string(rdp->settings, FreeRDP_DumpRemoteFxFile), TRUE);

			if (up->pcap_rfx)
				up->dump_rfx = TRUE;
		}

		pointer_cache_register_callbacks(instance->context->update);
		status = IFCALLRESULT(TRUE, instance->PostConnect, instance);
		instance->ConnectionCallbackState = CLIENT_STATE_POSTCONNECT_PASSED;

		if (status)
			status2 = freerdp_channels_post_connect(instance->context->channels, instance);
	}
	else
	{
		status2 = CHANNEL_RC_OK;
		if (freerdp_get_last_error(instance->context) == FREERDP_ERROR_CONNECT_TRANSPORT_FAILED)
			status = freerdp_reconnect(instance);
		else
			goto freerdp_connect_finally;
	}

	if (!status || (status2 != CHANNEL_RC_OK) || !update_post_connect(instance->context->update))
	{
		rdpContext* context = instance->context;
		WINPR_ASSERT(context);
		WLog_Print(context->log, WLOG_ERROR, "freerdp_post_connect failed");

		freerdp_set_last_error_if_not(context, FREERDP_ERROR_POST_CONNECT_FAILED);

		status = FALSE;
		goto freerdp_connect_finally;
	}

	if (freerdp_settings_get_bool(rdp->settings, FreeRDP_PlayRemoteFx))
	{
		wStream* s = nullptr;
		rdp_update_internal* update = update_cast(instance->context->update);
		pcap_record record = WINPR_C_ARRAY_INIT;

		WINPR_ASSERT(update);
		const char* PlayRemoteFxFile =
		    freerdp_settings_get_string(rdp->settings, FreeRDP_PlayRemoteFxFile);
		update->pcap_rfx = pcap_open(PlayRemoteFxFile, FALSE);
		status = FALSE;

		if (!update->pcap_rfx)
			goto freerdp_connect_finally;
		else
			update->play_rfx = TRUE;

		status = TRUE;

		while (pcap_has_next_record(update->pcap_rfx) && status)
		{
			if (!pcap_get_next_record_header(update->pcap_rfx, &record))
				break;

			s = transport_take_from_pool(rdp->transport, record.length);
			if (!s)
				break;

			record.data = Stream_Buffer(s);
			if (!pcap_get_next_record_content(update->pcap_rfx, &record))
				break;
			if (!Stream_SetLength(s, record.length))
			{
				status = FALSE;
				continue;
			}
			Stream_ResetPosition(s);

			if (!update_begin_paint(&update->common))
				status = FALSE;
			else
			{
				if (!update_recv_surfcmds(&update->common, s))
					status = FALSE;

				if (!update_end_paint(&update->common))
					status = FALSE;
			}

			Stream_Release(s);
		}

		pcap_close(update->pcap_rfx);
		update->pcap_rfx = nullptr;
		goto freerdp_connect_finally;
	}

	if (rdp->errorInfo == ERRINFO_SERVER_INSUFFICIENT_PRIVILEGES)
		freerdp_set_last_error_log(instance->context, FREERDP_ERROR_INSUFFICIENT_PRIVILEGES);

	if (status)
		status = transport_set_connected_event(rdp->transport);

freerdp_connect_finally:
	EventArgsInit(&e, "freerdp");
	e.result = status ? 0 : -1;
	if (PubSub_OnConnectionResult(rdp->pubSub, instance->context, &e) < 0)
		return FALSE;

	if (!status)
		freerdp_disconnect(instance);

	return status;
}

#if !defined(WITHOUT_FREERDP_3x_DEPRECATED)
BOOL freerdp_abort_connect(freerdp* instance)
{
	if (!instance)
		return FALSE;

	return freerdp_abort_connect_context(instance->context);
}
#endif

BOOL freerdp_abort_connect_context(rdpContext* context)
{
	if (!context)
		return FALSE;

	freerdp_set_last_error_if_not(context, FREERDP_ERROR_CONNECT_CANCELLED);
	return utils_abort_connect(context->rdp);
}

#if defined(WITH_FREERDP_DEPRECATED)
BOOL freerdp_get_fds(freerdp* instance, void** rfds, int* rcount, WINPR_ATTR_UNUSED void** wfds,
                     WINPR_ATTR_UNUSED int* wcount)
{
	rdpRdp* rdp = nullptr;

	WINPR_ASSERT(instance);
	WINPR_ASSERT(instance->context);

	rdp = instance->context->rdp;
	WINPR_ASSERT(rdp);

	transport_get_fds(rdp->transport, rfds, rcount);
	return TRUE;
}
#endif

BOOL freerdp_check_fds(freerdp* instance)
{
	int status = 0;
	rdpRdp* rdp = nullptr;

	if (!instance)
		return FALSE;

	if (!instance->context)
		return FALSE;

	if (!instance->context->rdp)
		return FALSE;

	rdp = instance->context->rdp;
	status = rdp_check_fds(rdp);

	if (status < 0)
	{
		TerminateEventArgs e;
		rdpContext* context = instance->context;
		WINPR_ASSERT(context);

		WLog_Print(context->log, WLOG_DEBUG, "rdp_check_fds() - %i", status);
		EventArgsInit(&e, "freerdp");
		e.code = 0;
		if (PubSub_OnTerminate(rdp->pubSub, context, &e) < 0)
			return FALSE;
		return FALSE;
	}

	return TRUE;
}

DWORD freerdp_get_event_handles(rdpContext* context, HANDLE* events, DWORD count)
{
	DWORD nCount = 0;

	WINPR_ASSERT(context);
	WINPR_ASSERT(context->rdp);
	WINPR_ASSERT(events || (count == 0));

	const size_t rrc = rdp_get_event_handles(context->rdp, &events[nCount], count - nCount);
	if (rrc == 0)
		return 0;

	nCount += WINPR_ASSERTING_INT_CAST(uint32_t, rrc);

	if (events && (nCount < count + 2))
	{
		events[nCount++] = freerdp_channels_get_event_handle(context->instance);
		events[nCount++] = getChannelErrorEventHandle(context);
	}
	else
		return 0;

	const SSIZE_T rc = freerdp_client_channel_get_registered_event_handles(
	    context->channels, &events[nCount], count - nCount);
	if (rc < 0)
		return 0;
	return nCount + (DWORD)rc;
}

/* Resend mouse cursor position to prevent session lock in prevent-session-lock mode */
static BOOL freerdp_prevent_session_lock(rdpContext* context)
{
	WINPR_ASSERT(context);
	WINPR_ASSERT(context->input);

	rdp_input_internal* in = input_cast(context->input);

	UINT32 FakeMouseMotionInterval =
	    freerdp_settings_get_uint32(context->settings, FreeRDP_FakeMouseMotionInterval);
	if (FakeMouseMotionInterval && in->lastInputTimestamp)
	{
		const time_t now = time(nullptr);
		if (WINPR_ASSERTING_INT_CAST(size_t, now) - in->lastInputTimestamp >
		    FakeMouseMotionInterval)
		{
			WLog_Print(context->log, WLOG_DEBUG,
			           "fake mouse move: x=%d y=%d lastInputTimestamp=%" PRIu64 " "
			           "FakeMouseMotionInterval=%" PRIu32,
			           in->lastX, in->lastY, in->lastInputTimestamp, FakeMouseMotionInterval);

			BOOL status = freerdp_input_send_mouse_event(context->input, PTR_FLAGS_MOVE, in->lastX,
			                                             in->lastY);
			if (!status)
			{
				if (freerdp_get_last_error(context) == FREERDP_ERROR_SUCCESS)
					WLog_Print(context->log, WLOG_ERROR,
					           "freerdp_prevent_session_lock() failed - %" PRIi32 "", status);

				return FALSE;
			}

			return status;
		}
	}

	return TRUE;
}

BOOL freerdp_check_event_handles(rdpContext* context)
{
	WINPR_ASSERT(context);

	BOOL status = freerdp_check_fds(context->instance);

	if (!status)
	{
		if (freerdp_get_last_error(context) == FREERDP_ERROR_SUCCESS)
			WLog_Print(context->log, WLOG_ERROR, "freerdp_check_fds() failed - %" PRIi32 "",
			           status);

		return FALSE;
	}

	status = freerdp_channels_check_fds(context->channels, context->instance);

	if (!status)
	{
		if (freerdp_get_last_error(context) == FREERDP_ERROR_SUCCESS)
			WLog_Print(context->log, WLOG_ERROR,
			           "freerdp_channels_check_fds() failed - %" PRIi32 "", status);

		return FALSE;
	}

	status = checkChannelErrorEvent(context);

	if (!status)
	{
		if (freerdp_get_last_error(context) == FREERDP_ERROR_SUCCESS)
			WLog_Print(context->log, WLOG_ERROR, "checkChannelErrorEvent() failed - %" PRIi32 "",
			           status);

		return FALSE;
	}

	status = freerdp_prevent_session_lock(context);

	return status;
}

wMessageQueue* freerdp_get_message_queue(freerdp* instance, DWORD id)
{
	wMessageQueue* queue = nullptr;

	WINPR_ASSERT(instance);

	rdpContext* context = instance->context;
	WINPR_ASSERT(context);

	switch (id)
	{
		case FREERDP_UPDATE_MESSAGE_QUEUE:
		{
			rdp_update_internal* update = update_cast(context->update);
			queue = update->queue;
		}
		break;

		case FREERDP_INPUT_MESSAGE_QUEUE:
		{
			rdp_input_internal* input = input_cast(context->input);
			queue = input->queue;
		}
		break;
		default:
			break;
	}

	return queue;
}

HANDLE freerdp_get_message_queue_event_handle(freerdp* instance, DWORD id)
{
	HANDLE event = nullptr;
	wMessageQueue* queue = freerdp_get_message_queue(instance, id);

	if (queue)
		event = MessageQueue_Event(queue);

	return event;
}

int freerdp_message_queue_process_message(freerdp* instance, DWORD id, wMessage* message)
{
	int status = -1;
	rdpContext* context = nullptr;

	WINPR_ASSERT(instance);

	context = instance->context;
	WINPR_ASSERT(context);

	switch (id)
	{
		case FREERDP_UPDATE_MESSAGE_QUEUE:
			status = update_message_queue_process_message(context->update, message);
			break;

		case FREERDP_INPUT_MESSAGE_QUEUE:
			status = input_message_queue_process_message(context->input, message);
			break;
		default:
			break;
	}

	return status;
}

int freerdp_message_queue_process_pending_messages(freerdp* instance, DWORD id)
{
	int status = -1;
	rdpContext* context = nullptr;

	WINPR_ASSERT(instance);

	context = instance->context;
	WINPR_ASSERT(context);

	switch (id)
	{
		case FREERDP_UPDATE_MESSAGE_QUEUE:
			status = update_message_queue_process_pending_messages(context->update);
			break;

		case FREERDP_INPUT_MESSAGE_QUEUE:
			status = input_message_queue_process_pending_messages(context->input);
			break;
		default:
			break;
	}

	return status;
}

static BOOL freerdp_send_channel_data(freerdp* instance, UINT16 channelId, const BYTE* data,
                                      size_t size)
{
	WINPR_ASSERT(instance);
	WINPR_ASSERT(instance->context);
	WINPR_ASSERT(instance->context->rdp);
	return rdp_send_channel_data(instance->context->rdp, channelId, data, size);
}

static BOOL freerdp_send_channel_packet(freerdp* instance, UINT16 channelId, size_t totalSize,
                                        UINT32 flags, const BYTE* data, size_t chunkSize)
{
	WINPR_ASSERT(instance);
	WINPR_ASSERT(instance->context);
	WINPR_ASSERT(instance->context->rdp);
	return rdp_channel_send_packet(instance->context->rdp, channelId, totalSize, flags, data,
	                               chunkSize);
}

BOOL freerdp_disconnect(freerdp* instance)
{
	BOOL rc = TRUE;

	if (!instance || !instance->context)
		return FALSE;

	rdpRdp* rdp = instance->context->rdp;
	if (rdp)
	{
		/* Try to send a [MS-RDPBCGR] 1.3.1.4.1 User-Initiated on Client PDU, we don't care about
		 * success */
		if (freerdp_get_last_error(instance->context) == FREERDP_ERROR_CONNECT_CANCELLED)
		{
			(void)mcs_send_disconnect_provider_ultimatum(rdp->mcs,
			                                             Disconnect_Ultimatum_user_requested);
		}
	}

	utils_abort_connect(rdp);

	if (!rdp_client_disconnect(rdp))
		rc = FALSE;

	rdp_update_internal* up = nullptr;
	if (rdp && rdp->update)
	{
		up = update_cast(rdp->update);

		update_post_disconnect(rdp->update);
	}

	IFCALL(instance->PostDisconnect, instance);

	if (up)
	{
		if (up->pcap_rfx)
		{
			up->dump_rfx = FALSE;
			pcap_close(up->pcap_rfx);
			up->pcap_rfx = nullptr;
		}
	}

	if (instance->context->channels)
		freerdp_channels_close(instance->context->channels, instance);

	IFCALL(instance->PostFinalDisconnect, instance);

	freerdp_del_signal_cleanup_handler(instance->context, sig_abort_connect);
	return rc;
}

#if !defined(WITHOUT_FREERDP_3x_DEPRECATED)
BOOL freerdp_disconnect_before_reconnect(freerdp* instance)
{
	WINPR_ASSERT(instance);
	return freerdp_disconnect_before_reconnect_context(instance->context);
}
#endif

BOOL freerdp_disconnect_before_reconnect_context(rdpContext* context)
{
	rdpRdp* rdp = nullptr;

	WINPR_ASSERT(context);

	rdp = context->rdp;
	return rdp_client_disconnect_and_clear(rdp);
}

BOOL freerdp_reconnect(freerdp* instance)
{
	rdpRdp* rdp = nullptr;

	WINPR_ASSERT(instance);
	WINPR_ASSERT(instance->context);

	if (freerdp_get_last_error(instance->context) == FREERDP_ERROR_CONNECT_CANCELLED)
		return FALSE;

	rdp = instance->context->rdp;

	if (!utils_reset_abort(instance->context->rdp))
		return FALSE;
	return rdp_client_reconnect(rdp);
}

#if !defined(WITHOUT_FREERDP_3x_DEPRECATED)
BOOL freerdp_shall_disconnect(const freerdp* instance)
{
	if (!instance)
		return FALSE;

	return freerdp_shall_disconnect_context(instance->context);
}
#endif

BOOL freerdp_shall_disconnect_context(const rdpContext* context)
{
	if (!context)
		return FALSE;

	return utils_abort_event_is_set(context->rdp);
}

BOOL freerdp_focus_required(freerdp* instance)
{
	rdpRdp* rdp = nullptr;
	BOOL bRetCode = FALSE;

	WINPR_ASSERT(instance);
	WINPR_ASSERT(instance->context);

	rdp = instance->context->rdp;
	WINPR_ASSERT(rdp);

	if (rdp->resendFocus)
	{
		bRetCode = TRUE;
		rdp->resendFocus = FALSE;
	}

	return bRetCode;
}

void freerdp_set_focus(freerdp* instance)
{
	rdpRdp* rdp = nullptr;

	WINPR_ASSERT(instance);
	WINPR_ASSERT(instance->context);

	rdp = instance->context->rdp;
	WINPR_ASSERT(rdp);

	rdp->resendFocus = TRUE;
}

void freerdp_get_version(int* major, int* minor, int* revision)
{
	if (major != nullptr)
		*major = FREERDP_VERSION_MAJOR;

	if (minor != nullptr)
		*minor = FREERDP_VERSION_MINOR;

	if (revision != nullptr)
		*revision = FREERDP_VERSION_REVISION;
}

const char* freerdp_get_version_string(void)
{
	return FREERDP_VERSION_FULL;
}

const char* freerdp_get_build_config(void)
{
	WINPR_PRAGMA_DIAG_PUSH
	WINPR_PRAGMA_DIAG_IGNORED_OVERLENGTH_STRINGS
	static const char build_config[] =
	    "Build configuration: " FREERDP_BUILD_CONFIG "\n"
	    "Build type:          " FREERDP_BUILD_TYPE "\n"
	    "CFLAGS:              " FREERDP_CFLAGS "\n"
	    "Compiler:            " FREERDP_COMPILER_ID ", " FREERDP_COMPILER_VERSION "\n"
	    "Target architecture: " FREERDP_TARGET_ARCH "\n";
	WINPR_PRAGMA_DIAG_POP
	return build_config;
}

const char* freerdp_get_build_revision(void)
{
	return FREERDP_GIT_REVISION;
}

static wEventType FreeRDP_Events[] = {
	DEFINE_EVENT_ENTRY(WindowStateChange),   DEFINE_EVENT_ENTRY(ResizeWindow),
	DEFINE_EVENT_ENTRY(LocalResizeWindow),   DEFINE_EVENT_ENTRY(EmbedWindow),
	DEFINE_EVENT_ENTRY(PanningChange),       DEFINE_EVENT_ENTRY(ZoomingChange),
	DEFINE_EVENT_ENTRY(ErrorInfo),           DEFINE_EVENT_ENTRY(Terminate),
	DEFINE_EVENT_ENTRY(ConnectionResult),    DEFINE_EVENT_ENTRY(ChannelConnected),
	DEFINE_EVENT_ENTRY(ChannelDisconnected), DEFINE_EVENT_ENTRY(MouseEvent),
	DEFINE_EVENT_ENTRY(Activated),           DEFINE_EVENT_ENTRY(Timer),
	DEFINE_EVENT_ENTRY(GraphicsReset)
};

/** Allocator function for a rdp context.
 *  The function will allocate a rdpRdp structure using rdp_new(), then copy
 *  its contents to the appropriate fields in the rdp_freerdp structure given in parameters.
 *  It will also initialize the 'context' field in the rdp_freerdp structure as needed.
 *  If the caller has set the ContextNew callback in the 'instance' parameter, it will be called at
 * the end of the function.
 *
 *  @param instance - Pointer to the rdp_freerdp structure that will be initialized with the new
 * context.
 */
BOOL freerdp_context_new(freerdp* instance)
{
	return freerdp_context_new_ex(instance, nullptr);
}

static BOOL freerdp_common_context(rdpContext* context, AccessTokenType tokenType, char** token,
                                   size_t count, ...)
{
	BOOL rc = FALSE;

	WINPR_ASSERT(context);
	if (!context->instance || !context->instance->GetAccessToken)
		return TRUE;

	va_list ap = WINPR_C_ARRAY_INIT;
	va_start(ap, count);
	switch (tokenType)
	{
		case ACCESS_TOKEN_TYPE_AAD:
			if (count != 2)
			{
				WLog_ERR(TAG,
				         "ACCESS_TOKEN_TYPE_AAD expected 2 additional arguments, but got %" PRIuz
				         ", aborting",
				         count);
			}
			else
			{
				const char* scope = va_arg(ap, const char*);
				const char* req_cnf = va_arg(ap, const char*);
				rc = context->instance->GetAccessToken(context->instance, tokenType, token, count,
				                                       scope, req_cnf);
			}
			break;
		case ACCESS_TOKEN_TYPE_AVD:
			if (count != 0)
			{
				WLog_WARN(TAG,
				          "ACCESS_TOKEN_TYPE_AVD expected 0 additional arguments, but got %" PRIuz
				          ", ignoring",
				          count);
			}
			else
			{
				rc = context->instance->GetAccessToken(context->instance, tokenType, token, count);
			}
			break;
		default:
			break;
	}
	va_end(ap);

	if (!rc)
		freerdp_set_last_error_if_not(context, FREERDP_ERROR_CONNECT_ACCESS_DENIED);

	return rc;
}

BOOL freerdp_context_new_ex(freerdp* instance, rdpSettings* settings)
{
	rdpRdp* rdp = nullptr;
	rdpContext* context = nullptr;
	BOOL ret = TRUE;

	WINPR_ASSERT(instance);

	instance->context = context = (rdpContext*)calloc(1, instance->ContextSize);

	if (!context)
		return FALSE;

	context->log = WLog_Get(TAG);
	if (!context->log)
		goto fail;

	/* Set to external settings, prevents rdp_new from creating its own instance */
	context->settings = settings;
	context->instance = instance;
	context->ServerMode = FALSE;
	context->disconnectUltimatum = 0;

	context->metrics = metrics_new(context);

	if (!context->metrics)
		goto fail;

	rdp = rdp_new(context);

	if (!rdp)
		goto fail;

	context->rdp = rdp;
	context->pubSub = rdp->pubSub;

	if (!context->pubSub)
		goto fail;

	PubSub_AddEventTypes(rdp->pubSub, FreeRDP_Events, ARRAYSIZE(FreeRDP_Events));

#if defined(WITH_FREERDP_DEPRECATED)
	instance->input = rdp->input;
	instance->update = rdp->update;
	instance->settings = rdp->settings;
	instance->autodetect = rdp->autodetect;
#endif

	instance->heartbeat = rdp->heartbeat;
	context->graphics = graphics_new(context);

	if (!context->graphics)
		goto fail;

	context->input = rdp->input;
	context->update = rdp->update;
	context->settings = rdp->settings;
	context->autodetect = rdp->autodetect;

	if (!(context->errorDescription = calloc(1, 500)))
	{
		WLog_Print(context->log, WLOG_ERROR, "calloc failed!");
		goto fail;
	}

	if (!(context->channelErrorEvent = CreateEvent(nullptr, TRUE, FALSE, nullptr)))
	{
		WLog_Print(context->log, WLOG_ERROR, "CreateEvent failed!");
		goto fail;
	}

	update_register_client_callbacks(rdp->update);

	if (!(context->channels = freerdp_channels_new(instance)))
		goto fail;

	context->dump = stream_dump_new();
	if (!context->dump)
		goto fail;

	/* Fallback:
	 * Client common library might set a function pointer to handle this, but here we provide a
	 * default implementation that simply calls instance->GetAccessToken.
	 */
	if (!freerdp_set_common_access_token(context, freerdp_common_context))
		goto fail;

	IFCALLRET(instance->ContextNew, ret, instance, context);

	if (!ret)
		goto fail;

	return TRUE;

fail:
	freerdp_context_free(instance);
	return FALSE;
}

BOOL freerdp_context_reset(freerdp* instance)
{
	if (!instance)
		return FALSE;

	WINPR_ASSERT(instance->context);
	rdpRdp* rdp = instance->context->rdp;

	return rdp_reset_runtime_settings(rdp);
}

/** Deallocator function for a rdp context.
 *  The function will deallocate the resources from the 'instance' parameter that were allocated
 * from a call to freerdp_context_new(). If the ContextFree callback is set in the 'instance'
 * parameter, it will be called before deallocation occurs.
 *
 *  @param instance - Pointer to the rdp_freerdp structure that was initialized by a call to
 * freerdp_context_new(). On return, the fields associated to the context are invalid.
 */
void freerdp_context_free(freerdp* instance)
{
	rdpContext* ctx = nullptr;

	if (!instance)
		return;

	if (!instance->context)
		return;

	ctx = instance->context;

	IFCALL(instance->ContextFree, instance, ctx);
	rdp_free(ctx->rdp);
	ctx->rdp = nullptr;
	ctx->settings = nullptr; /* owned by rdpRdp */

	graphics_free(ctx->graphics);
	ctx->graphics = nullptr;

	metrics_free(ctx->metrics);
	ctx->metrics = nullptr;

	if (ctx->channelErrorEvent)
		(void)CloseHandle(ctx->channelErrorEvent);
	ctx->channelErrorEvent = nullptr;

	free(ctx->errorDescription);
	ctx->errorDescription = nullptr;

	freerdp_channels_free(ctx->channels);
	ctx->channels = nullptr;

	freerdp_client_codecs_free(ctx->codecs);
	ctx->codecs = nullptr;

	stream_dump_free(ctx->dump);
	ctx->dump = nullptr;

	ctx->input = nullptr;      /* owned by rdpRdp */
	ctx->update = nullptr;     /* owned by rdpRdp */
	ctx->settings = nullptr;   /* owned by rdpRdp */
	ctx->autodetect = nullptr; /* owned by rdpRdp */

	free(ctx);
	instance->context = nullptr;
#if defined(WITH_FREERDP_DEPRECATED)
	instance->input = nullptr;      /* owned by rdpRdp */
	instance->update = nullptr;     /* owned by rdpRdp */
	instance->settings = nullptr;   /* owned by rdpRdp */
	instance->autodetect = nullptr; /* owned by rdpRdp */
#endif
	instance->heartbeat = nullptr; /* owned by rdpRdp */
}

int freerdp_get_disconnect_ultimatum(const rdpContext* context)
{
	WINPR_ASSERT(context);
	return context->disconnectUltimatum;
}

UINT32 freerdp_error_info(const freerdp* instance)
{
	WINPR_ASSERT(instance);
	WINPR_ASSERT(instance->context);
	WINPR_ASSERT(instance->context->rdp);
	return instance->context->rdp->errorInfo;
}

void freerdp_set_error_info(rdpRdp* rdp, UINT32 error)
{
	if (!rdp)
		return;

	rdp_set_error_info(rdp, error);
}

BOOL freerdp_send_error_info(rdpRdp* rdp)
{
	if (!rdp)
		return FALSE;

	return rdp_send_error_info(rdp);
}

UINT32 freerdp_get_last_error(const rdpContext* context)
{
	WINPR_ASSERT(context);
	return context->LastError;
}

const char* freerdp_get_last_error_name(UINT32 code)
{
	const char* name = nullptr;
	const UINT32 cls = GET_FREERDP_ERROR_CLASS(code);
	const UINT32 type = GET_FREERDP_ERROR_TYPE(code);

	switch (cls)
	{
		case FREERDP_ERROR_ERRBASE_CLASS:
			name = freerdp_get_error_base_name(type);
			break;

		case FREERDP_ERROR_ERRINFO_CLASS:
			name = freerdp_get_error_info_name(type);
			break;

		case FREERDP_ERROR_CONNECT_CLASS:
			name = freerdp_get_error_connect_name(type);
			break;

		default:
			name = rpc_error_to_string(code);
			break;
	}

	return name;
}

const char* freerdp_get_last_error_string(UINT32 code)
{
	const char* string = nullptr;
	const UINT32 cls = GET_FREERDP_ERROR_CLASS(code);
	const UINT32 type = GET_FREERDP_ERROR_TYPE(code);

	switch (cls)
	{
		case FREERDP_ERROR_ERRBASE_CLASS:
			string = freerdp_get_error_base_string(type);
			break;

		case FREERDP_ERROR_ERRINFO_CLASS:
			string = freerdp_get_error_info_string(type);
			break;

		case FREERDP_ERROR_CONNECT_CLASS:
			string = freerdp_get_error_connect_string(type);
			break;

		default:
			string = rpc_error_to_string(code);
			break;
	}

	return string;
}

const char* freerdp_get_last_error_category(UINT32 code)
{
	const char* string = nullptr;
	const UINT32 cls = GET_FREERDP_ERROR_CLASS(code);
	const UINT32 type = GET_FREERDP_ERROR_TYPE(code);

	switch (cls)
	{
		case FREERDP_ERROR_ERRBASE_CLASS:
			string = freerdp_get_error_base_category(type);
			break;

		case FREERDP_ERROR_ERRINFO_CLASS:
			string = freerdp_get_error_info_category(type);
			break;

		case FREERDP_ERROR_CONNECT_CLASS:
			string = freerdp_get_error_connect_category(type);
			break;

		default:
			string = rpc_error_to_category(code);
			break;
	}

	return string;
}

void freerdp_set_last_error_ex(rdpContext* context, UINT32 lastError, const char* fkt,
                               const char* file, int line)
{
	WINPR_ASSERT(context);
	WINPR_ASSERT(line >= 0);

	if (lastError)
	{
		if (WLog_IsLevelActive(context->log, WLOG_ERROR))
		{
			WLog_PrintTextMessage(context->log, WLOG_ERROR, (size_t)line, file, fkt,
			                      "%s [0x%08" PRIX32 "]", freerdp_get_last_error_name(lastError),
			                      lastError);
		}
	}

	if (lastError == FREERDP_ERROR_SUCCESS)
	{
		if (WLog_IsLevelActive(context->log, WLOG_DEBUG))
			WLog_PrintTextMessage(context->log, WLOG_DEBUG, (size_t)line, file, fkt,
			                      "resetting error state");
	}
	else if (context->LastError != FREERDP_ERROR_SUCCESS)
	{
		if (WLog_IsLevelActive(context->log, WLOG_ERROR))
		{
			WLog_PrintTextMessage(context->log, WLOG_ERROR, (size_t)line, file, fkt,
			                      "TODO: Trying to set error code %s, but %s already set!",
			                      freerdp_get_last_error_name(lastError),
			                      freerdp_get_last_error_name(context->LastError));
		}
	}
	context->LastError = lastError;
}

const char* freerdp_get_logon_error_info_type_ex(UINT32 type, char* buffer, size_t size)
{
	const char* str = freerdp_get_logon_error_info_type(type);
	(void)_snprintf(buffer, size, "%s(0x%04" PRIx32 ")", str, type);
	return buffer;
}

const char* freerdp_get_logon_error_info_type(UINT32 type)
{
#define CASE_ENTRY(x) \
	case x:           \
		return #x
	switch (type)
	{
		CASE_ENTRY(LOGON_MSG_SESSION_BUSY_OPTIONS);
		CASE_ENTRY(LOGON_MSG_DISCONNECT_REFUSED);
		CASE_ENTRY(LOGON_MSG_NO_PERMISSION);
		CASE_ENTRY(LOGON_MSG_BUMP_OPTIONS);
		CASE_ENTRY(LOGON_MSG_RECONNECT_OPTIONS);
		CASE_ENTRY(LOGON_MSG_SESSION_TERMINATE);
		CASE_ENTRY(LOGON_MSG_SESSION_CONTINUE);
		CASE_ENTRY(ERROR_CODE_ACCESS_DENIED);

		default:
			return "UNKNOWN";
	}
#undef CASE_ENTRY
}

const char* freerdp_get_logon_error_info_data(UINT32 data)
{
	switch (data)
	{
		case LOGON_FAILED_BAD_PASSWORD:
			return "LOGON_FAILED_BAD_PASSWORD";

		case LOGON_FAILED_UPDATE_PASSWORD:
			return "LOGON_FAILED_UPDATE_PASSWORD";

		case LOGON_FAILED_OTHER:
			return "LOGON_FAILED_OTHER";

		case LOGON_WARNING:
			return "LOGON_WARNING";

		default:
			return "SESSION_ID";
	}
}

const char* freerdp_get_logon_error_info_data_ex(UINT32 data, char* buffer, size_t size)
{
	const char* str = freerdp_get_logon_error_info_data(data);
	(void)_snprintf(buffer, size, "%s(0x%04" PRIx32 ")", str, data);
	return buffer;
}

/** Allocator function for the rdp_freerdp structure.
 *  @return an allocated structure filled with 0s. Need to be deallocated using freerdp_free()
 */
freerdp* freerdp_new(void)
{
	freerdp* instance = nullptr;
	instance = (freerdp*)calloc(1, sizeof(freerdp));

	if (!instance)
		return nullptr;

	instance->ContextSize = sizeof(rdpContext);
	instance->SendChannelData = freerdp_send_channel_data;
	instance->SendChannelPacket = freerdp_send_channel_packet;
	instance->ReceiveChannelData = freerdp_channels_data;
	return instance;
}

/** Deallocator function for the rdp_freerdp structure.
 *  @param instance - pointer to the rdp_freerdp structure to deallocate.
 *                    On return, this pointer is not valid anymore.
 */
void freerdp_free(freerdp* instance)
{
	free(instance);
}

ULONG freerdp_get_transport_sent(const rdpContext* context, BOOL resetCount)
{
	WINPR_ASSERT(context);
	WINPR_ASSERT(context->rdp);
	UINT64 rc = transport_get_bytes_sent(context->rdp->transport, resetCount);
	return WINPR_CXX_COMPAT_CAST(ULONG, MIN(rc, UINT32_MAX));
}

BOOL freerdp_nla_impersonate(rdpContext* context)
{
	rdpNla* nla = nullptr;

	if (!context)
		return FALSE;

	if (!context->rdp)
		return FALSE;

	if (!context->rdp->transport)
		return FALSE;

	nla = transport_get_nla(context->rdp->transport);
	return nla_impersonate(nla);
}

BOOL freerdp_nla_revert_to_self(rdpContext* context)
{
	rdpNla* nla = nullptr;

	if (!context)
		return FALSE;

	if (!context->rdp)
		return FALSE;

	if (!context->rdp->transport)
		return FALSE;

	nla = transport_get_nla(context->rdp->transport);
	return nla_revert_to_self(nla);
}

UINT32 freerdp_get_nla_sspi_error(const rdpContext* context)
{
	WINPR_ASSERT(context);
	WINPR_ASSERT(context->rdp);
	WINPR_ASSERT(context->rdp->transport);

	rdpNla* nla = transport_get_nla(context->rdp->transport);
	return (UINT32)nla_get_sspi_error(nla);
}

BOOL freerdp_nla_encrypt(rdpContext* context, const SecBuffer* inBuffer, SecBuffer* outBuffer)
{
	WINPR_ASSERT(context);
	WINPR_ASSERT(context->rdp);

	rdpNla* nla = context->rdp->nla;
	return nla_encrypt(nla, inBuffer, outBuffer);
}

BOOL freerdp_nla_decrypt(rdpContext* context, const SecBuffer* inBuffer, SecBuffer* outBuffer)
{
	WINPR_ASSERT(context);
	WINPR_ASSERT(context->rdp);

	rdpNla* nla = context->rdp->nla;
	return nla_decrypt(nla, inBuffer, outBuffer);
}

SECURITY_STATUS freerdp_nla_QueryContextAttributes(rdpContext* context, DWORD ulAttr, PVOID pBuffer)
{
	WINPR_ASSERT(context);
	WINPR_ASSERT(context->rdp);

	rdpNla* nla = context->rdp->nla;
	if (!nla)
		nla = transport_get_nla(context->rdp->transport);

	WINPR_ASSERT(nla);

	return nla_QueryContextAttributes(nla, ulAttr, pBuffer);
}

SECURITY_STATUS freerdp_nla_FreeContextBuffer(rdpContext* context, PVOID pBuffer)
{
	WINPR_ASSERT(context);
	WINPR_ASSERT(context->rdp);

	rdpNla* nla = context->rdp->nla;
	if (!nla)
		nla = transport_get_nla(context->rdp->transport);

	WINPR_ASSERT(nla);

	return nla_FreeContextBuffer(nla, pBuffer);
}

HANDLE getChannelErrorEventHandle(rdpContext* context)
{
	WINPR_ASSERT(context);
	return context->channelErrorEvent;
}

BOOL checkChannelErrorEvent(rdpContext* context)
{
	WINPR_ASSERT(context);

	if (WaitForSingleObject(context->channelErrorEvent, 0) == WAIT_OBJECT_0)
	{
		WLog_Print(context->log, WLOG_ERROR, "%s. Error was %" PRIu32 "", context->errorDescription,
		           context->channelErrorNum);
		return FALSE;
	}

	return TRUE;
}

/**
 * Function description
 *
 * @return 0 on success, otherwise a Win32 error code
 */
UINT getChannelError(const rdpContext* context)
{
	WINPR_ASSERT(context);
	return context->channelErrorNum;
}

const char* getChannelErrorDescription(const rdpContext* context)
{
	WINPR_ASSERT(context);
	return context->errorDescription;
}

void clearChannelError(rdpContext* context)
{
	WINPR_ASSERT(context);
	context->channelErrorNum = 0;
	memset(context->errorDescription, 0, 500);
	(void)ResetEvent(context->channelErrorEvent);
}

WINPR_ATTR_FORMAT_ARG(3, 4)
void setChannelError(rdpContext* context, UINT errorNum, WINPR_FORMAT_ARG const char* format, ...)
{
	va_list ap = WINPR_C_ARRAY_INIT;
	va_start(ap, format);

	WINPR_ASSERT(context);

	context->channelErrorNum = errorNum;
	(void)vsnprintf(context->errorDescription, 499, format, ap);
	va_end(ap);
	(void)SetEvent(context->channelErrorEvent);
}

const char* freerdp_nego_get_routing_token(const rdpContext* context, DWORD* length)
{
	if (!context || !context->rdp)
		return nullptr;

	return (const char*)nego_get_routing_token(context->rdp->nego, length);
}

BOOL freerdp_io_callback_set_event(rdpContext* context, BOOL set)
{
	WINPR_ASSERT(context);
	return rdp_io_callback_set_event(context->rdp, set);
}

const rdpTransportIo* freerdp_get_io_callbacks(rdpContext* context)
{
	WINPR_ASSERT(context);
	return rdp_get_io_callbacks(context->rdp);
}

BOOL freerdp_set_io_callbacks(rdpContext* context, const rdpTransportIo* io_callbacks)
{
	WINPR_ASSERT(context);
	return rdp_set_io_callbacks(context->rdp, io_callbacks);
}

BOOL freerdp_set_io_callback_context(rdpContext* context, void* usercontext)
{
	WINPR_ASSERT(context);
	return rdp_set_io_callback_context(context->rdp, usercontext);
}

void* freerdp_get_io_callback_context(rdpContext* context)
{
	WINPR_ASSERT(context);
	return rdp_get_io_callback_context(context->rdp);
}

CONNECTION_STATE freerdp_get_state(const rdpContext* context)
{
	WINPR_ASSERT(context);
	return rdp_get_state(context->rdp);
}

const char* freerdp_state_string(CONNECTION_STATE state)
{
	return rdp_state_string(state);
}

BOOL freerdp_is_active_state(const rdpContext* context)
{
	WINPR_ASSERT(context);
	return rdp_is_active_state(context->rdp);
}

BOOL freerdp_channels_from_mcs(rdpSettings* settings, const rdpContext* context)
{
	WINPR_ASSERT(context);
	return rdp_channels_from_mcs(settings, context->rdp);
}

HANDLE freerdp_abort_event(rdpContext* context)
{
	WINPR_ASSERT(context);
	return utils_get_abort_event(context->rdp);
}

static void test_mcs_free(rdpMcs* mcs)
{
	if (!mcs)
		return;

	if (mcs->context)
	{
		rdpSettings* settings = mcs->context->settings;
		freerdp_settings_free(settings);
	}
	free(mcs->context);

	mcs_free(mcs);
}

static rdpMcs* test_mcs_new(void)
{
	rdpSettings* settings = freerdp_settings_new(0);
	rdpContext* context = calloc(1, sizeof(rdpContext));

	if (!settings)
		goto fail;
	if (!freerdp_settings_set_bool(settings, FreeRDP_TransportDumpReplay, TRUE))
		goto fail;

	if (!context)
		goto fail;
	context->settings = settings;
	return mcs_new(context);

fail:
	free(context);
	freerdp_settings_free(settings);

	return nullptr;
}

BOOL freerdp_is_valid_mcs_create_request(const BYTE* data, size_t size)
{

	wStream sbuffer = WINPR_C_ARRAY_INIT;
	wStream* s = Stream_StaticConstInit(&sbuffer, data, size);

	WINPR_ASSERT(data || (size == 0));
	WINPR_ASSERT(s);

	rdpMcs* mcs = test_mcs_new();
	WINPR_ASSERT(mcs);

	BOOL result = mcs_recv_connect_initial(mcs, s);
	test_mcs_free(mcs);
	return result;
}

BOOL freerdp_is_valid_mcs_create_response(const BYTE* data, size_t size)
{

	wStream sbuffer = WINPR_C_ARRAY_INIT;
	wStream* s = Stream_StaticConstInit(&sbuffer, data, size);

	WINPR_ASSERT(data || (size == 0));
	WINPR_ASSERT(s);

	rdpMcs* mcs = test_mcs_new();
	WINPR_ASSERT(mcs);

	BOOL result = mcs_recv_connect_response(mcs, s);
	test_mcs_free(mcs);
	return result;
}

BOOL freerdp_persist_credentials(rdpContext* context)
{
	if (!context)
		return FALSE;
	WINPR_ASSERT(context->rdp);
	return utils_persist_credentials(context->rdp->originalSettings, context->rdp->settings);
}

const char* freerdp_disconnect_reason_string(int reason)
{
	switch (reason)
	{
		case Disconnect_Ultimatum_domain_disconnected:
			return "rn-domain-disconnected";
		case Disconnect_Ultimatum_provider_initiated:
			return "rn-provider-initiated";
		case Disconnect_Ultimatum_token_purged:
			return "rn-token-purged";
		case Disconnect_Ultimatum_user_requested:
			return "rn-user-requested";
		case Disconnect_Ultimatum_channel_purged:
			return "rn-channel-purged";
		default:
			return "rn-unknown";
	}
}

BOOL freerdp_set_common_access_token(rdpContext* context,
                                     pGetCommonAccessToken GetCommonAccessToken)
{
	WINPR_ASSERT(context);
	WINPR_ASSERT(context->rdp);
	context->rdp->GetCommonAccessToken = GetCommonAccessToken;
	return TRUE;
}

pGetCommonAccessToken freerdp_get_common_access_token(const rdpContext* context)
{
	WINPR_ASSERT(context);
	WINPR_ASSERT(context->rdp);
	return context->rdp->GetCommonAccessToken;
}
