/**

 * FreeRDP: A Remote Desktop Protocol Implementation

 * RAIL common functions

 *

 * Copyright 2011 Marc-Andre Moreau <marcandre.moreau@gmail.com>

 * Copyright 2011 Roman Barabanov <romanbarabanov@gmail.com>

 * Copyright 2011 Vic Lee

 * 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 "rail_common.h"

#include <winpr/crt.h>

const char* const RAIL_ORDER_TYPE_STRINGS[] =

{

	"",

	"Execute",

	"Activate",

	"System Parameters Update",

	"System Command",

	"Handshake",

	"Notify Event",

	"",

	"Window Move",

	"Local Move/Size",

	"Min Max Info",

	"Client Status",

	"System Menu",

	"Language Bar Info",

	"Get Application ID Request",

	"Get Application ID Response",

	"Execute Result",

	"",

	"",

	"",

	"",

	"",

	""

};

BOOL rail_string_to_unicode_string(const char* string, RAIL_UNICODE_STRING* unicode_string)

{

	WCHAR* buffer = NULL;

	int length = 0;

	free(unicode_string->string);

	unicode_string->string = NULL;

	unicode_string->length = 0;

	if (!string || strlen(string) < 1)

		return TRUE;

	length = ConvertToUnicode(CP_UTF8, 0, string, -1, &buffer, 0);

	if ((length < 0) || ((size_t)length * sizeof(WCHAR) > UINT16_MAX))

	{

		free(buffer);

		return FALSE;

	}

	unicode_string->string = (BYTE*) buffer;

	unicode_string->length = (UINT16) length * sizeof(WCHAR);

	return TRUE;

}

/**

 * Function description

 *

 * @return 0 on success, otherwise a Win32 error code

 */

UINT rail_read_pdu_header(wStream* s, UINT16* orderType, UINT16* orderLength)

{

	if (!s || !orderType || !orderLength)

		return ERROR_INVALID_PARAMETER;

	if (Stream_GetRemainingLength(s) < 4)

		return ERROR_INVALID_DATA;

	Stream_Read_UINT16(s, *orderType); /* orderType (2 bytes) */

	Stream_Read_UINT16(s, *orderLength); /* orderLength (2 bytes) */

	return CHANNEL_RC_OK;

}

void rail_write_pdu_header(wStream* s, UINT16 orderType, UINT16 orderLength)

{

	Stream_Write_UINT16(s, orderType); /* orderType (2 bytes) */

	Stream_Write_UINT16(s, orderLength); /* orderLength (2 bytes) */

}

wStream* rail_pdu_init(size_t length)

{

	wStream* s;

	s = Stream_New(NULL, length + RAIL_PDU_HEADER_LENGTH);

	if (!s)

		return NULL;

	Stream_Seek(s, RAIL_PDU_HEADER_LENGTH);

	return s;

}

/**

 * Function description

 *

 * @return 0 on success, otherwise a Win32 error code

 */

UINT rail_read_handshake_order(wStream* s, RAIL_HANDSHAKE_ORDER* handshake)

{

	if (Stream_GetRemainingLength(s) < 4)

		return ERROR_INVALID_DATA;

	Stream_Read_UINT32(s, handshake->buildNumber); /* buildNumber (4 bytes) */

	return CHANNEL_RC_OK;

}

void rail_write_handshake_order(wStream* s, const RAIL_HANDSHAKE_ORDER* handshake)

{

	Stream_Write_UINT32(s, handshake->buildNumber); /* buildNumber (4 bytes) */

}

/**

 * Function description

 *

 * @return 0 on success, otherwise a Win32 error code

 */

UINT rail_read_handshake_ex_order(wStream* s, RAIL_HANDSHAKE_EX_ORDER* handshakeEx)

{

	if (Stream_GetRemainingLength(s) < 8)

		return ERROR_INVALID_DATA;

	Stream_Read_UINT32(s, handshakeEx->buildNumber); /* buildNumber (4 bytes) */

	Stream_Read_UINT32(s, handshakeEx->railHandshakeFlags); /* railHandshakeFlags (4 bytes) */

	return CHANNEL_RC_OK;

}

void rail_write_handshake_ex_order(wStream* s, const RAIL_HANDSHAKE_EX_ORDER* handshakeEx)

{

	Stream_Write_UINT32(s, handshakeEx->buildNumber); /* buildNumber (4 bytes) */

	Stream_Write_UINT32(s, handshakeEx->railHandshakeFlags); /* railHandshakeFlags (4 bytes) */

}