// Copyright (c) 2006-2009 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:

//

/**

 @file

*/

#include "mm_node.h"

#include "mm_nodepeer.h"

#include "mm_activities.h"

#include <elements/mm_context_internal.h>

#include <elements/mm_log.h>

#include <elements/nm_messages_base.h>

#include <elements/nm_messages_child.h>

#ifdef _DEBUG

// Panic category for "absolutely impossible!" vanilla ASSERT()-type panics from this module

// (if it could happen through user error then you should give it an explicit, documented, category + code)

_LIT(KSpecAssert_ElemMeshMachNodC, "ElemMeshMachNodC");

#endif

using namespace Messages;

using namespace MeshMachine;

using namespace NetStateMachine;

//-=========================================================

//

//Panics

//

//-=========================================================

_LIT (KMMNodePanic,"MMNodePanic");

enum

	{

	EPanicPreallocatedSpaceAlreadyTaken = 1,

	EPanicPreallocatedSpaceReturnedOther = 2,

	EPanicInvalidClientId = 3,

	EPanicOriginatorIsNotAPeer = 4,

	EPanicNotSynchronisedActivity = 5,

	};

//-=========================================================

//

//AMMNodeBase

//

//-=========================================================

EXPORT_C AMMNodeBase::AMMNodeBase(const TNodeActivityMap& aActivityMap, const TNodeId& aNodeId)

:	ANodeBase(aNodeId),

	TIfStaticFetcherNearestInHierarchy(this),

	iActivityMap(aActivityMap),

	iPreallocatedSpace(NULL)

	{

	};

EXPORT_C void AMMNodeBase::ReturnInterfacePtrL(AMMNodeBase*& aInterface)

    {

    aInterface = this;

    }

EXPORT_C TUint AMMNodeBase::CountActivities(TUint8 aActivitySigId) const

    {

	TInt i = iActivities.Count();

	TUint count = 0;

	while (i>0)

		{

		CNodeActivityBase* a = iActivities[--i];

		if (aActivitySigId == a->ActivitySigId() && !a->IsIdle())

			{

			count++;

			}

		}

	return count;

    }

EXPORT_C TUint AMMNodeBase::CountAllActivities(TUint8 aActivitySigId) const

    {

	TInt i = iActivities.Count();

	TUint count = 0;

	while (i>0)

		{

		CNodeActivityBase* a = iActivities[--i];

		if (aActivitySigId == a->ActivitySigId())

			{

			count++;

			}

		}

	return count;

    }

EXPORT_C TUint AMMNodeBase::CountAllActivities() const

	{

	return iActivities.Count();

	}

EXPORT_C const RPointerArray<CNodeActivityBase>& AMMNodeBase::Activities() const

	{

	return iActivities;

	}

EXPORT_C void AMMNodeBase::ConstructL(TInt aSize)

    {

    //Preallocate space for preallocated activities

    if (aSize==KUseDefaultPreallocatedSize)

    	{

    	//Nodes will usually use the default parameter (-1) and rely on KDefaultPreallocatedActivityBufferSize.

    	//We do not use KDefaultPreallocatedActivityBufferSize as the default argument to avoid publishing of

    	//freely changed at any time).

    	aSize = KDefaultPreallocatedActivityBufferSize;

    	}

    if (aSize>0)

    	{

    	PreallocateSpaceL(aSize);

    	}

	}

EXPORT_C AMMNodeBase::~AMMNodeBase()

	{

	//We may still have some idle activities in the iActivities array.

	//This may be considered normal since there is no strict requirement

	//for a node to destroy all of them before going into destruction phase.

	//What is crutial however, is that there are no running activities left.

	for (TInt i = iActivities.Count() - 1; i >= 0; i--)

		{

		CNodeActivityBase* activity = iActivities[i];

		if (activity->IsIdle())

			{

			iActivities.Remove(i);

			activity->Destroy();

			}

		else

			{

			MESH_LOG((KMeshMachineSubTag, _L8("ERROR: ~AMMNodeBase(%08x) - Running activity found! (%08x) %s (%d)"),

			    this, activity, activity->ActivityName(), activity->ActivityId()));

			}

		}

#ifdef _DEBUG

	for (TInt i = iActivities.Count() -1; i >= 0; i--)

		{

		CNodeActivityBase* activity = iActivities[i];

		if (!activity->IsIdle())

			{

			User::Invariant(); //Running activities still present! Please fix your node's cleanup!

			}

		}

#endif

    for (TInt i = iActivities.Count() -1; i >= 0; i--)

		{

		CNodeActivityBase* activity = iActivities[i];

		iActivities.Remove(i);

		activity->Destroy();

		}

    iActivities.Close();

    User::Free(iPreallocatedSpace);

    iPreallocatedSpace = NULL;

	}

EXPORT_C void AMMNodeBase::RemoveClient(const TRuntimeCtxId& aClientId)

	{

	TNodeNullContext ctx(*this);

    RemoveClient(aClientId, ctx);

	}

EXPORT_C void AMMNodeBase::RemoveClient(const TRuntimeCtxId& aClientId, TNodeContextBase& aContext)

	{

	TInt foundAt = 0;

	RNodeInterface* client = DoFindClient(aClientId, foundAt);

	__ASSERT_DEBUG(client, User::Panic(KMMNodePanic, EPanicInvalidClientId));

	//here we need to cancel all activities originating aClientId (except the current one)

	__ASSERT_DEBUG(!client->RecipientId().IsNull(), User::Panic(KSpecAssert_ElemMeshMachNodC, 1));

	AbortActivitiesOriginatedBy(aContext, client->RecipientId());

    if (aContext.iSender == aClientId)

        {

        aContext.iPeer = NULL;

        }	

	ANodeBase::RemoveClient(foundAt);

	}

EXPORT_C CNodeActivityBase* AMMNodeBase::FindActivityById(TUint aActivityId) const

	{

	CNodeActivityBase* a = NULL;

	for (TInt i = iActivities.Count() - 1; i>=0 && a==NULL; i--)

		{

		CNodeActivityBase* comp = iActivities[i];

		if (!comp->IsIdle()	&& comp->ActivityId() == aActivityId)

			{

			a = comp;

			}

		}

	return a;

	}

EXPORT_C CNodeActivityBase* AMMNodeBase::FindAddressedActivity(const TNodeContextBase& aContext) const

	{

	const TNodeCtxId* recipient = address_cast<const TNodeCtxId>(&aContext.iRecipient);

	if (recipient==NULL)

		{

		return NULL;

		}

	CNodeActivityBase* a = NULL;

	for (TInt i = iActivities.Count() - 1; i>=0 && a==NULL; i--)

		{

		CNodeActivityBase* act = iActivities[i];

		const TNodeId& postedTo = act->PostedToNodeId();

		if (!act->IsIdle()

			&& (postedTo.IsNull() || aContext.iSender == postedTo)

				&& (recipient->NodeCtx() == act->ActivityId()))

			{

			a = act;

			}

		}

	return a;

	}

EXPORT_C void AMMNodeBase::Received(const TNodeSignal::TMessageId aNoPeerMsgIds[], TNodeContextBase& aContext)

    {

	aContext.iReturn = KErrNone; //Clear the return value.

    CNodeActivityBase* sendersActivity = NULL;

    if (!Accept(aContext, sendersActivity) && aContext.iReturn == KErrNone )

        {//start new ones? but only if we know who sent a message or if it's a noPeer message

        //or if it's us sending message to ourselves

	    //is it a message that must have a peer

	    TBool noPeer = EFalse;

	    for (const TNodeSignal::TMessageId* p = &aNoPeerMsgIds[0] ; !p->IsNull() ; ++p)

	    	{

	    	noPeer = (*p == aContext.iMessage.MessageId());

	    	if (noPeer)

	    		{

	    		break;

	    		}

	    	}

        if (noPeer || aContext.IsPeer() || aContext.IsSelf())

            {

			//At this stage we are about to add the peer to the activiy originators.

			//For all "peer requests" we must ensure the peer is in the client's list.

			//If this assertion fires, your environment is critically corrupted.

			//You must repair it by reconsidering your protocols.

			__ASSERT_ALWAYS(noPeer || aContext.iPeer!=NULL, User::Panic(KMMNodePanic, EPanicOriginatorIsNotAPeer));

           	TRAP(aContext.iReturn, StartNewActivitiesL(aContext, sendersActivity));

            }

        else

            {

            //unknown sender => ignore the message

			NM_LOG_ADDRESS_EXT(KMeshMachineSubTag, aContext.iSender, _L8("AMMNodeBase:\tReceived() - Unknown sender. Message must have a peer => ignoring message"));

            }

        }

    else

        {

        __ASSERT_DEBUG(aContext.iMessage.MessageId().IsNull() || aContext.iReturn != KErrNone, User::Panic(KSpecAssert_ElemMeshMachNodC, 2)); //message must be processed

        }

	}

EXPORT_C void AMMNodeBase::HandlePassThroughMessage(TNodeContextBase& aContext)

    {

	if (aContext.iMessage.IsMessage<TEBase::TCancel>())

	    {

	    ASSERT(aContext.iNodeActivity == NULL); //by definition, since nobody accepted this;

	    TInt originatorIndex = KErrNotFound;

		//Please note:

		//The same originator must not attempt to start more than one activities on the same node

		//(so there is only ever one activity on the node with the originator == that this TCancel is addressed to)!

		//In general one originator (identified by a unique channel id) must never originate more

		//than one activities at all.

	    for (TInt i = 0; i < iActivities.Count(); i++)

	    	{

	    	if ((originatorIndex = iActivities[i]->FindOriginator(aContext.iSender)) != KErrNotFound)

	    		{

	    		aContext.iNodeActivity = iActivities[i];

	    		break;

	    		}

	    	}

	    if (aContext.iNodeActivity)

    	    {

    	    ASSERT(originatorIndex != KErrNotFound);

     	    if(aContext.iNodeActivity->iOriginators.Count() == 1) // only if this is the final originator

				{

				aContext.iNodeActivity->Cancel(aContext);

				}

			else

				{

				// just tell this originator that we've stopped with respect to it.. otherwise keep running

				TEBase::TError err(TEBase::TCancel::Id(), KErrCancel);

				aContext.PostToSender(err);

				aContext.iNodeActivity->RemoveOriginator(originatorIndex);

				}

    		}

	    }    

	else if (aContext.iReturn==KErrNone && !aContext.iMessage.MessageId().IsNull())

	    { //this is the pass-through message, forward to .....

        //Find the activity being addressed, if any

		aContext.iNodeActivity = FindAddressedActivity(aContext);

		if (aContext.iNodeActivity)

			{ //Peer message, forward to originators

			//aContext.iSender = Id();

			NM_LOG_MESSAGE_EXT(KMeshMachineSubTag, aContext.iMessage, _L8("AMMNodeBase::HandlePassThroughMessage - message not served by the node but matching activity found=>SEND TO ACTIVITY'S ORIGINATORS"));

			aContext.iNodeActivity->PostToOriginators(aContext.iMessage);

			if (aContext.iMessage.IsMessage<TEBase::TError>())

			    {

			    //An activity received an error=>remove it from active array

			    //This scenario will usually happen when there is no error activity on the node,

			    //or when the error activity decides not to handle this particular error.

			    aContext.iNodeActivity->SetIdle();

			    }

			aContext.iMessage.ClearMessageId();

			}

		else

			{ //A stray message

			//Please note that we should not implement forwarding for non-response messages here.

			//Any non-response message that needs forwarding should use TMessageDispatcher or other

			//supported mechanisms.

			NM_LOG_MESSAGE_EXT(KMeshMachineSubTag, aContext.iMessage, _L8("ERROR: AMMNodeBase::HandlePassThroughMessage - A stray message received!"));

			}

        }

    }

EXPORT_C void AMMNodeBase::HandleMessageReturnValue(TNodeContextBase& aContext)

    {

	if (aContext.iReturn != KErrNone)

		{

		//keep the original activity in case no aCurrentActivity

		if(aContext.iNodeActivity && aContext.iNodeActivity->iOriginators.Count() > 0)

			{

			aContext.iNodeActivity->SetError(aContext.iReturn);

			// Mark the activity for destruction

			// When the activity is destroyed the Originators will be errored

			aContext.iNodeActivity->SetIdle();

			}

		// Check to see if the node was unable to handle an error (e.g., an error recovery activity failed to start

		// because there was insufficient memory).  If so, we manually error the activity and set it to idle causing

		// its originators to be errored.  This is the most we can do if an error activity cannot be run.

		else if(aContext.iMessage.IsMessage<TEBase::TError>())

			{

			CNodeActivityBase* activity = FindAddressedActivity(aContext);

			if(activity)

				{

				activity->SetError(static_cast<TEBase::TError&>(aContext.iMessage).iValue);

				// Mark the activity for destruction

				// When the activity is destroyed the Originators will be errored

				activity->SetIdle();

				}

			}

		else

			{ //No activity == single triple activity or an error from the first transition, so use the sender's message id.

			TEBase::TError err(aContext.iMessage.MessageId(), aContext.iReturn);

			aContext.PostToSender(err);

			}

		aContext.iReturn = KErrNone;

		aContext.iMessage.ClearMessageId();

		}

    }

//N.B. This function may delete the object pointed to by aCurrentActivity.

//So don't use it after this function is called.

//Also, it may delete the node itself, so don't use it after this function is called.

EXPORT_C void AMMNodeBase::PostReceived(TNodeContextBase& aContext)

	{

    HandleMessageReturnValue(aContext);

	HandlePassThroughMessage(aContext);

    SignalActivities();

    }

void AMMNodeBase::SignalActivities()

	{

	TBool awoke = ETrue;

	TInt c = iActivities.Count();

	TNodeNullContext context(*this);

	//Clean up..

	while (awoke)

		{

		awoke = EFalse;

		for (TInt i = iActivities.Count() - 1 ; i>=0 ; --i)

			{

			//Signal to waiting activities if:

			//1) an event was received (& the state has potentialy changed as a result)

			//- remember that an activity does not have to get idle as a result of processing an event which changes the state!!!!

			//2) another waiting activity awoke == reacted to the signal (& the state has potentialy changed as a result)

			//If any signalled activity reacted, the state could change and all other activities need to be signalled again.

			context.iNodeActivity = iActivities[i];

			awoke |= context.iNodeActivity->Signal(context);

			}

		}

	for (TInt i = iActivities.Count() - 1 ; i>=0 ; --i)

		{

		context.iNodeActivity = iActivities[i];

		if(context.iNodeActivity->IsIdle())

			{

			iActivities.Remove(i);

			context.iNodeActivity->Destroy();

			context.iNodeActivity = NULL;

			}

		}

	}

EXPORT_C void AMMNodeBase::AbortActivitiesOriginatedBy(TNodeContextBase& aContext, const TNodeId& aCommsId, TBool aIsNodeBeingDestroyed)

    {

    CNodeActivityBase* caller = aContext.iNodeActivity;

    TBool abortAll = aCommsId.IsNull();

    for (TInt i = iActivities.Count() - 1; i>=0; i--)

        {

        aContext.iNodeActivity = iActivities[i];

        if (!abortAll && aContext.iNodeActivity->PostedToNodeId() == aCommsId)

            {//clear postedto if a leaver has been set as a postedto at any of the running activities.

             //No other messages will ever come from the leaver and it is not gonna be safe to forward TCancels

             //to the leaver, so at least postedto must be cleared to avoid the crash. It could be speculated that

             //if the postedto is still set, then either the postedto node failed to respond or the local activity failed to

             //clear postedto when it had responded. Worth putting a speculative ASSERT here to catch misdeeds. 

             //clearing postedto shouldn't be done in here (AbortActivitiesOriginatedBy), but I (RZ) have expressed my disrespect to the

             //this method before and the suggestion that it should go (replaced by a CancelActivitiesOriginatedBy).

             //So instead of introducing another method that  loops through activities i decided to piggyback the function in an 

             //existing method. Note that the final logic should be based on RNodeInterfaces and not TNodeIds.

            aContext.iNodeActivity->ClearPostedTo();

            }

		// We dont want to abort already idle activities or they may error.		

		if(aContext.iNodeActivity->IsIdle())

			{

			continue;

			}

        if (caller != aContext.iNodeActivity)

            {

        	if (abortAll)

        	    {

        	    //Abort the whole activity (Cancel it & error all originators)

        	    aContext.iNodeActivity->Abort(aContext,aIsNodeBeingDestroyed);

        	    }

        	else

      	        {

      	        //Abort for one originator only (Cancel the activity if last originator & error just this one originator)

      	        TInt idx = aContext.iNodeActivity->FindOriginator(aCommsId);

      	        if (KErrNotFound!=idx)

      	        	{

                    TBool canSend = ETrue;

					if(aContext.iNodeActivity->iOriginators.Count() == 1) // only if this is the final originator

						{

						aContext.iNodeActivity->SetError(KErrAbort);

           	        	aContext.iNodeActivity->Cancel(aContext);

           	        	//This is a workaround for CCommsBinderRequest. The proper fix is to abolish the concept of aborting activities.

                        //Aborting activities is a bad idea as an aborted activity isn't given a chance to perform graceful cleanup.

           	        	//Today activities get aborted because their orinators urgently leave. I.e.: they are trully leaving now! Last orders!

           	        	//It is then incorrect to leave the activity d'tor to finish the wrap up - because the node will be gone by then.

           	        	//So whether and when to send an error must be decided here, by this generic code that has no clue on the subtleties

           	        	//of individual activities. If there is no abort - there is urgent leavers. They send TLeaveRequest and they politely

           	        	//wait for the completion and all this code is unnecessary.

           	        	canSend = (aContext.iNodeActivity->Error() != KErrNone);

						}

                    //In the "quiet mode", when the hosting node is being destroyed, we can not afford sending

                    //an error to the node as it would hit void.

                    TNodePeerId& originator = aContext.iNodeActivity->iOriginators[idx];

                    canSend &= !((aIsNodeBeingDestroyed && originator == aContext.NodeId())

                        || aContext.iMessage.IsMessage<TEChild::TLeft>()); 

                    if (canSend)

                        {

                        aContext.iNodeActivity->PostToOriginator(originator, TEBase::TError(aContext.iMessage.MessageId(), KErrAbort).CRef());

                        }

                    aContext.iNodeActivity->RemoveOriginator(idx);

      	        	}

      	        }

            }

        }

    aContext.iReturn = KErrNone;

    aContext.iNodeActivity = caller;

    }

//This fn finds the (first) activity that has an originator matching that passed in

EXPORT_C CNodeActivityBase* AMMNodeBase::FindActivityOriginatedBy(const TRuntimeCtxId& aPeerId)

    {

    for (TInt i = 0; i < iActivities.Count(); i++)

		{

		CNodeActivityBase* a = iActivities[i];

		if (!a->IsIdle()

			&& a->FindOriginator(aPeerId) != KErrNotFound)

		    {

			return a;

		    }

		}

	return NULL;

    }

TBool AMMNodeBase::Accept(TNodeContextBase& aContext, CNodeActivityBase*& aFoundActivity)

    {

//    NM_LOG_MESSAGE_EXT(KMeshMachineSubTag, aContext.iMessage, _L8("AMMNodeBase:\tAccept"));

    //process the existing ones

    for (TInt i = 0; i < iActivities.Count(); i++)

		{

		CNodeActivityBase* a = aContext.iNodeActivity = iActivities[i];