/*

* Copyright (c) 2001-2006 Nokia Corporation and/or its subsidiary(-ies).

* All rights reserved.

* This component and the accompanying materials are made available

* under the terms of the License "Eclipse Public License v1.0"

* which accompanies this distribution, and is available

* at the URL "http://www.eclipse.org/legal/epl-v10.html".

*

* Initial Contributors:

* Nokia Corporation - initial contribution.

*

* Contributors:

*

* Description:  EAP and WLAN authentication protocols.

*

*/

// This is enumeration of EAPOL source code.

#if defined(USE_EAP_MINIMUM_RELEASE_TRACES)

	#undef EAP_FILE_NUMBER_ENUM

	#define EAP_FILE_NUMBER_ENUM 40 

	#undef EAP_FILE_NUMBER_DATE 

	#define EAP_FILE_NUMBER_DATE 1127594498 

#endif //#if defined(USE_EAP_MINIMUM_RELEASE_TRACES)

#include "eap_am_memory.h"

#include "eap_automatic_variable.h"

#include "eapol_handle_tlv_message_data.h"

#include "eap_variable_data.h"

#include "eap_am_network_id.h"

#include "eap_buffer.h"

#include "eapol_session_key.h"

#include "abs_eap_state_notification.h"

#include "eap_state_notification.h"

#if defined(USE_EAP_SIMPLE_CONFIG)

#include "simple_config_types.h"

#include "simple_config_credential.h"

#include "simple_config_payloads.h"

#endif // #if defined(USE_EAP_SIMPLE_CONFIG)

/** @file */

//--------------------------------------------------

EAP_FUNC_EXPORT eapol_handle_tlv_message_data_c::~eapol_handle_tlv_message_data_c()

{

}

//--------------------------------------------------

EAP_FUNC_EXPORT eapol_handle_tlv_message_data_c::eapol_handle_tlv_message_data_c(

	abs_eap_am_tools_c * const tools)

	: eap_tlv_message_data_c(tools)

	, m_am_tools(tools)

	, m_is_valid(true)

{

}

//-------------------------------------------------------------------

/**

 * This function should increase reference count.

 */

EAP_FUNC_EXPORT void eapol_handle_tlv_message_data_c::object_increase_reference_count()

{

}

//-------------------------------------------------------------------

/**

 * This function should first decrease reference count

 * and second return the remaining reference count.

 * Reference count must not be decreased when it is zero.

 */

EAP_FUNC_EXPORT u32_t eapol_handle_tlv_message_data_c::object_decrease_reference_count()

{

	return 0;

}

//--------------------------------------------------

EAP_FUNC_EXPORT bool eapol_handle_tlv_message_data_c::get_is_valid()

{

	return m_is_valid && eap_tlv_message_data_c::get_is_valid();

}

//--------------------------------------------------

EAP_FUNC_EXPORT u32_t eapol_handle_tlv_message_data_c::get_payload_size(

	const eap_am_network_id_c * const network_id) const

{

	return

		(3ul * eap_tlv_header_c::get_header_length() // Each attribute have their own header.

		+ network_id->get_source_id()->get_data_length()

		+ network_id->get_destination_id()->get_data_length()

		+ sizeof(network_id->get_type()));

}

//--------------------------------------------------

EAP_FUNC_EXPORT u32_t eapol_handle_tlv_message_data_c::get_payload_size(

	const abs_eap_state_notification_c * const state) const

{

	return

		(7ul * eap_tlv_header_c::get_header_length()) // Each attribute have their own header.

		+ (get_payload_size(state->get_send_network_id())

		+ sizeof(u32_t) // eap_protocol_layer_e

		+ sizeof(state->get_protocol())

		+ eap_expanded_type_c::get_eap_expanded_type_size()

		+ sizeof(state->get_current_state())

		+ sizeof(u32_t) // bool is_client

		+ sizeof(u32_t) // eap_status_e authentication error

		);

}

//--------------------------------------------------

EAP_FUNC_EXPORT u32_t eapol_handle_tlv_message_data_c::get_payload_size(

	const eapol_session_key_c * const session_key) const

{

	return

		((5ul * eap_tlv_header_c::get_header_length()) // Each attribute have their own header.

		+ session_key->get_key()->get_data_length()

		+ session_key->get_sequence_number()->get_data_length()

		+ sizeof(u32_t) // const eapol_key_type_e m_key_type

		+ sizeof(session_key->get_key_index())

		+ sizeof(u32_t) // const bool m_key_tx_bit

		);

}

//--------------------------------------------------

#if defined(USE_EAP_SIMPLE_CONFIG)

EAP_FUNC_EXPORT u32_t eapol_handle_tlv_message_data_c::get_payload_size(

	network_key_and_index_c * key) const

{

	u32_t size(0ul);

	if (key != 0)

	{

		size += eap_tlv_header_c::get_header_length()

			+ sizeof(key->get_network_key_index()) // Size of Network Key Index

			+ eap_tlv_header_c::get_header_length()

			+ key->get_network_key()->get_data_length() // Size of Network Key

			;

	}

	return (size);

}

#endif // #if defined(USE_EAP_SIMPLE_CONFIG)

//--------------------------------------------------

#if defined(USE_EAP_SIMPLE_CONFIG)

EAP_FUNC_EXPORT u32_t eapol_handle_tlv_message_data_c::get_payload_size(

	EAP_TEMPLATE_CONST eap_array_c<network_key_and_index_c> * network_keys) const

{

	u32_t size(0ul);

	for (u32_t ind_network_key = 0ul; ind_network_key < network_keys->get_object_count(); ind_network_key++)

	{

		network_key_and_index_c * const key = network_keys->get_object(ind_network_key);

		if (key != 0)

		{

			size += eap_tlv_header_c::get_header_length() // Size of structure header

				+ get_payload_size(key); // Size of Network Key

		}

	} // for ()

	return (size);

}

#endif // #if defined(USE_EAP_SIMPLE_CONFIG)

//--------------------------------------------------

#if defined(USE_EAP_SIMPLE_CONFIG)

EAP_FUNC_EXPORT u32_t eapol_handle_tlv_message_data_c::get_payload_size(

	simple_config_credential_c * const credential) const

{

	u32_t size(0ul);

	if (credential != 0)

	{

		size += eap_tlv_header_c::get_header_length()

			+ sizeof(credential->get_network_index()) // Size of Network Index

			+ eap_tlv_header_c::get_header_length()

			+ credential->get_SSID()->get_data_length() // Size of SSID

			+ eap_tlv_header_c::get_header_length()

			+ sizeof(u16_t) // Size of Authentiction type

			+ eap_tlv_header_c::get_header_length()

			+ sizeof(u16_t) // Size of Encryption type

			;

		size += eap_tlv_header_c::get_header_length() // Size of header of Array

			+ get_payload_size(credential->get_network_keys());

		size += eap_tlv_header_c::get_header_length()

			+ credential->get_MAC_address()->get_data_length() // Size of MAC Address

			;

	}

	return (size);

}

#endif // #if defined(USE_EAP_SIMPLE_CONFIG)

//--------------------------------------------------

#if defined(USE_EAP_SIMPLE_CONFIG)

EAP_FUNC_EXPORT u32_t eapol_handle_tlv_message_data_c::get_payload_size(

	EAP_TEMPLATE_CONST eap_array_c<simple_config_credential_c> * const credential_array) const

{

	u32_t size(0ul);

	for (u32_t ind_credential = 0ul; ind_credential < credential_array->get_object_count(); ind_credential++)

	{

		simple_config_credential_c * const credential = credential_array->get_object(ind_credential);

		if (credential != 0)

		{

			size += eap_tlv_header_c::get_header_length() // Size of structure header

				+ get_payload_size(credential);

		}

	} // for ()

	return (size);

}

#endif // #if defined(USE_EAP_SIMPLE_CONFIG)

//--------------------------------------------------

EAP_FUNC_EXPORT eap_status_e eapol_handle_tlv_message_data_c::add_structured_parameter_header(

	const eapol_tlv_message_type_e type,

	const u32_t length)

{

	EAP_TRACE_DEBUG(

		m_am_tools,

		EAP_TRACE_FLAGS_MESSAGE_DATA,

		(EAPL("eapol_handle_tlv_message_data_c::add_structured_parameter_header(): type=%s\n"),

		 get_type_string(type)));

	return add_message_header(

		type,

		length);

}

//--------------------------------------------------

EAP_FUNC_EXPORT eap_status_e eapol_handle_tlv_message_data_c::add_parameter_data(

	const eapol_tlv_message_type_e type,

	const u32_t integer)

{

	EAP_TRACE_BEGIN(m_am_tools, TRACE_FLAGS_DEFAULT);

	EAP_TRACE_DEBUG(

		m_am_tools,

		EAP_TRACE_FLAGS_MESSAGE_DATA,

		(EAPL("eapol_handle_tlv_message_data_c::add_parameter_data(): type=%s\n"),

		 get_type_string(type)));

	const u32_t network_order_integer(eap_htonl(integer));

	eap_status_e status = add_message_data(

		type,

		sizeof(network_order_integer),

		&network_order_integer);

	EAP_TRACE_END(m_am_tools, TRACE_FLAGS_DEFAULT);

	return EAP_STATUS_RETURN(m_am_tools, status);

}

//--------------------------------------------------

EAP_FUNC_EXPORT eap_status_e eapol_handle_tlv_message_data_c::add_parameter_data(

	const u64_t long_integer)

{

	EAP_TRACE_BEGIN(m_am_tools, TRACE_FLAGS_DEFAULT);

	EAP_TRACE_DEBUG(

		m_am_tools,

		EAP_TRACE_FLAGS_MESSAGE_DATA,

		(EAPL("eapol_handle_tlv_message_data_c::add_parameter_data(): type=%s\n"),

		 get_type_string(eapol_tlv_message_type_u64_t)));

	const u64_t network_order_long_integer(eap_htonll(long_integer));

	eap_status_e status = add_message_data(

		eapol_tlv_message_type_u64_t,

		sizeof(network_order_long_integer),

		&network_order_long_integer);

	EAP_TRACE_END(m_am_tools, TRACE_FLAGS_DEFAULT);

	return EAP_STATUS_RETURN(m_am_tools, status);

}

//--------------------------------------------------

EAP_FUNC_EXPORT eap_status_e eapol_handle_tlv_message_data_c::add_parameter_data(

	const u32_t integer)

{

	EAP_TRACE_BEGIN(m_am_tools, TRACE_FLAGS_DEFAULT);

	EAP_TRACE_DEBUG(

		m_am_tools,

		EAP_TRACE_FLAGS_MESSAGE_DATA,

		(EAPL("eapol_handle_tlv_message_data_c::add_parameter_data(): type=%s\n"),

		 get_type_string(eapol_tlv_message_type_u32_t)));

	const u32_t network_order_integer(eap_htonl(integer));

	eap_status_e status = add_message_data(

		eapol_tlv_message_type_u32_t,

		sizeof(network_order_integer),

		&network_order_integer);

	EAP_TRACE_END(m_am_tools, TRACE_FLAGS_DEFAULT);

	return EAP_STATUS_RETURN(m_am_tools, status);

}

//--------------------------------------------------

EAP_FUNC_EXPORT eap_status_e eapol_handle_tlv_message_data_c::add_parameter_data(

	const u16_t short_integer)

{

	EAP_TRACE_BEGIN(m_am_tools, TRACE_FLAGS_DEFAULT);

	EAP_TRACE_DEBUG(

		m_am_tools,

		EAP_TRACE_FLAGS_MESSAGE_DATA,

		(EAPL("eapol_handle_tlv_message_data_c::add_parameter_data(): type=%s\n"),

		 get_type_string(eapol_tlv_message_type_u16_t)));

	const u16_t network_order_short_integer(eap_htons(short_integer));

	eap_status_e status = add_message_data(

		eapol_tlv_message_type_u16_t,

		sizeof(network_order_short_integer),

		&network_order_short_integer);

	EAP_TRACE_END(m_am_tools, TRACE_FLAGS_DEFAULT);

	return EAP_STATUS_RETURN(m_am_tools, status);

}

//--------------------------------------------------

EAP_FUNC_EXPORT eap_status_e eapol_handle_tlv_message_data_c::add_parameter_data(

	const u8_t byte_integer)

{

	EAP_TRACE_BEGIN(m_am_tools, TRACE_FLAGS_DEFAULT);

	EAP_TRACE_DEBUG(

		m_am_tools,

		EAP_TRACE_FLAGS_MESSAGE_DATA,

		(EAPL("eapol_handle_tlv_message_data_c::add_parameter_data(): type=%s\n"),

		 get_type_string(eapol_tlv_message_type_u8_t)));

	eap_status_e status = add_message_data(

		eapol_tlv_message_type_u8_t,

		sizeof(byte_integer),

		&byte_integer);

	EAP_TRACE_END(m_am_tools, TRACE_FLAGS_DEFAULT);

	return EAP_STATUS_RETURN(m_am_tools, status);

}

//--------------------------------------------------

EAP_FUNC_EXPORT eap_status_e eapol_handle_tlv_message_data_c::add_parameter_data(

	const bool boolean)

{

	const u32_t value((boolean == false) ? 0u: 1u);

	return add_parameter_data(

		eapol_tlv_message_type_boolean,

		value);

}

//--------------------------------------------------

EAP_FUNC_EXPORT eap_status_e eapol_handle_tlv_message_data_c::add_parameter_data(

	const eap_status_e status)

{

	const u32_t value(static_cast<u32_t>(status));

	return add_parameter_data(

		eapol_tlv_message_type_eap_status,

		value);

}

//--------------------------------------------------

EAP_FUNC_EXPORT eap_status_e eapol_handle_tlv_message_data_c::add_parameter_data(

	const eapol_tlv_message_type_function_e function)

{

	EAP_TRACE_BEGIN(m_am_tools, TRACE_FLAGS_DEFAULT);

	EAP_TRACE_DEBUG(

		m_am_tools,

		EAP_TRACE_FLAGS_MESSAGE_DATA,

		(EAPL("eapol_handle_tlv_message_data_c::add_parameter_data(): type=%s, function=%s\n"),

		 get_type_string(eapol_tlv_message_type_function),

		 get_function_string(function)));

	if (function < eapol_tlv_message_type_function_none

		|| function >= eapol_tlv_message_type_function_illegal_value)

	{

		EAP_TRACE_END(m_am_tools, TRACE_FLAGS_DEFAULT);

		return EAP_STATUS_RETURN(m_am_tools, eap_status_illegal_parameter);

	}

	const u32_t network_order_function(eap_htonl(function));

	eap_status_e status = add_message_data(

		eapol_tlv_message_type_function,

		sizeof(network_order_function),

		&network_order_function);

	EAP_TRACE_END(m_am_tools, TRACE_FLAGS_DEFAULT);

	return EAP_STATUS_RETURN(m_am_tools, status);

}

//--------------------------------------------------

EAP_FUNC_EXPORT eap_status_e eapol_handle_tlv_message_data_c::add_parameter_data(

	const eap_variable_data_c * const variable_data)

{

	EAP_TRACE_BEGIN(m_am_tools, TRACE_FLAGS_DEFAULT);

	EAP_TRACE_DEBUG(

		m_am_tools,

		EAP_TRACE_FLAGS_MESSAGE_DATA,

		(EAPL("eapol_handle_tlv_message_data_c::add_parameter_data(): type=%s\n"),

		 get_type_string(eapol_tlv_message_type_variable_data)));

	if (variable_data == 0)

	{

		EAP_TRACE_END(m_am_tools, TRACE_FLAGS_DEFAULT);

		return EAP_STATUS_RETURN(m_am_tools, eap_status_illegal_parameter);

	}

	eap_status_e status(eap_status_process_general_error);

	if (variable_data->get_is_valid_data() == false)

	{

		// Empty variable data. Add just the header.

		status = add_structured_parameter_header(

			eapol_tlv_message_type_variable_data,

			0ul);

	}

	else

	{

		status = add_message_data(

			eapol_tlv_message_type_variable_data,

			variable_data->get_data_length(),

			variable_data->get_data());

	}

	EAP_TRACE_END(m_am_tools, TRACE_FLAGS_DEFAULT);

	return EAP_STATUS_RETURN(m_am_tools, status);

}

//--------------------------------------------------

EAP_FUNC_EXPORT eap_status_e eapol_handle_tlv_message_data_c::add_parameter_data(

	const eap_am_network_id_c * const network_id)

{

	EAP_TRACE_BEGIN(m_am_tools, TRACE_FLAGS_DEFAULT);

	EAP_TRACE_DEBUG(

		m_am_tools,

		EAP_TRACE_FLAGS_MESSAGE_DATA,

		(EAPL("eapol_handle_tlv_message_data_c::add_parameter_data(): type=%s\n"),

		 get_type_string(eapol_tlv_message_type_network_id)));

	if (network_id == 0

		|| network_id->get_is_valid() == false)

	{

		EAP_TRACE_END(m_am_tools, TRACE_FLAGS_DEFAULT);

		return EAP_STATUS_RETURN(m_am_tools, eap_status_illegal_parameter);

	}

	const u32_t size_of_network_id = get_payload_size(network_id);

	eap_status_e status = add_structured_parameter_header(

		eapol_tlv_message_type_network_id,

		size_of_network_id);

	if (status != eap_status_ok)

	{

		EAP_TRACE_END(m_am_tools, TRACE_FLAGS_DEFAULT);

		return EAP_STATUS_RETURN(m_am_tools, status);

	}

	status = add_parameter_data(

		network_id->get_source_id());

	if (status != eap_status_ok)

	{

		EAP_TRACE_END(m_am_tools, TRACE_FLAGS_DEFAULT);

		return EAP_STATUS_RETURN(m_am_tools, status);

	}

	status = add_parameter_data(

		network_id->get_destination_id());

	if (status != eap_status_ok)

	{

		EAP_TRACE_END(m_am_tools, TRACE_FLAGS_DEFAULT);

		return EAP_STATUS_RETURN(m_am_tools, status);

	}

	status = add_parameter_data(

		network_id->get_type());

	if (status != eap_status_ok)

	{

		EAP_TRACE_END(m_am_tools, TRACE_FLAGS_DEFAULT);

		return EAP_STATUS_RETURN(m_am_tools, status);

	}