/*

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

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

#include "eap_session_core.h"

#include "eap_state_notification.h"

#include "eap_network_id_selector.h"

#include "abs_eap_am_mutex.h"

#include "eap_config.h"

#include "eap_core.h"

#include "eap_buffer.h"

#include "eap_base_type.h"

#include "eap_automatic_variable.h"

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

// 

EAP_FUNC_EXPORT eap_session_core_c::~eap_session_core_c()

{

	EAP_TRACE_BEGIN(m_am_tools, TRACE_FLAGS_DEFAULT);

	EAP_TRACE_DEBUG(

		m_am_tools, 

		TRACE_FLAGS_DEFAULT, 

		(EAPL("eap_session_core_c::~eap_session_core_c(): this = 0x%08x => 0x%08x.\n"),

		this,

		dynamic_cast<abs_eap_base_timer_c *>(this)));

	EAP_ASSERT(m_shutdown_was_called == true);

	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 eap_session_core_c::eap_session_core_c(

	abs_eap_am_tools_c * const tools,

	abs_eap_session_core_c * const partner,

	const bool is_client_when_true)

: m_partner(partner)

, m_am_tools(tools)

, m_session_map(tools, this)

, m_remove_session_timeout(EAP_SESSION_CORE_REMOVE_SESSION_TIMEOUT)

, m_is_client(is_client_when_true)

, m_is_valid(false)

, m_use_eap_session_core_reset_session(true)

, m_shutdown_was_called(false)

{

	EAP_TRACE_BEGIN(m_am_tools, TRACE_FLAGS_DEFAULT);

	EAP_TRACE_DEBUG(

		m_am_tools, 

		TRACE_FLAGS_DEFAULT, 

		(EAPL("eap_session_core_c::eap_session_core_c(): this = 0x%08x => 0x%08x.\n"),

		this,

		dynamic_cast<abs_eap_base_timer_c *>(this)));

	set_is_valid();

	EAP_TRACE_END(m_am_tools, TRACE_FLAGS_DEFAULT);

}

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

//

EAP_FUNC_EXPORT abs_eap_session_core_c * eap_session_core_c::get_partner()

{

	EAP_ASSERT(m_am_tools->get_global_mutex()->get_is_reserved() == true);

	return m_partner;

}

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

//

EAP_FUNC_EXPORT void eap_session_core_c::set_is_valid()

{

	m_is_valid = true;

}

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

//

EAP_FUNC_EXPORT bool eap_session_core_c::get_is_valid()

{

	return m_is_valid;

}

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

EAP_FUNC_EXPORT eap_status_e eap_session_core_c::reset()

{

	EAP_TRACE_BEGIN(m_am_tools, TRACE_FLAGS_DEFAULT);

	EAP_TRACE_DEBUG(

		m_am_tools, 

		TRACE_FLAGS_DEFAULT, 

		(EAPL("eap_session_core_c::reset(): this = 0x%08x => 0x%08x.\n"),

		this,

		dynamic_cast<abs_eap_base_timer_c *>(this)));

	eap_status_e status = m_session_map.for_each(shutdown_operation, true);

	(void)EAP_STATUS_RETURN(m_am_tools, status);

	status = m_session_map.reset();

	(void)EAP_STATUS_RETURN(m_am_tools, status);

	m_partner->cancel_timer(this, EAP_SESSION_CORE_REMOVE_SESSION_ID);

	EAP_TRACE_DEBUG(

		m_am_tools, 

		TRACE_FLAGS_DEFAULT, 

		(EAPL("TIMER: EAP_SESSION_CORE_REMOVE_SESSION_ID cancelled, %s.\n"),

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

	EAP_TRACE_END(m_am_tools, TRACE_FLAGS_DEFAULT);

	return EAP_STATUS_RETURN(m_am_tools, status);

}

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

//

EAP_FUNC_EXPORT eap_status_e eap_session_core_c::load_module(

	const eap_type_value_e type,

	const eap_type_value_e tunneling_type,

	abs_eap_base_type_c * const partner,

	eap_base_type_c ** const eap_type,

	const bool is_client_when_true,

	const eap_am_network_id_c * const receive_network_id)

{

	EAP_TRACE_BEGIN(m_am_tools, TRACE_FLAGS_DEFAULT);

	EAP_ASSERT(m_am_tools->get_global_mutex()->get_is_reserved() == true);

	eap_status_e status = m_partner->load_module(

		type,

		tunneling_type,

		partner,

		eap_type,

		is_client_when_true,

		receive_network_id);

	EAP_TRACE_END(m_am_tools, TRACE_FLAGS_DEFAULT);

	return EAP_STATUS_RETURN(m_am_tools, status);

}

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

//

EAP_FUNC_EXPORT eap_core_c * eap_session_core_c::create_new_session(

	const eap_am_network_id_c * const receive_network_id)

{

	eap_status_e status = eap_status_process_general_error;

	// Create a new session.

	eap_core_c *session = new eap_core_c(

		m_am_tools,

		this,

		m_is_client,

		receive_network_id,

		false);

	if (session == 0)

	{

		EAP_TRACE_END(m_am_tools, TRACE_FLAGS_DEFAULT);

		(void)EAP_STATUS_RETURN(m_am_tools, eap_status_allocation_error);

		return 0;

	}

	if (session->get_is_valid() == false)

	{

		session->shutdown();

		delete session;

		EAP_TRACE_END(m_am_tools, TRACE_FLAGS_DEFAULT);

		(void)EAP_STATUS_RETURN(m_am_tools, eap_status_allocation_error);

		return 0;

	}

	status = session->configure();

	if (status != eap_status_ok)

	{

		session->shutdown();

		delete session;

		EAP_TRACE_END(m_am_tools, TRACE_FLAGS_DEFAULT);

		(void)EAP_STATUS_RETURN(m_am_tools, status);

		return 0;

	}

	// 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_data() == false)

	{

		session->shutdown();

		delete session;

		EAP_TRACE_END(m_am_tools, TRACE_FLAGS_DEFAULT);

		(void)EAP_STATUS_RETURN(m_am_tools, eap_status_allocation_error);

		return 0;

	}

	eap_network_id_selector_c selector(

		m_am_tools,

		&send_network_id);

	if (selector.get_is_valid() == false)

	{

		session->shutdown();

		delete session;

		EAP_TRACE_END(m_am_tools, TRACE_FLAGS_DEFAULT);

		(void)EAP_STATUS_RETURN(m_am_tools, eap_status_allocation_error);

		return 0;

	}

	EAP_TRACE_DATA_DEBUG(

		m_am_tools,

		TRACE_FLAGS_DEFAULT,

		(EAPL("eap_session_core_c::create_new_session() EAP-session"),

		 selector.get_data(selector.get_data_length()),

		 selector.get_data_length()));

	status = m_session_map.add_handler(&selector, session);

	if (status != eap_status_ok)

	{

		session->shutdown();

		delete session;

		EAP_TRACE_END(m_am_tools, TRACE_FLAGS_DEFAULT);

		(void)EAP_STATUS_RETURN(m_am_tools, status);

		return 0;

	}

	return session;

}

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

//

EAP_FUNC_EXPORT eap_status_e eap_session_core_c::reset_or_remove_session(

	eap_core_c ** const session,

	const eap_network_id_selector_c * const selector,

	const bool reset_immediately)

{

	EAP_TRACE_BEGIN(m_am_tools, TRACE_FLAGS_DEFAULT);

	EAP_TRACE_DEBUG(

		m_am_tools, 

		TRACE_FLAGS_DEFAULT, 

		(EAPL("eap_session_core_c::reset_or_remove_session(): %s.\n"),

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

	EAP_TRACE_RETURN_STRING_FLAGS(m_am_tools, TRACE_FLAGS_DEFAULT, "returns: eap_session_core_c::reset_or_remove_session()");

	eap_status_e status(eap_status_process_general_error);

	if (session == 0

		|| *session == 0

		|| selector == 0)

	{

		EAP_TRACE_END(m_am_tools, TRACE_FLAGS_DEFAULT);

		return EAP_STATUS_RETURN(m_am_tools, eap_status_illegal_parameter);

	}

	if (m_use_eap_session_core_reset_session == true)

	{

		// This will reuse session.

		EAP_TRACE_DEBUG(

			m_am_tools, 

			TRACE_FLAGS_DEFAULT, 

			(EAPL("eap_session_core_c::reset_or_remove_session(): resets session, session 0x%08x.\n"),

			(*session)));

		// NOTE, this delayed reset of session is used bacause the device is so slow it cannot respond to

		// 4-Way Handshake message fast enough.

		if (reset_immediately == true)

		{

			(*session)->unset_marked_removed();

			status = (*session)->reset();

		}

		else

		{

			// This will delay reset to eap_core_c::packet_process().

			status = eap_status_ok;

		}

	}

	else

	{

		//  This will cause shutdown of the session.

		status = eap_status_process_general_error;

	}

	if (status != eap_status_ok)

	{

		// We cannot reuse the session.

		EAP_TRACE_DEBUG(

			m_am_tools, 

			TRACE_FLAGS_DEFAULT, 

			(EAPL("eap_session_core_c::reset_or_remove_session(): shutdown session, session 0x%08x.\n"),

			(*session)));

		(*session)->shutdown();

		(*session) = 0;

		status = m_session_map.remove_handler(selector, true);

		if (status != eap_status_ok)

		{

			EAP_TRACE_DEBUG(

				m_am_tools, 

				TRACE_FLAGS_DEFAULT, 

				(EAPL("ERROR: eap_session_core_c::reset_or_remove_session(): m_session_map.remove_type(), eap_status_e %d\n"),

				status));

			EAP_TRACE_END(m_am_tools, TRACE_FLAGS_DEFAULT);

			return EAP_STATUS_RETURN(m_am_tools, status);

		}

		EAP_TRACE_DEBUG(

			m_am_tools, 

			TRACE_FLAGS_DEFAULT, 

			(EAPL("eap_session_core_c::reset_or_remove_session(): session NOT reused.\n")));

	}

	else

	{

		EAP_TRACE_DEBUG(

			m_am_tools, 

			TRACE_FLAGS_DEFAULT, 

			(EAPL("eap_session_core_c::reset_or_remove_session(): session reused, session 0x%08x.\n"),

			(*session)));

	}

	EAP_TRACE_END(m_am_tools, TRACE_FLAGS_DEFAULT);

	return EAP_STATUS_RETURN(m_am_tools, status);

}

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

//

EAP_FUNC_EXPORT eap_status_e eap_session_core_c::packet_process(

	const eap_am_network_id_c * const receive_network_id,

	eap_general_header_base_c * const packet_data,

	const u32_t packet_length)

{

	EAP_TRACE_BEGIN(m_am_tools, TRACE_FLAGS_DEFAULT);

	EAP_ASSERT(m_am_tools->get_global_mutex()->get_is_reserved() == true);

	eap_status_e status = eap_status_process_general_error;

	// Each EAP authentication session includes own eap_core_c object.

	// EAP authentication sessions are separated by eap_am_network_id_c object.

	if (packet_data == 0

		|| packet_length < eap_header_base_c::get_header_length())

	{

		EAP_TRACE_END(m_am_tools, TRACE_FLAGS_DEFAULT);

		return EAP_STATUS_RETURN(m_am_tools, eap_status_process_illegal_packet_error);

	}

	if (receive_network_id == 0

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

	{

		EAP_TRACE_END(m_am_tools, TRACE_FLAGS_DEFAULT);

		return EAP_STATUS_RETURN(m_am_tools, eap_status_illegal_parameter);

	}

	eap_header_wr_c eap(

		m_am_tools,

		packet_data->get_header_buffer(packet_data->get_header_buffer_length()),

		packet_data->get_header_buffer_length());

	if (eap.get_is_valid() == false)

	{

		EAP_TRACE_END(m_am_tools, TRACE_FLAGS_DEFAULT);

		return EAP_STATUS_RETURN(m_am_tools, eap_status_process_illegal_packet_error);

	}

	if (packet_length < eap.get_length())

	{

		EAP_TRACE_END(m_am_tools, TRACE_FLAGS_DEFAULT);

		return EAP_STATUS_RETURN(m_am_tools, eap_status_process_illegal_packet_error);

	}

	if (eap.get_code() == eap_code_none)

	{

		EAP_TRACE_END(m_am_tools, TRACE_FLAGS_DEFAULT);

		return EAP_STATUS_RETURN(m_am_tools, eap_status_process_illegal_packet_error);

	}

	EAP_TRACE_DEBUG(

		m_am_tools, 

		TRACE_FLAGS_DEFAULT, 

		(EAPL("-> EAP_session: %s, code=0x%02x=%s, identifier=0x%02x, ")

		 EAPL("length=0x%04x, type=0x%08x=%s, packet length 0x%04x\n"),

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

		eap.get_code(),

		eap.get_code_string(),

		eap.get_identifier(),

		eap.get_length(),

		convert_eap_type_to_u32_t(eap.get_type()),

		eap.get_type_string(),

		packet_length));

	status = eap.check_header();

	if (status != eap_status_ok)

	{

		EAP_TRACE_END(m_am_tools, TRACE_FLAGS_DEFAULT);

		return EAP_STATUS_RETURN(m_am_tools, status);

	}

	// 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_data() == false)

	{

		EAP_TRACE_END(m_am_tools, TRACE_FLAGS_DEFAULT);

		return EAP_STATUS_RETURN(m_am_tools, eap_status_allocation_error);

	}

	eap_network_id_selector_c selector(

		m_am_tools,

		&send_network_id);

	if (selector.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_TRACE_DATA_DEBUG(

		m_am_tools,

		TRACE_FLAGS_DEFAULT,

		(EAPL("eap_session_core_c::packet_process() EAP-session"),

		 selector.get_data(selector.get_data_length()),

		 selector.get_data_length()));

	eap_core_c *session = m_session_map.get_handler(&selector);

	if (session == 0)

	{

		// Create a new session.

		session = create_new_session(receive_network_id);

	}

	if (session != 0)

	{

		status = session->packet_process(

			receive_network_id,

			&eap,

			packet_length);

		EAP_GENERAL_HEADER_COPY_ERROR_PARAMETERS(packet_data, &eap);

	}

	else

	{

		status = eap_status_illegal_eap_type;

	}

	EAP_TRACE_END(m_am_tools, TRACE_FLAGS_DEFAULT);

	return EAP_STATUS_RETURN(m_am_tools, status);

}

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

//

EAP_FUNC_EXPORT eap_status_e eap_session_core_c::packet_send(

	const eap_am_network_id_c * const send_network_id,

	eap_buf_chain_wr_c * const sent_packet,

	const u32_t header_offset,

	const u32_t data_length,

	const u32_t buffer_length)

{

	EAP_TRACE_BEGIN(m_am_tools, TRACE_FLAGS_DEFAULT);

	EAP_ASSERT(m_am_tools->get_global_mutex()->get_is_reserved() == true);

	eap_header_wr_c eap(

		m_am_tools,

		sent_packet->get_data_offset(header_offset, data_length),

		data_length);

	if (eap.get_is_valid() == false)

	{

		EAP_TRACE_DEBUG(

			m_am_tools, 

			TRACE_FLAGS_DEFAULT, 

			(EAPL("packet_send: packet buffer corrupted.\n")));

		EAP_TRACE_END(m_am_tools, TRACE_FLAGS_DEFAULT);

		return EAP_STATUS_RETURN(m_am_tools, eap_status_header_corrupted);