/*

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

	#undef EAP_FILE_NUMBER_DATE 

	#define EAP_FILE_NUMBER_DATE 1127594498 

#endif //#if defined(USE_EAP_MINIMUM_RELEASE_TRACES)

#if defined(USE_EAP_SIMPLE_CONFIG)

#include "eap_am_memory.h"

#include "eap_am_export.h"

#include "eap_am_tools.h"

#include "eap_tools.h"

#include "eap_crypto_api.h"

#include "abs_simple_config_base_record.h"

#include "simple_config_base_record.h"

#include "simple_config_record.h"

#include "simple_config_am_services.h"

#include "simple_config_types.h"

#include "simple_config_message.h"

#include "eap_automatic_variable.h"

#include "eap_state_notification.h"

#include "eap_type_simple_config_types.h"

#include "eap_header_string.h"

#include "abs_eap_am_mutex.h"

#include "simple_config_credential.h"

#include "eapol_key_types.h"

#include "eap_config.h"

#include "eap_network_id_selector.h"

#include "eap_tlv_message_data.h"

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

EAP_FUNC_EXPORT simple_config_record_c::~simple_config_record_c()

{

	EAP_TRACE_BEGIN(m_am_tools, TRACE_FLAGS_DEFAULT);

	EAP_TRACE_DEBUG(m_am_tools, TRACE_FLAGS_DEFAULT,

					(EAPL("%s: function: simple_config_record_c::~simple_config_record_c(): this = 0x%08x, m_am_simple_config_services")

					 EAPL(" = 0x%08x (validity %d).\n"),

					 (m_is_client == true ? "client": "server"),

					 this,

					 m_am_simple_config_services,

					 m_am_simple_config_services->get_is_valid()));

	EAP_ASSERT(m_shutdown_was_called == true);

	completion_action_clenup();

	if (m_free_am_simple_config_services == true)

	{

		delete m_am_simple_config_services;

	}

	m_am_simple_config_services = 0;

	EAP_TRACE_END(m_am_tools, TRACE_FLAGS_DEFAULT);

}

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

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

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

#endif

EAP_FUNC_EXPORT simple_config_record_c::simple_config_record_c(

	abs_eap_am_tools_c * const tools, ///< tools is pointer to the tools class. @see abs_eap_am_tools_c.

	simple_config_am_services_c * const am_simple_config_services, ///< This is pointer to adaptation module of SIMPLE_CONFIG.

	const bool free_am_simple_config_services,

	const bool is_client_when_true, ///< Indicates whether this is client (true) or server (false).

	const eap_am_network_id_c * const receive_network_id)

	: simple_config_base_record_c(tools /*, partner */)

	, m_am_tools(tools)

	, m_am_simple_config_services(am_simple_config_services)

	, m_free_am_simple_config_services(free_am_simple_config_services)

	, m_completion_queue(tools)

	, m_M2D_payloads(tools)

	, m_received_simple_config_message(tools, is_client_when_true)

	, m_received_payloads(tools)

	, m_previous_simple_config_message(tools, is_client_when_true)

	, m_new_simple_config_message(tools, is_client_when_true)

	, m_current_simple_config_message_type(simple_config_Message_Type_None)

	, m_enrollee_nonce(tools)

	, m_enrollee_mac(tools)

	, m_registrar_nonce(tools)

	, m_device_password(tools)

	, m_PSK1(tools)

	, m_PSK2(tools)

	, m_E_SNonce1(tools)

	, m_E_SNonce2(tools)

	, m_EHash1(tools)

	, m_EHash2(tools)

	, m_R_SNonce1(tools)

	, m_R_SNonce2(tools)

	, m_RHash1(tools)

	, m_RHash2(tools)

	, m_own_private_dhe_key(tools)

	, m_own_public_dhe_key(tools)

	, m_peer_public_dhe_key(tools)

	, m_shared_dh_key(tools)

	, m_dhe_prime(tools)

	, m_dhe_group_generator(tools)

	, m_kdk(tools)

	, m_auth_key(tools)

	, m_key_wrap_key(tools)

	, m_EMSK(tools)

	, m_SSID(tools)

	, m_signed_message_hash(tools)

	, m_NAI(tools)

	, m_NAI_realm(tools)

	, m_send_network_id(tools)

#if defined(USE_EAP_TYPE_SERVER_SIMPLE_CONFIG)

	, m_network_key(tools)

	, m_authentication_type(simple_config_Authentication_Type_None)

#endif //#if defined(USE_EAP_TYPE_SERVER_SIMPLE_CONFIG)

	, m_simple_config_state(simple_config_state_wait_simple_config_start)

	, m_handshake_error(eap_status_ok)

	, m_UUID_E(tools)

	, m_UUID_R(tools)

	, m_Rf_Bands(simple_config_RF_Bands_2_4_GHz)

	, m_MAC_address(tools)

	, m_local_Device_Password_ID(simple_config_Device_Password_ID_Default_PIN)

	, m_received_Device_Password_ID(simple_config_Device_Password_ID_Default_PIN)

	, m_new_password(tools)

	, m_new_Device_Password_ID(simple_config_Device_Password_ID_Default_PIN)

	, m_error_message_received_timeout(SIMPLE_CONFIG_RECORD_ERROR_MESSAGE_RECEIVED_TIMEOUT)

	, m_is_valid(false)

	, m_is_client(is_client_when_true)

	, m_allow_message_send(true)

	, m_already_in_completion_action_check(false)

	, m_pending_query_network_and_device_parameters(false)

	, m_simple_config_test_version(false)

	, m_key_material_generated(false)

	, m_force_simple_config_message_send(false)

	, m_shutdown_was_called(false)

	, m_M2D_received_timeout_active(false)

{

	if (m_am_tools == 0

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

	{

		// No need to delete anything here because it is done in destructor.

		EAP_TRACE_END(m_am_tools, TRACE_FLAGS_DEFAULT);

		return;

	}

	EAP_TRACE_BEGIN(m_am_tools, TRACE_FLAGS_DEFAULT);

	EAP_TRACE_DEBUG(

		m_am_tools,

		TRACE_FLAGS_DEFAULT,

		(EAPL("SIMPLE_CONFIG: %s: function: simple_config_record_c::simple_config_record_c(): ")

		 EAPL("this = 0x%08x\n"),

		(m_is_client == true ? "client": "server"),

		this));

	if (receive_network_id == 0

		|| receive_network_id->get_is_valid_data() == false)

	{

		// No need to delete anything here because it is done in destructor.

		EAP_TRACE_END(m_am_tools, TRACE_FLAGS_DEFAULT);

		return;

	}

	if (m_am_simple_config_services == 0

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

	{

		EAP_TRACE_ERROR(m_am_tools, TRACE_FLAGS_SIMPLE_CONFIG_ERROR,

						(EAPL("ERROR: %s: function: simple_config_record_c::simple_config_record_c() failed,")

						 EAPL(" m_am_simple_config_services = 0x%08x (validity %d) is invalid.\n"),

						 (m_is_client == true ? "client": "server"),

						 m_am_simple_config_services, (m_am_simple_config_services != 0) ? m_am_simple_config_services->get_is_valid(): false));

		EAP_TRACE_END(m_am_tools, TRACE_FLAGS_DEFAULT);

		return;

	}

	m_am_simple_config_services->set_simple_config_am_partner(this);

	// Here we swap the addresses.

	eap_am_network_id_c send_network_id(

		m_am_tools,

		receive_network_id->get_destination_id(),

		receive_network_id->get_source_id(),

		receive_network_id->get_type());

	if (send_network_id.get_is_valid() == false)

	{

		EAP_TRACE_END(m_am_tools, TRACE_FLAGS_DEFAULT);

		return;

	}

	if (m_send_network_id.get_is_valid() == false)

	{

		EAP_TRACE_END(m_am_tools, TRACE_FLAGS_DEFAULT);

		return;

	}

	eap_status_e status = m_send_network_id.set_copy_of_network_id(&send_network_id);

	if (status != eap_status_ok)

	{

		EAP_TRACE_END(m_am_tools, TRACE_FLAGS_DEFAULT);

		return;

	}

	if (m_send_network_id.get_is_valid() == false

		|| m_received_simple_config_message.get_is_valid() == false

		|| m_received_payloads.get_is_valid() == false

		|| m_previous_simple_config_message.get_is_valid() == false

		|| m_new_simple_config_message.get_is_valid() == false

		|| m_enrollee_nonce.get_is_valid() == false

		|| m_enrollee_mac.get_is_valid() == false

		|| m_registrar_nonce.get_is_valid() == false

		|| m_device_password.get_is_valid() == false

		|| m_PSK1.get_is_valid() == false

		|| m_PSK2.get_is_valid() == false

		|| m_E_SNonce1.get_is_valid() == false

		|| m_E_SNonce2.get_is_valid() == false

		|| m_EHash1.get_is_valid() == false

		|| m_EHash2.get_is_valid() == false

		|| m_R_SNonce1.get_is_valid() == false

		|| m_R_SNonce2.get_is_valid() == false

		|| m_RHash1.get_is_valid() == false

		|| m_RHash2.get_is_valid() == false

		|| m_own_private_dhe_key.get_is_valid() == false

		|| m_own_public_dhe_key.get_is_valid() == false

		|| m_peer_public_dhe_key.get_is_valid() == false

		|| m_shared_dh_key.get_is_valid() == false

		|| m_dhe_prime.get_is_valid() == false

		|| m_dhe_group_generator.get_is_valid() == false

		|| m_kdk.get_is_valid() == false

		|| m_auth_key.get_is_valid() == false

		|| m_key_wrap_key.get_is_valid() == false

		|| m_EMSK.get_is_valid() == false

		|| m_SSID.get_is_valid() == false

		|| m_signed_message_hash.get_is_valid() == false

		|| m_NAI.get_is_valid() == false

		|| m_NAI_realm.get_is_valid() == false

		|| m_send_network_id.get_is_valid() == false

		|| m_UUID_E.get_is_valid() == false

		|| m_UUID_R.get_is_valid() == false

		|| m_MAC_address.get_is_valid() == false

		|| m_new_password.get_is_valid() == false)

	{

		EAP_TRACE_END(m_am_tools, TRACE_FLAGS_DEFAULT);

		return;

	}

#if defined(USE_EAP_TYPE_SERVER_SIMPLE_CONFIG)

	if (m_is_client == false)

	{

		set_state(simple_config_state_wait_M1);

	}

#endif //#if defined(USE_EAP_TYPE_SERVER_SIMPLE_CONFIG)

	set_is_valid();

	EAP_TRACE_END(m_am_tools, TRACE_FLAGS_DEFAULT);

}

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

EAP_FUNC_EXPORT void simple_config_record_c::set_state(const simple_config_state_e state)

{

	EAP_TRACE_BEGIN(m_am_tools, TRACE_FLAGS_DEFAULT);

	eap_simple_config_trace_string_c state_string;

	EAP_TRACE_DEBUG(

		m_am_tools,

		TRACE_FLAGS_DEFAULT,

		(EAPL("SIMPLE_CONFIG: %s: state_function: set_state() from %s to %s\n"),

		(m_is_client == true ? "client": "server"),

		state_string.get_state_string(m_simple_config_state),

		state_string.get_state_string(state)));

	if (m_simple_config_state != simple_config_state_failure)

	{

		m_simple_config_state = state;

	}

	EAP_TRACE_END(m_am_tools, TRACE_FLAGS_DEFAULT);

}

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

EAP_FUNC_EXPORT simple_config_state_e simple_config_record_c::get_state() const

{

	EAP_TRACE_BEGIN(m_am_tools, TRACE_FLAGS_DEFAULT);

	EAP_TRACE_END(m_am_tools, TRACE_FLAGS_DEFAULT);

	return m_simple_config_state;

}

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

EAP_FUNC_EXPORT bool simple_config_record_c::verify_state(const simple_config_state_e state)

{

	EAP_TRACE_BEGIN(m_am_tools, TRACE_FLAGS_DEFAULT);

	eap_simple_config_trace_string_c state_string;

	if (m_simple_config_state == state)

	{

		EAP_TRACE_DEBUG(

			m_am_tools,

			TRACE_FLAGS_DEFAULT,

			(EAPL("SIMPLE_CONFIG: %s: state_function: verify_state(): current state %s == %s\n"),

			 (m_is_client == true ? "client": "server"),

			 state_string.get_state_string(m_simple_config_state),

			 state_string.get_state_string(state)));

		EAP_TRACE_END(m_am_tools, TRACE_FLAGS_DEFAULT);

		return true;

	}

	else

	{

		EAP_TRACE_DEBUG(

			m_am_tools,

			TRACE_FLAGS_DEFAULT,

			(EAPL("ERROR: SIMPLE_CONFIG: %s: state_function: verify_state(): current state %s != %s\n"),

			 (m_is_client == true ? "client": "server"),

			 state_string.get_state_string(m_simple_config_state),

			 state_string.get_state_string(state)));

		EAP_TRACE_END(m_am_tools, TRACE_FLAGS_DEFAULT);

		return false;

	}

}

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

EAP_FUNC_EXPORT void simple_config_record_c::set_is_valid()

{

	EAP_TRACE_BEGIN(m_am_tools, TRACE_FLAGS_DEFAULT);

	m_is_valid = true;

	EAP_TRACE_END(m_am_tools, TRACE_FLAGS_DEFAULT);

}

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

EAP_FUNC_EXPORT eap_status_e simple_config_record_c::configure()

{

	EAP_TRACE_BEGIN(m_am_tools, TRACE_FLAGS_DEFAULT);

	EAP_TRACE_DEBUG(

		m_am_tools,

		TRACE_FLAGS_DEFAULT,

		(EAPL("SIMPLE_CONFIG: %s: function: simple_config_record_c::configure(): ")

		 EAPL("this = 0x%08x\n"),

		(m_is_client == true ? "client": "server"),

		this));

	eap_status_e status = m_am_simple_config_services->configure();

	if (status != eap_status_ok)

	{

		EAP_TRACE_END(m_am_tools, TRACE_FLAGS_DEFAULT);

		return EAP_STATUS_RETURN(m_am_tools, status);

	}

	if (get_type_partner() == 0)

	{

		EAP_TRACE_END(m_am_tools, TRACE_FLAGS_DEFAULT);

		return EAP_STATUS_RETURN(m_am_tools, eap_status_allocation_error);

	}

	if (m_is_client == true)

	{

		// Creates Enrollee Nonce.

		status = generate_nonce(

			&m_enrollee_nonce,

			SIMPLE_CONFIG_ENROLLEE_NONCE_SIZE);

		if (status != eap_status_ok)

		{

			EAP_TRACE_END(m_am_tools, TRACE_FLAGS_DEFAULT);

			return EAP_STATUS_RETURN(m_am_tools, status);

		}

	}

	else

	{

		// Creates Registrar Nonce.

		status = generate_nonce(

			&m_registrar_nonce,

			SIMPLE_CONFIG_REGISTRAR_NONCE_SIZE);

		if (status != eap_status_ok)

		{

			EAP_TRACE_END(m_am_tools, TRACE_FLAGS_DEFAULT);

			return EAP_STATUS_RETURN(m_am_tools, status);

		}

	}

	status = generate_dhe_keys();

	if (status != eap_status_ok)

	{

		EAP_TRACE_END(m_am_tools, TRACE_FLAGS_DEFAULT);

		return EAP_STATUS_RETURN(m_am_tools, status);

	}

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

	if (m_is_client == false)

	{

		status = get_type_partner()->read_configure(

			cf_str_EAP_SIMPLE_CONFIG_device_password.get_field(),

			&m_device_password);

		if (status != eap_status_ok

			|| m_device_password.get_is_valid_data() == false)

		{

			// This is mandatory value.

			EAP_TRACE_ERROR(

				m_am_tools,

				TRACE_FLAGS_DEFAULT,

				(EAPL("ERROR: SIMPLE_CONFIG: %s: simple_config_record_c::configure(): Missing device password.\n"),

				 (m_is_client == true ? "client": "server")));

			EAP_TRACE_END(m_am_tools, TRACE_FLAGS_DEFAULT);

			return EAP_STATUS_RETURN(m_am_tools, eap_status_wrong_password);

		}

		(void) get_type_partner()->read_configure(

			cf_str_EAP_SIMPLE_CONFIG_server_device_password.get_field(),

			&m_device_password);

	}

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

	{

		eap_variable_data_c test_version(m_am_tools);

		if (test_version.get_is_valid() == false)

		{

			EAP_TRACE_END(m_am_tools, TRACE_FLAGS_DEFAULT);

			return EAP_STATUS_RETURN(m_am_tools, eap_status_allocation_error);

		}

		status = get_type_partner()->read_configure(

			cf_str_EAP_SIMPLE_CONFIG_test_version.get_field(),

			&test_version);

		if (status == eap_status_ok

			&& test_version.get_is_valid_data() == true

			&& test_version.get_data_length() == sizeof(u32_t)

			&& test_version.get_data(sizeof(u32_t)) != 0)

		{

			// This is optional value.

			u32_t *flag = reinterpret_cast<u32_t *>(test_version.get_data(sizeof(u32_t)));

			if (flag != 0)

			{

				if (*flag == 0)

				{

					m_simple_config_test_version = false;

				}

				else

				{

					m_simple_config_test_version = true;

				}

			}

		}

		status = eap_status_ok;

	}

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

	{

		eap_variable_data_c error_message_received_timeout(m_am_tools);

		if (error_message_received_timeout.get_is_valid() == false)

		{

			EAP_TRACE_END(m_am_tools, TRACE_FLAGS_DEFAULT);

			return EAP_STATUS_RETURN(m_am_tools, eap_status_allocation_error);

		}

		eap_status_e status = read_configure(

			cf_str_SIMPLE_CONFIG_error_message_received_timeout.get_field(),

			&error_message_received_timeout);

		if (status == eap_status_ok

			&& error_message_received_timeout.get_is_valid_data() == true)

		{

			u32_t *error_message_received_timeout_value = reinterpret_cast<u32_t *>(

				error_message_received_timeout.get_data(sizeof(u32_t)));

			if (error_message_received_timeout_value != 0)

			{

				m_error_message_received_timeout = *error_message_received_timeout_value;

			}

		}

	}

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

#if defined(USE_EAP_TYPE_SERVER_SIMPLE_CONFIG)

	if (m_is_client == false)

	{

		{

			// This is optional.

			(void) read_configure(

				cf_str_SIMPLE_CONFIG_new_password.get_field(),

				&m_new_password);

		}

		{

			// This is optional.

			(void) read_configure(

				cf_str_SIMPLE_CONFIG_network_key.get_field(),

				&m_network_key);

		}

		{

			eap_variable_data_c authentication_type(m_am_tools);