/*
* 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.
*
*/




#if !defined(_TLS_MESSAGE_H_)
#define _TLS_MESSAGE_H_

#include "eap_tools.h"
#include "eap_array.h"
#include "tls_record_message.h"
#include "abs_tls_message_hash.h"
#include "abs_tls_apply_cipher_spec.h"

/** @file */


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


/// This class defines one TLS-message. One TLS message could include many TLS-records.
/**
 * This class defined one TLS-message.
 * Parse and analyse of TLS-message is asyncronous.
 * m_analyse_index tells the index of message where asyncronous
 * analyse of TLS-message must continue.
 * Analysed messages are skipped during the asyncronous
 * analyse of messages. Asyncronous analyse is needed
 * because of the PKI functions are asyncronous in
 * Symbian.
 */
class EAP_EXPORT tls_message_c
{
private:
	//--------------------------------------------------

	/// This is pointer to the tools class. @see abs_eap_am_tools_c.
	abs_eap_am_tools_c * const m_am_tools;

	/// This is pointer to interface of message HASH creation. See abs_tls_message_hash_c.
	abs_tls_message_hash_c * const m_message_hash;

	/// This is pointer to interface of apply cipher spec. See abs_tls_apply_cipher_spec_c.
	abs_tls_apply_cipher_spec_c * const m_apply_cipher_spec;

	/// This is pointer to interface of change cipher spec. See abs_tls_change_cipher_spec_c.
	abs_tls_change_cipher_spec_c * m_change_cipher_spec;

	/// This buffer includes copy of the whole received TLS-message data.
	eap_variable_data_c m_tls_message_data;

	/// This is EAP-identifier of the EAP-packet that includes TLS-message. This is needed in XP-PEAP.
	u8_t m_received_eap_identifier;

	/// This is the index of message where asyncronous analyse of TLS-message must continue.
	u32_t m_analyse_index;

	/// This array includes one or more records.
	eap_array_c<tls_record_message_c> m_record_messages;

	/// This indicates whether this object is client (true) or server (false). This is mostly for traces.
	const bool m_is_client;

	/// This flag tells the send message includes TLS-Handshake message.
	/// Note the received messages are not marked.
	/// This information is needed in special PEAP version.
	/// All messages including TLS-Handshake message must have PEAP L bit and four octet TLS message length field.
	bool m_includes_tls_handshake_message;

	//--------------------------------------------------
protected:
	//--------------------------------------------------

	//--------------------------------------------------
public:
	//--------------------------------------------------

	/**
	 * The destructor of the tls_message_c class does nothing special.
	 */
	EAP_FUNC_IMPORT virtual ~tls_message_c();

	/**
	 * The constructor of the tls_message_c class simply initializes the attributes.
	 */
	EAP_FUNC_IMPORT tls_message_c(
		abs_eap_am_tools_c * const tools,
		abs_tls_message_hash_c * const message_hash,
		abs_tls_apply_cipher_spec_c * const apply_cipher_spec,
		abs_tls_change_cipher_spec_c * const change_cipher_spec,
		const bool is_client);

	/**
	 * This function resets this object.
	 */
	EAP_FUNC_IMPORT eap_status_e reset();

	/**
	 * This function returns the index of message where analyse must continue.
	 */
	EAP_FUNC_IMPORT u32_t get_analyse_index() const;

	/**
	 * This function saves the index of message where analyse must continue.
	 */
	EAP_FUNC_IMPORT void save_analyse_index(const u32_t analyse_index);

	/**
	 * This function copies the received TLS-message data and EAP-identifier.
	 * EAP-identifier is needed in XP PEAPv0. That stupid version uses
	 * same EAP-identifier with PEAP header and tunneled EAP-header.
	 */
	EAP_FUNC_IMPORT eap_status_e set_tls_message_data(
		eap_variable_data_c * const tls_message_data,
		const u8_t received_eap_identifier);

	/**
	 * This function returns the TLS-message data.
	 */
	EAP_FUNC_IMPORT eap_variable_data_c * get_tls_message_data();

	/**
	 * This function returns the EAP-identifier.
	 */
	EAP_FUNC_IMPORT u8_t get_received_eap_identifier();

	/**
	 * This function adds TLS-record to m_record_messages.
	 * Parameter free_record tells whether record must be freed in destructor.
	 */
	EAP_FUNC_IMPORT eap_status_e add_record_message(
		tls_record_message_c * const record,
		const bool free_record,
		const bool includes_tls_handshake_message);

	/**
	 * This function fragments TLS-protocol messages to one or more TLS-record messages.
	 */
	EAP_FUNC_IMPORT eap_status_e fragment_tls_records(
		tls_record_message_c * const tls_record_message,
		eap_array_c<tls_record_message_c> * const tls_fragments);

	/**
	 * This function adds data of every TLS-record to tls_message_buffer.
	 */
	EAP_FUNC_IMPORT eap_status_e add_message_data(
		eap_variable_data_c * const tls_message_buffer,
		bool * const includes_tls_handshake_message);

	/**
	 * This function returns count of the TLS-records.
	 */
	EAP_FUNC_IMPORT u32_t get_record_message_count() const;

	/**
	 * This function returns pointer to the TLS-record selected by index.
	 */
	EAP_FUNC_IMPORT tls_record_message_c * get_record_message(
		const u32_t index) const;

	/**
	 * This function removes the TLS-record selected by index.
	 */
	EAP_FUNC_IMPORT eap_status_e remove_record_message(
		const u32_t index);

	/**
	 * This function returns pointer to the last TLS-record.
	 */
	EAP_FUNC_IMPORT tls_record_message_c * get_last_record_message() const;

	// 
	//--------------------------------------------------
}; // class tls_message_c


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

#endif //#if !defined(_TLS_MESSAGE_H_)



// End.
