/**

 * 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.

 */

#ifdef HAVE_CONFIG_H

#include "config.h"

#endif

#include "rdp.h"

#include "input.h"

#include "update.h"

#include "surface.h"

#include "transport.h"

#include "connection.h"

#include "message.h"

#include "buildflags.h"

#include "gateway/rpc_fault.h"

#include <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/error.h>

#include <freerdp/event.h>

#include <freerdp/locale/keyboard.h>

#include <freerdp/channels/channels.h>

#include <freerdp/version.h>

#include <freerdp/log.h>

#include <freerdp/cache/pointer.h>

#include "settings.h"

#define TAG FREERDP_TAG("core")

/* connectErrorCode is 'extern' in error.h. See comment there.*/

UINT freerdp_channel_add_init_handle_data(rdpChannelHandles* handles, void* pInitHandle,

                                          void* pUserData)

{

	if (!handles->init)

		handles->init = ListDictionary_New(TRUE);

	if (!handles->init)

	{

		WLog_ERR(TAG, "ListDictionary_New failed!");

		return ERROR_NOT_ENOUGH_MEMORY;

	}

	if (!ListDictionary_Add(handles->init, pInitHandle, pUserData))

	{

		WLog_ERR(TAG, "ListDictionary_Add failed!");

		return ERROR_INTERNAL_ERROR;

	}

	return CHANNEL_RC_OK;

}

void* freerdp_channel_get_init_handle_data(rdpChannelHandles* handles, void* pInitHandle)

{

	void* pUserData = NULL;

	pUserData = ListDictionary_GetItemValue(handles->init, pInitHandle);

	return pUserData;

}

void freerdp_channel_remove_init_handle_data(rdpChannelHandles* handles, void* pInitHandle)

{

	ListDictionary_Remove(handles->init, pInitHandle);

	if (ListDictionary_Count(handles->init) < 1)

	{

		ListDictionary_Free(handles->init);

		handles->init = NULL;

	}

}

UINT freerdp_channel_add_open_handle_data(rdpChannelHandles* handles, DWORD openHandle,

                                          void* pUserData)

{

	void* pOpenHandle = (void*)(size_t)openHandle;

	if (!handles->open)

		handles->open = ListDictionary_New(TRUE);

	if (!handles->open)

	{

		WLog_ERR(TAG, "ListDictionary_New failed!");

		return ERROR_NOT_ENOUGH_MEMORY;

	}

	if (!ListDictionary_Add(handles->open, pOpenHandle, pUserData))

	{

		WLog_ERR(TAG, "ListDictionary_Add failed!");

		return ERROR_INTERNAL_ERROR;

	}

	return CHANNEL_RC_OK;

}

void* freerdp_channel_get_open_handle_data(rdpChannelHandles* handles, DWORD openHandle)

{

	void* pUserData = NULL;

	void* pOpenHandle = (void*)(size_t)openHandle;

	pUserData = ListDictionary_GetItemValue(handles->open, pOpenHandle);

	return pUserData;

}

void freerdp_channel_remove_open_handle_data(rdpChannelHandles* handles, DWORD openHandle)

{

	void* pOpenHandle = (void*)(size_t)openHandle;

	ListDictionary_Remove(handles->open, pOpenHandle);

	if (ListDictionary_Count(handles->open) < 1)

	{

		ListDictionary_Free(handles->open);

		handles->open = NULL;

	}

}

/** 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.

 *

 */

BOOL freerdp_connect(freerdp* instance)

{

	UINT status2 = CHANNEL_RC_OK;

	rdpRdp* rdp;

	BOOL status = TRUE;

	rdpSettings* settings;

	ConnectionResultEventArgs e;

	if (!instance)

		return FALSE;

	/* We always set the return code to 0 before we start the connect sequence*/

	instance->ConnectionCallbackState = CLIENT_STATE_INITIAL;

	connectErrorCode = 0;

	freerdp_set_last_error_log(instance->context, FREERDP_ERROR_SUCCESS);

	clearChannelError(instance->context);

	ResetEvent(instance->context->abortEvent);

	rdp = instance->context->rdp;

	settings = instance->settings;

	if (!freerdp_settings_set_default_order_support(settings))

		return FALSE;

	IFCALLRET(instance->PreConnect, status, instance);

	instance->ConnectionCallbackState = CLIENT_STATE_PRECONNECT_PASSED;

	if (status)

		status2 = freerdp_channels_pre_connect(instance->context->channels, instance);

	if (settings->KeyboardLayout == KBD_JAPANESE_INPUT_SYSTEM_MS_IME2002)

	{

		settings->KeyboardType = 7;

		settings->KeyboardSubType = 2;

		settings->KeyboardFunctionKey = 12;

	}

	if (!status || (status2 != CHANNEL_RC_OK))

	{

		freerdp_set_last_error_if_not(instance->context, FREERDP_ERROR_PRE_CONNECT_FAILED);

		WLog_ERR(TAG, "freerdp_pre_connect failed");

		goto freerdp_connect_finally;

	}

	status = rdp_client_connect(rdp);

	/* --authonly tests the connection without a UI */

	if (instance->settings->AuthenticationOnly)

	{

		WLog_ERR(TAG, "Authentication only, exit status %" PRId32 "", !status);

		goto freerdp_connect_finally;

	}

	if (instance->settings->DumpRemoteFx)

	{

		instance->update->pcap_rfx = pcap_open(instance->settings->DumpRemoteFxFile, TRUE);

		if (instance->update->pcap_rfx)

			instance->update->dump_rfx = TRUE;

	}

	if (status)

	{

		pointer_cache_register_callbacks(instance->context->update);

		IFCALLRET(instance->PostConnect, status, instance);

		instance->ConnectionCallbackState = CLIENT_STATE_POSTCONNECT_PASSED;

		if (status)

			status2 = freerdp_channels_post_connect(instance->context->channels, instance);

	}

	else

	{

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

	{

		WLog_ERR(TAG, "freerdp_post_connect failed");

		freerdp_set_last_error_if_not(instance->context, FREERDP_ERROR_POST_CONNECT_FAILED);

		status = FALSE;

		goto freerdp_connect_finally;

	}

	if (instance->settings->PlayRemoteFx)

	{

		wStream* s;

		rdpUpdate* update;

		pcap_record record;

		update = instance->update;

		update->pcap_rfx = pcap_open(settings->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)

		{

			pcap_get_next_record_header(update->pcap_rfx, &record);

			if (!(s = StreamPool_Take(rdp->transport->ReceivePool, record.length)))

				break;

			record.data = Stream_Buffer(s);

			pcap_get_next_record_content(update->pcap_rfx, &record);

			Stream_SetLength(s, record.length);

			Stream_SetPosition(s, 0);

			if (!update_begin_paint(update))

				status = FALSE;

			else

			{

				if (update_recv_surfcmds(update, s) < 0)

					status = FALSE;

				if (!update_end_paint(update))

					status = FALSE;

			}

			Stream_Release(s);

		}

		pcap_close(update->pcap_rfx);

		update->pcap_rfx = NULL;

		goto freerdp_connect_finally;

	}

	if (rdp->errorInfo == ERRINFO_SERVER_INSUFFICIENT_PRIVILEGES)

		freerdp_set_last_error_log(instance->context, FREERDP_ERROR_INSUFFICIENT_PRIVILEGES);

	SetEvent(rdp->transport->connectedEvent);

freerdp_connect_finally:

	EventArgsInit(&e, "freerdp");

	e.result = status ? 0 : -1;

	PubSub_OnConnectionResult(instance->context->pubSub, instance->context, &e);

	if (!status)

		freerdp_disconnect(instance);

	return status;

}

BOOL freerdp_abort_connect(freerdp* instance)

{

	if (!instance || !instance->context)

		return FALSE;

	return SetEvent(instance->context->abortEvent);

}

BOOL freerdp_get_fds(freerdp* instance, void** rfds, int* rcount, void** wfds, int* wcount)

{

	rdpRdp* rdp = instance->context->rdp;

	transport_get_fds(rdp->transport, rfds, rcount);

	return TRUE;

}

BOOL freerdp_check_fds(freerdp* instance)

{

	int status;

	rdpRdp* rdp;

	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;

		WLog_DBG(TAG, "rdp_check_fds() - %i", status);

		EventArgsInit(&e, "freerdp");

		e.code = 0;

		PubSub_OnTerminate(context->pubSub, context, &e);

		return FALSE;

	}

	return TRUE;

}

DWORD freerdp_get_event_handles(rdpContext* context, HANDLE* events, DWORD count)

{

	DWORD nCount = 0;

	nCount += transport_get_event_handles(context->rdp->transport, events, count);

	if (nCount == 0)

		return 0;

	if (events && (nCount < count + 2))

	{

		events[nCount++] = freerdp_channels_get_event_handle(context->instance);

		events[nCount++] = getChannelErrorEventHandle(context);

		events[nCount++] = context->abortEvent;

	}

	else

		return 0;

	if (context->settings->AsyncInput)

	{

		if (nCount >= count)

			return 0;

		events[nCount++] =

		    freerdp_get_message_queue_event_handle(context->instance, FREERDP_INPUT_MESSAGE_QUEUE);

	}

	return nCount;

}

BOOL freerdp_check_event_handles(rdpContext* context)

{

	BOOL status;

	status = freerdp_check_fds(context->instance);

	if (!status)

	{

		if (freerdp_get_last_error(context) == FREERDP_ERROR_SUCCESS)

			WLog_ERR(TAG, "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_ERR(TAG, "freerdp_channels_check_fds() failed - %" PRIi32 "", status);

		return FALSE;

	}

	status = checkChannelErrorEvent(context);

	if (!status)

	{

		if (freerdp_get_last_error(context) == FREERDP_ERROR_SUCCESS)

			WLog_ERR(TAG, "checkChannelErrorEvent() failed - %" PRIi32 "", status);

		return FALSE;

	}

	if (context->settings->AsyncInput)

	{

		int rc = freerdp_message_queue_process_pending_messages(context->instance,

		                                                        FREERDP_INPUT_MESSAGE_QUEUE);

		if (rc < 0)

			return FALSE;

		else

			status = TRUE;

	}

	return status;

}

wMessageQueue* freerdp_get_message_queue(freerdp* instance, DWORD id)

{

	wMessageQueue* queue = NULL;

	switch (id)

	{

		case FREERDP_UPDATE_MESSAGE_QUEUE:

			queue = instance->update->queue;

			break;

		case FREERDP_INPUT_MESSAGE_QUEUE:

			queue = instance->input->queue;

			break;

	}

	return queue;

}

HANDLE freerdp_get_message_queue_event_handle(freerdp* instance, DWORD id)

{

	HANDLE event = NULL;

	wMessageQueue* queue = NULL;

	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;

	switch (id)

	{

		case FREERDP_UPDATE_MESSAGE_QUEUE:

			status = update_message_queue_process_message(instance->update, message);

			break;

		case FREERDP_INPUT_MESSAGE_QUEUE:

			status = input_message_queue_process_message(instance->input, message);

			break;

	}

	return status;

}

int freerdp_message_queue_process_pending_messages(freerdp* instance, DWORD id)

{

	int status = -1;

	switch (id)

	{

		case FREERDP_UPDATE_MESSAGE_QUEUE:

			status = update_message_queue_process_pending_messages(instance->update);

			break;

		case FREERDP_INPUT_MESSAGE_QUEUE:

			status = input_message_queue_process_pending_messages(instance->input);

			break;

	}

	return status;

}

static BOOL freerdp_send_channel_data(freerdp* instance, UINT16 channelId, const BYTE* data,

                                      size_t size)

{

	return rdp_send_channel_data(instance->context->rdp, channelId, data, size);

}

BOOL freerdp_disconnect(freerdp* instance)

{

	BOOL rc = TRUE;

	rdpRdp* rdp;

	if (!instance || !instance->context || !instance->context->rdp)

		return FALSE;

	rdp = instance->context->rdp;

	if (!rdp_client_disconnect(rdp))

		rc = FALSE;

	update_post_disconnect(instance->update);

	if (instance->settings->AsyncInput)

	{

		wMessageQueue* inputQueue =

		    freerdp_get_message_queue(instance, FREERDP_INPUT_MESSAGE_QUEUE);

		MessageQueue_PostQuit(inputQueue, 0);

	}

	IFCALL(instance->PostDisconnect, instance);

	if (instance->update->pcap_rfx)

	{

		instance->update->dump_rfx = FALSE;

		pcap_close(instance->update->pcap_rfx);

		instance->update->pcap_rfx = NULL;

	}

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

	return rc;

}

BOOL freerdp_disconnect_before_reconnect(freerdp* instance)

{

	rdpRdp* rdp = instance->context->rdp;

	return rdp_client_disconnect_and_clear(rdp);

}

BOOL freerdp_reconnect(freerdp* instance)

{

	BOOL status;

	rdpRdp* rdp = instance->context->rdp;

	ResetEvent(instance->context->abortEvent);

	status = rdp_client_reconnect(rdp);

	return status;

}

BOOL freerdp_shall_disconnect(freerdp* instance)

{

	if (!instance || !instance->context)

		return FALSE;

	if (WaitForSingleObject(instance->context->abortEvent, 0) != WAIT_OBJECT_0)

		return FALSE;

	return TRUE;

}

BOOL freerdp_focus_required(freerdp* instance)

{

	rdpRdp* rdp;

	BOOL bRetCode = FALSE;

	rdp = instance->context->rdp;

	if (rdp->resendFocus)

	{

		bRetCode = TRUE;

		rdp->resendFocus = FALSE;

	}

	return bRetCode;

}

void freerdp_set_focus(freerdp* instance)

{

	rdpRdp* rdp;

	rdp = instance->context->rdp;

	rdp->resendFocus = TRUE;

}

void freerdp_get_version(int* major, int* minor, int* revision)

{

	if (major != NULL)

		*major = FREERDP_VERSION_MAJOR;

	if (minor != NULL)

		*minor = FREERDP_VERSION_MINOR;

	if (revision != NULL)

		*revision = FREERDP_VERSION_REVISION;

}

const char* freerdp_get_version_string(void)

{

	return FREERDP_VERSION_FULL;

}

const char* freerdp_get_build_date(void)

{

	static char build_date[] = __DATE__ " " __TIME__;

	return build_date;

}

const char* freerdp_get_build_config(void)

{

	static const char build_config[] =

	    "Build configuration: " BUILD_CONFIG "\n"

	    "Build type:          " BUILD_TYPE "\n"

	    "CFLAGS:              " CFLAGS "\n"

	    "Compiler:            " COMPILER_ID ", " COMPILER_VERSION "\n"

	    "Target architecture: " TARGET_ARCH "\n";

	return build_config;

}

const char* freerdp_get_build_revision(void)

{

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

{

	rdpRdp* rdp;

	rdpContext* context;

	BOOL ret = TRUE;

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

	if (!instance->context)

		return FALSE;

	context = instance->context;

	context->instance = instance;

	context->ServerMode = FALSE;

	context->settings = instance->settings;

	context->disconnectUltimatum = 0;

	context->pubSub = PubSub_New(TRUE);

	if (!context->pubSub)

		goto fail;

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

	context->metrics = metrics_new(context);

	if (!context->metrics)

		goto fail;

	rdp = rdp_new(context);

	if (!rdp)

		goto fail;

	instance->input = rdp->input;

	instance->update = rdp->update;

	instance->settings = rdp->settings;

	instance->autodetect = rdp->autodetect;

	instance->heartbeat = rdp->heartbeat;

	context->graphics = graphics_new(context);

	if (!context->graphics)

		goto fail;

	context->rdp = rdp;

	context->input = instance->input;

	context->update = instance->update;

	context->settings = instance->settings;

	context->autodetect = instance->autodetect;

	instance->update->context = instance->context;

	instance->update->pointer->context = instance->context;

	instance->update->primary->context = instance->context;

	instance->update->secondary->context = instance->context;

	instance->update->altsec->context = instance->context;

	instance->input->context = context;

	instance->autodetect->context = context;

	if (!(context->errorDescription = calloc(1, 500)))

	{

		WLog_ERR(TAG, "calloc failed!");

		goto fail;

	}

	if (!(context->channelErrorEvent = CreateEvent(NULL, TRUE, FALSE, NULL)))

	{

		WLog_ERR(TAG, "CreateEvent failed!");

		goto fail;

	}

	update_register_client_callbacks(rdp->update);

	instance->context->abortEvent = CreateEvent(NULL, TRUE, FALSE, NULL);

	if (!instance->context->abortEvent)

		goto fail;

	if (!(context->channels = freerdp_channels_new(instance)))

		goto fail;

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

	if (ret)

		return TRUE;

fail:

	freerdp_context_free(instance);

	return FALSE;

}

/** 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)

{

	if (!instance)

		return;

	if (!instance->context)

		return;

	IFCALL(instance->ContextFree, instance, instance->context);

	rdp_free(instance->context->rdp);

	instance->context->rdp = NULL;

	graphics_free(instance->context->graphics);

	instance->context->graphics = NULL;

	PubSub_Free(instance->context->pubSub);

	metrics_free(instance->context->metrics);

	CloseHandle(instance->context->channelErrorEvent);

	free(instance->context->errorDescription);

	CloseHandle(instance->context->abortEvent);

	instance->context->abortEvent = NULL;

	freerdp_channels_free(instance->context->channels);

	free(instance->context);

	instance->context = NULL;

}

int freerdp_get_disconnect_ultimatum(rdpContext* context)

{

	return context->disconnectUltimatum;

}

UINT32 freerdp_error_info(freerdp* instance)

{

	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(rdpContext* context)

{

	return context->LastError;

}

const char* freerdp_get_last_error_name(UINT32 code)

{

	const char* name = NULL;

	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 = NULL;

	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 = NULL;

	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(rdpContext* context, UINT32 lastError)

{

	freerdp_set_last_error_ex(context, lastError, NULL, NULL, -1);

}

void freerdp_set_last_error_ex(rdpContext* context, UINT32 lastError, const char* fkt,

                               const char* file, int line)

{

	if (lastError)