/*

* Copyright (c) 2001-2007 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 589 

	#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 "simple_config_payloads.h"

#include "simple_config_tlv_header.h"

#include "simple_config_message.h"

#include "abs_eap_am_tools.h"

#include "eap_tools.h"

#include "eap_array_algorithms.h"

#include "eap_automatic_variable.h"

#include "eap_crypto_api.h"

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

EAP_FUNC_EXPORT simple_config_variable_data_c::~simple_config_variable_data_c()

{

	delete m_next_payload_with_same_attribute_type;

	m_next_payload_with_same_attribute_type = 0;

}

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

EAP_FUNC_EXPORT simple_config_variable_data_c::simple_config_variable_data_c(

	abs_eap_am_tools_c * const tools)

	: m_am_tools(tools)

	  , m_data(tools)

	  , m_header(tools, 0, 0ul)

	  , m_next_payload_with_same_attribute_type(0)

	  , m_is_mandatory(false)

	  , m_is_valid(false)

{

	if (m_data.get_is_valid() == false)

	{

		return;

	}

	m_is_valid = true;

}

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

EAP_FUNC_EXPORT bool simple_config_variable_data_c::get_is_valid() const

{

	return m_is_valid;

}

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

EAP_FUNC_EXPORT eap_status_e simple_config_variable_data_c::set_copy_of_buffer(

	const simple_config_Attribute_Type_e current_payload,

	const bool is_mandatory,

	const void * const buffer,

	const u32_t buffer_length)

{

	eap_status_e status = m_data.set_buffer_length(

		simple_config_tlv_header_c::get_header_length() + buffer_length);

	if (status != eap_status_ok)

	{

		EAP_TRACE_END(m_am_tools, TRACE_FLAGS_DEFAULT);

		return EAP_STATUS_RETURN(m_am_tools, status);

	}

	status = m_data.set_data_length(

		simple_config_tlv_header_c::get_header_length());

	if (status != eap_status_ok)

	{

		EAP_TRACE_END(m_am_tools, TRACE_FLAGS_DEFAULT);

		return EAP_STATUS_RETURN(m_am_tools, status);

	}

	m_header.set_header_buffer(

		m_data.get_buffer(m_data.get_buffer_length()),

		m_data.get_buffer_length());

	m_header.set_tlv_type(current_payload);

	if (buffer_length > 0xffff)

	{

		EAP_TRACE_END(m_am_tools, TRACE_FLAGS_DEFAULT);

		return EAP_STATUS_RETURN(m_am_tools, eap_status_illegal_parameter);

	}

	m_header.set_data_length(static_cast<u16_t>(buffer_length));

	status = m_data.add_data(buffer, buffer_length);

	if (status != eap_status_ok)

	{

		EAP_TRACE_END(m_am_tools, TRACE_FLAGS_DEFAULT);

		return EAP_STATUS_RETURN(m_am_tools, status);

	}

	m_is_mandatory = is_mandatory;

	EAP_TRACE_END(m_am_tools, TRACE_FLAGS_DEFAULT);

	return EAP_STATUS_RETURN(m_am_tools, status);

}

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

EAP_FUNC_EXPORT eap_status_e simple_config_variable_data_c::add_data(

	const void * const buffer,

	const u32_t buffer_length)

{

	const simple_config_Attribute_Type_e current_payload = m_header.get_tlv_type();

	eap_status_e status = m_data.add_data(

		buffer,

		buffer_length);

	if (status != eap_status_ok)

	{

		EAP_TRACE_END(m_am_tools, TRACE_FLAGS_DEFAULT);

		return EAP_STATUS_RETURN(m_am_tools, status);

	}

	m_header.set_header_buffer(

		m_data.get_buffer(m_data.get_buffer_length()),

		m_data.get_buffer_length());

	m_header.set_tlv_type(current_payload);

	if ((m_header.get_data_length() + buffer_length) > 0xffff)

	{

		EAP_TRACE_END(m_am_tools, TRACE_FLAGS_DEFAULT);

		return EAP_STATUS_RETURN(m_am_tools, eap_status_illegal_parameter);

	}

	m_header.set_data_length(static_cast<u16_t>(m_header.get_data_length() + buffer_length));

	EAP_TRACE_END(m_am_tools, TRACE_FLAGS_DEFAULT);

	return EAP_STATUS_RETURN(m_am_tools, status);

}

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

EAP_FUNC_EXPORT u32_t simple_config_variable_data_c::get_data_length() const

{

	return m_header.get_data_length();

}

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

EAP_FUNC_EXPORT u8_t * simple_config_variable_data_c::get_data(

	const u32_t data_length) const

{

	return m_header.get_data(data_length);

}

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

EAP_FUNC_EXPORT eap_variable_data_c * simple_config_variable_data_c::get_full_attribute_buffer()

{

	return &m_data;

}

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

EAP_FUNC_EXPORT simple_config_tlv_header_c * simple_config_variable_data_c::get_header()

{

	return &m_header;

}

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

EAP_FUNC_EXPORT simple_config_Attribute_Type_e simple_config_variable_data_c::get_attribute_type() const

{

	return m_header.get_tlv_type();

}

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

EAP_FUNC_EXPORT bool simple_config_variable_data_c::get_is_mandatory() const

{

	return m_is_mandatory;

}

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

EAP_FUNC_EXPORT void simple_config_variable_data_c::set_attribute_type(

	const simple_config_Attribute_Type_e type)

{

	m_header.set_tlv_type(type);

}

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

EAP_FUNC_EXPORT void simple_config_variable_data_c::set_next_payload_with_same_attribute_type(

	simple_config_variable_data_c * const attribute)

{

	m_next_payload_with_same_attribute_type = attribute;

}

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

EAP_FUNC_EXPORT simple_config_variable_data_c * simple_config_variable_data_c::get_next_payload_with_same_attribute_type()

{

	return m_next_payload_with_same_attribute_type;

}

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

EAP_FUNC_EXPORT void simple_config_variable_data_c::add_next_payload_with_same_attribute_type(

	simple_config_variable_data_c * const attribute)

{

	simple_config_variable_data_c *payload = get_next_payload_with_same_attribute_type();

	simple_config_variable_data_c *prev_payload = this;

	while (payload != 0)

	{

		prev_payload = payload;

		payload = payload->get_next_payload_with_same_attribute_type();

	}

	if (prev_payload != 0)

	{

		prev_payload->set_next_payload_with_same_attribute_type(attribute);

	}

}

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

EAP_FUNC_EXPORT simple_config_variable_data_c * simple_config_variable_data_c::copy() const

{

	simple_config_variable_data_c * new_data = new simple_config_variable_data_c(m_am_tools);

	if (new_data != 0)

	{

		eap_status_e status = new_data->set_copy_of_buffer(

			get_attribute_type(),

			get_is_mandatory(),

			get_data(get_data_length()),

			get_data_length());

		if (status != eap_status_ok)

		{

			delete new_data;

			EAP_TRACE_END(m_am_tools, TRACE_FLAGS_DEFAULT);

			(void) EAP_STATUS_RETURN(m_am_tools, eap_status_allocation_error);

			return 0;

		}

	}

	return new_data;

}

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

EAP_FUNC_EXPORT  void simple_config_variable_data_c::object_increase_reference_count()

{

}

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

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

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

EAP_FUNC_EXPORT simple_config_payloads_c::~simple_config_payloads_c()

{

}

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

#if defined(_WIN32) && !defined(__GNUC__)

	#pragma warning( disable : 4355 ) // 'this' : used in base member initializer list

#endif

EAP_FUNC_EXPORT simple_config_payloads_c::simple_config_payloads_c(

	abs_eap_am_tools_c * const tools)

	: m_am_tools(tools)

	  , m_payload_map(tools, this)

	  , m_read_payloads(tools)

	  , m_payload_index(0ul)

	  , m_is_valid(false)

{

	m_is_valid = true;

}

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

EAP_FUNC_EXPORT simple_config_variable_data_c * simple_config_payloads_c::get_attribute_pointer(

	const simple_config_Attribute_Type_e current_payload,

	u32_t index) const

{

	eap_variable_data_c selector(m_am_tools);

	if (selector.get_is_valid() == false)

	{

		EAP_TRACE_END(m_am_tools, TRACE_FLAGS_DEFAULT);

		(void) EAP_STATUS_RETURN(m_am_tools, eap_status_allocation_error);

		return 0;

	}

	selector.set_buffer(

		&current_payload,

		sizeof(current_payload),

		false,

		false);

	simple_config_variable_data_c *payload = m_payload_map.get_handler(&selector);

	while (index != 0ul && payload != 0)

	{

		--index;

		payload = payload->get_next_payload_with_same_attribute_type();

	}

	return payload;

}

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

EAP_FUNC_EXPORT simple_config_variable_data_c * simple_config_payloads_c::get_attribute_pointer(

	const simple_config_Attribute_Type_e current_payload) const

{

	return get_attribute_pointer(current_payload, 0ul);

}

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

EAP_FUNC_EXPORT u32_t simple_config_payloads_c::get_attribute_count() const

{

	return m_read_payloads.get_object_count();

}

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

EAP_FUNC_EXPORT simple_config_variable_data_c * simple_config_payloads_c::get_attribute(

	const u32_t attribute_index) const

{

	EAP_TRACE_DEBUG(

		m_am_tools,

		TRACE_FLAGS_DEFAULT,

		(EAPL("SIMPLE_CONFIG:     message_function: simple_config_payloads_c::get_attribute(index %d, max %d)\n"),

		attribute_index,

		m_read_payloads.get_object_count()));

	simple_config_variable_data_c *payload = m_read_payloads.get_object(attribute_index);

	return payload;

}

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

EAP_FUNC_EXPORT eap_status_e simple_config_payloads_c::check_payloads_existense(

	const simple_config_Attribute_Type_e * const needed_payloads,

	const u32_t count_of_needed_payloads) const

{

	EAP_TRACE_BEGIN(m_am_tools, TRACE_FLAGS_DEFAULT);

	EAP_TRACE_DEBUG(

		m_am_tools,

		TRACE_FLAGS_DEFAULT,

		(EAPL("SIMPLE_CONFIG:     message_function: simple_config_payloads_c::check_payloads_existense()\n")));

	for (u32_t ind = 0ul; ind < count_of_needed_payloads; ind++)

	{

		const simple_config_Attribute_Type_e required_avp_code = needed_payloads[ind];

		if (required_avp_code == 0)

		{

			EAP_TRACE_END(m_am_tools, TRACE_FLAGS_DEFAULT);

			return EAP_STATUS_RETURN(m_am_tools, eap_status_allocation_error);

		}

		if (get_attribute_pointer(required_avp_code) == 0)

		{

			EAP_TRACE_DEBUG(

				m_am_tools, 

				TRACE_FLAGS_DEFAULT, 

				(EAPL("not received TLV 0x%08x.\n"),

				 required_avp_code));

			EAP_TRACE_END(m_am_tools, TRACE_FLAGS_DEFAULT);

			return EAP_STATUS_RETURN(m_am_tools, eap_status_not_found);

		}

	}

	EAP_TRACE_END(m_am_tools, TRACE_FLAGS_DEFAULT);

	return EAP_STATUS_RETURN(m_am_tools, eap_status_ok);

}

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

EAP_FUNC_EXPORT eap_status_e simple_config_payloads_c::check_payloads_existense(

	EAP_TEMPLATE_CONST eap_array_c<simple_config_Attribute_Type_e> * const needed_payloads) const

{

	EAP_TRACE_BEGIN(m_am_tools, TRACE_FLAGS_DEFAULT);

	EAP_TRACE_DEBUG(

		m_am_tools,

		TRACE_FLAGS_DEFAULT,

		(EAPL("SIMPLE_CONFIG:     message_function: simple_config_payloads_c::check_payloads_existense()\n")));

	for (u32_t ind = 0ul; ind < needed_payloads->get_object_count(); ind++)

	{

		const simple_config_Attribute_Type_e * const required_avp_code = needed_payloads->get_object(ind);

		if (required_avp_code == 0)

		{

			EAP_TRACE_END(m_am_tools, TRACE_FLAGS_DEFAULT);

			return EAP_STATUS_RETURN(m_am_tools, eap_status_allocation_error);

		}

		if (get_attribute_pointer(*required_avp_code) == 0)

		{

			EAP_TRACE_DEBUG(

				m_am_tools, 

				TRACE_FLAGS_DEFAULT, 

				(EAPL("not received TLV 0x%08x.\n"),

				 *required_avp_code));

			EAP_TRACE_END(m_am_tools, TRACE_FLAGS_DEFAULT);

			return EAP_STATUS_RETURN(m_am_tools, eap_status_not_found);

		}

	}

	EAP_TRACE_END(m_am_tools, TRACE_FLAGS_DEFAULT);

	return EAP_STATUS_RETURN(m_am_tools, eap_status_ok);

}

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

EAP_FUNC_EXPORT eap_status_e simple_config_payloads_c::check_mandatory_payloads(

	EAP_TEMPLATE_CONST eap_array_c<simple_config_Attribute_Type_e> * const used_payloads) const

{

	EAP_TRACE_BEGIN(m_am_tools, TRACE_FLAGS_DEFAULT);

	EAP_TRACE_DEBUG(

		m_am_tools,

		TRACE_FLAGS_DEFAULT,

		(EAPL("SIMPLE_CONFIG:     message_function: simple_config_payloads_c::check_mandatory_payloads()\n")));

	for (u32_t ind = 0ul; ind < m_read_payloads.get_object_count(); ind++)

	{

		const simple_config_variable_data_c * const read_payload = m_read_payloads.get_object(ind);

		if (read_payload == 0)

		{

			EAP_TRACE_END(m_am_tools, TRACE_FLAGS_DEFAULT);

			return EAP_STATUS_RETURN(m_am_tools, eap_status_allocation_error);

		}

		if (read_payload->get_is_mandatory() == true)

		{

			simple_config_Attribute_Type_e attribute_type = read_payload->get_attribute_type();

			i32_t index = find_simple<simple_config_Attribute_Type_e>(

				used_payloads,

				&attribute_type,

				m_am_tools);

			if (index < 0ul)

			{

				// ERROR: not used mandatory TLV.

				EAP_TRACE_ERROR(

					m_am_tools, 

					TRACE_FLAGS_ERROR, 

					(EAPL("ERROR: not used mandatory TLV 0x%08x.\n"),

					 attribute_type));

				EAP_TRACE_END(m_am_tools, TRACE_FLAGS_DEFAULT);

				return EAP_STATUS_RETURN(m_am_tools, eap_status_unsupported_payload);

			}

		}

	}

	EAP_TRACE_END(m_am_tools, TRACE_FLAGS_DEFAULT);

	return EAP_STATUS_RETURN(m_am_tools, eap_status_ok);

}

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

EAP_FUNC_EXPORT eap_status_e simple_config_payloads_c::copy_attribute(

	const simple_config_payloads_c * const source,

	const simple_config_Attribute_Type_e attribute)

{

	EAP_TRACE_DEBUG(

		m_am_tools,

		TRACE_FLAGS_DEFAULT,

		(EAPL("SIMPLE_CONFIG:     message_function: simple_config_payloads_c::copy_attribute(TLV 0x%08x)\n"),

		attribute));

	const simple_config_variable_data_c * const payload

		= source->get_attribute_pointer(attribute, 0ul);

	if (payload == 0)

	{

		EAP_TRACE_END(m_am_tools, TRACE_FLAGS_DEFAULT);

		return EAP_STATUS_RETURN(m_am_tools, eap_status_missing_payload);

	}

	eap_status_e status = add_attribute(

		payload->copy());

	if (status != eap_status_ok)

	{

		EAP_TRACE_END(m_am_tools, TRACE_FLAGS_DEFAULT);

		return EAP_STATUS_RETURN(m_am_tools, status);

	}

	EAP_TRACE_END(m_am_tools, TRACE_FLAGS_DEFAULT);

	return EAP_STATUS_RETURN(m_am_tools, status);

}

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

EAP_FUNC_EXPORT eap_status_e simple_config_payloads_c::add_attribute(

	simple_config_variable_data_c * const new_payload)