// Copyright (c) 2010 Nokia Corporation and/or its subsidiary(-ies).
// All rights reserved.
// This component and the accompanying materials are made available
// under the terms of "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:
//

#include <bluetooth/logger.h>

#include "encryptionkeyrefreshhelper.h"
#include "hcifacade.h"
#include "linkmgr.h"

#ifdef __FLOG_ACTIVE
_LIT8(KLogComponent, LOG_COMPONENT_LINKMGR);
#endif

CEncryptionKeyRefresher::CEncryptionKeyRefresher(CPhysicalLinksManager& aLinkMgr, CPhysicalLink& aLink)
	: CPhysicalLinkHelper(aLinkMgr, aLink)
	{
	LOG_FUNC
	}

CEncryptionKeyRefresher* CEncryptionKeyRefresher::NewL(CPhysicalLinksManager& aLinkMgr, CPhysicalLink& aLink)
	{
	LOG_STATIC_FUNC
	CEncryptionKeyRefresher* self = new(ELeave) CEncryptionKeyRefresher(aLinkMgr, aLink);
	CleanupStack::PushL(self);
	self->ConstructL();
	CleanupStack::Pop(self);
	return self;
	}
	
void CEncryptionKeyRefresher::ConstructL()
	{
	LOG_FUNC
	
	BaseConstructL();
	
	iStateFactory = CEncryptionKeyRefresherStateFactory::NewL();
	iState = &(iStateFactory->GetState(CEncryptionKeyRefresherStateFactory::EIdle));
	}
	
CEncryptionKeyRefresher::~CEncryptionKeyRefresher()
	{
	LOG_FUNC
	}

void CEncryptionKeyRefresher::EncryptionKeyRefreshComplete(TInt aError)
	{
	LOG_FUNC
	iState->EncryptionKeyRefreshComplete(*this, aError);
	}

void CEncryptionKeyRefresher::TimerExpired()
	{
	LOG_FUNC
	iState->TimerExpired(*this);
	}

void CEncryptionKeyRefresher::HandleError( TInt aError)
	{
	LOG_FUNC
	iState->Error(*this, aError);
	}

void CEncryptionKeyRefresher::StartHelper()
	{
	LOG_FUNC
	iState->Start(*this);
	}
	
void CEncryptionKeyRefresher::EventReceived(TBTBasebandEventNotification& aEvent)
	{
	LOG_FUNC
	iState->EventReceived(*this, aEvent);
	}
		

//----------------------------------------------------------------------------------
// STATE FACTORY
//----------------------------------------------------------------------------------

CEncryptionKeyRefresherStateFactory* CEncryptionKeyRefresherStateFactory::NewL()
	{
	LOG_STATIC_FUNC
	CEncryptionKeyRefresherStateFactory* ret=new (ELeave) CEncryptionKeyRefresherStateFactory();
	CleanupStack::PushL(ret);
	ret->ConstructL();
	CleanupStack::Pop(ret);
	return ret;
	}

void CEncryptionKeyRefresherStateFactory::ConstructL()
	{
	LOG_FUNC	
	iStates[EIdle]					=new (ELeave) TEkrStateIdle(*this);
	iStates[EDisablingLPM]			=new (ELeave) TEkrStateDisablingLPM(*this);
	iStates[ERefreshingKey]			=new (ELeave) TEkrStateRefreshingKey(*this);
	}

CEncryptionKeyRefresherStateFactory::CEncryptionKeyRefresherStateFactory()
	{
	LOG_FUNC
	iStates.DeleteAll();
	}

const TEncryptionKeyRefresherState& CEncryptionKeyRefresherStateFactory::GetState(CEncryptionKeyRefresherStateFactory::TEncryptionKeyRefresherStates aState) const
	{
	LOG_FUNC
	__ASSERT_DEBUG(iStates[aState],  Panic(EEncryptionKeyRefresherInvalidState));
	return *iStates[aState];
	}

TInt CEncryptionKeyRefresherStateFactory::StateIndex(const TEncryptionKeyRefresherState* aState) const
	{
	LOG_FUNC
	TInt state;
	for (state = 0; state < EEncryptionKeyRefresherMaxState; state++)
		{
		if (iStates[state] == aState)
			{
			return state;
			}
		}
	
	return KUnknownState;
	}


//----------------------------------------------------------------------------------
// STATES
//----------------------------------------------------------------------------------

TEncryptionKeyRefresherState::TEncryptionKeyRefresherState(CEncryptionKeyRefresherStateFactory& aFactory)
: iFactory(aFactory)
	{
	LOG_FUNC
	}

void TEncryptionKeyRefresherState::PanicInState(TLinkPanic aPanic) const
	{
	LOG_FUNC
	Panic(aPanic, iFactory.StateIndex(this));
	}

void TEncryptionKeyRefresherState::ChangeState(CEncryptionKeyRefresher& aContext, CEncryptionKeyRefresherStateFactory::TEncryptionKeyRefresherStates aState) const
	{
	LOG_FUNC
	
	aContext.iState->Exit(aContext);

#ifdef __FLOG_ACTIVE
	const TEncryptionKeyRefresherState* state=&iFactory.GetState(aState);
	LOG2(_L("EncryptionKeyRefresher: State %S -> %S"), &aContext.iState->iName, &state->iName);
#endif //__FLOG_ACTIVE
	aContext.iState=&iFactory.GetState(aState);

	aContext.iState->Enter(aContext);
	}

void TEncryptionKeyRefresherState::Enter(CEncryptionKeyRefresher& /*aContext*/) const
	{
	LOG_FUNC
	// do nothing
	}

void TEncryptionKeyRefresherState::Exit(CEncryptionKeyRefresher& /*aContext*/) const
	{
	LOG_FUNC
	// do nothing
	}

void TEncryptionKeyRefresherState::Start(CEncryptionKeyRefresher& /*aContext*/) const
	{
	LOG_FUNC
	PanicInState(EEncryptionKeyRefresherStateMachineInvalidEvent);
	}

void TEncryptionKeyRefresherState::Error(CEncryptionKeyRefresher& aContext, TInt /*aErr*/) const
	{
	LOG_FUNC
	aContext.CancelNotify();
	aContext.RemoveTimer();
	ChangeState(aContext, CEncryptionKeyRefresherStateFactory::EIdle);
	}

void TEncryptionKeyRefresherState::EventReceived(CEncryptionKeyRefresher& /*aContext*/, TBTBasebandEventNotification& /*aEvent*/) const
	{
	LOG_FUNC
	// do nothing
	}

void TEncryptionKeyRefresherState::TimerExpired(CEncryptionKeyRefresher& aContext) const
	{
	LOG_FUNC
	aContext.CancelNotify();
	ChangeState(aContext, CEncryptionKeyRefresherStateFactory::EIdle);
	}

void TEncryptionKeyRefresherState::EncryptionKeyRefreshComplete(CEncryptionKeyRefresher& /*aContext*/, TInt /*aError*/) const
	{
	LOG_FUNC
	// do nothing
	}

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

TEkrStateIdle::TEkrStateIdle(CEncryptionKeyRefresherStateFactory& aFactory)
: TEncryptionKeyRefresherState(aFactory)
	{
	LOG_FUNC
	STATENAME("TEkrStateIdle");
	}

void TEkrStateIdle::Start(CEncryptionKeyRefresher& aContext) const
	{
	LOG_FUNC
	aContext.QueueTimer(KTimeoutEncryptionKeyRefresh);	// watchdog timer
	ChangeState(aContext, CEncryptionKeyRefresherStateFactory::EDisablingLPM);
	}	

void TEkrStateIdle::Enter(CEncryptionKeyRefresher& aContext) const
	{
	LOG_FUNC
	// async call to delete the helper
	aContext.iLink.AsyncDeleteKeyRefresher();
	}	

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

TEkrStateDisablingLPM::TEkrStateDisablingLPM(CEncryptionKeyRefresherStateFactory& aFactory)
: TEncryptionKeyRefresherState(aFactory)
	{
	LOG_FUNC
	STATENAME("TEkrStateDisablingLPM");
	}

void TEkrStateDisablingLPM::Enter(CEncryptionKeyRefresher& aContext) const
	{
	LOG_FUNC
	
	if (aContext.iLink.LinkMode() == EActiveMode)
		{
		// Skip straight on to refresh the key
		ChangeState(aContext, CEncryptionKeyRefresherStateFactory::ERefreshingKey);
		}
	else
		{
		aContext.NotifyBasebandEvent(ENotifyActiveMode);
		}
	
	// DisableLPM even if link is active to prevent possible LPM requests during key refresh
	aContext.DisableLPM();
	}

void TEkrStateDisablingLPM::EventReceived(CEncryptionKeyRefresher& aContext, TBTBasebandEventNotification& aEvent) const
	{
	LOG_FUNC
	if (aEvent.EventType()==ENotifyActiveMode && aEvent.ErrorCode()==KErrNone)
		{
		ChangeState(aContext, CEncryptionKeyRefresherStateFactory::ERefreshingKey);
		}
	else 
		{
		LOG(_L("CEncryptionKeyRefresher failed in DisableLPM"));
		// If we can't put ourselves into active mode then don't bother with 
		// refresh, it'll probably cause more problems than it would solve
		// we can quit SM, don't need to rewind
		ChangeState(aContext, CEncryptionKeyRefresherStateFactory::EIdle);
		}
	}

//----------------------------------------------------------------------------------
TEkrStateRefreshingKey::TEkrStateRefreshingKey(CEncryptionKeyRefresherStateFactory& aFactory)
: TEncryptionKeyRefresherState(aFactory)
	{
	LOG_FUNC
	STATENAME("TEkrStateRefreshingKey");
	}

void TEkrStateRefreshingKey::Enter(CEncryptionKeyRefresher& aContext) const
	{
	LOG_FUNC
	
	TRAPD(err, aContext.iLinkMgr.HCIFacade().RefreshEncryptionKeyL(aContext.iLink.Handle()));
	
	if(err)
		{
		ChangeState(aContext, CEncryptionKeyRefresherStateFactory::EIdle);
		}
	}

void TEkrStateRefreshingKey::EncryptionKeyRefreshComplete(CEncryptionKeyRefresher& aContext, TInt __DEBUG_ONLY(aError)) const
	{
	LOG_FUNC
	LOG1(_L("EncryptionKeyRefresh completed with error %d"), aError);

	// Don't really care what the error is, this is just a best effort service
	// anyway
	ChangeState(aContext, CEncryptionKeyRefresherStateFactory::EIdle);
	}