/*

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

*

*/

#include "rtsecmgrscriptmanager.h"

#include "rtsecmgrserverdef.h"

#include "rtsecmgrstore.h"

#include "rtsecmgrpolicy.h"

#include "rtsecmgrpolicymanager.h"

#include "rtsecmgrscriptmanager.h"

#include <e32debug.h>

//#include <centralrepository.h>

_LIT(KDefaultTrustPolicy,"UnTrusted");

CScriptManager* CScriptManager::NewL(CSecMgrStore* aSecMgrDB,CPolicyManager* aPolicyMgr)

{	

	CScriptManager* pThis = CScriptManager::NewLC(aSecMgrDB,aPolicyMgr);

	CleanupStack::Pop(pThis);

	return pThis;

}

CScriptManager* CScriptManager::NewLC(CSecMgrStore* aSecMgrDB,CPolicyManager* aPolicyMgr)

{	

	CScriptManager* pThis = new (ELeave) CScriptManager(aSecMgrDB,aPolicyMgr);

	CleanupStack::PushL(pThis);

	pThis->ConstructL();

	return pThis;

}

CScriptManager::~CScriptManager()

{

	if(iSecMgrDB)

	{

		iSecMgrDB->WriteCounter(iID);	

	}

	iScripts.ResetAndDestroy();

}

void CScriptManager::ConstructL()	

{

	iSecMgrDB->ReadCounter(iID);

}

TExecutableID CScriptManager::RegisterScript(TPolicyID aPolicyID/*,const CTrustInfo& aTrustPolicy*/)

{	

	TInt ret(ErrPolicyIDDoesNotExist);

	CPolicy* policy = iPolicyMgr->Policy(aPolicyID);

	if(policy)

		{

		CProtectionDomain* domain = policy->ProtectionDomain(KDefaultTrustPolicy);

		if(domain)

			{

				TExecutableID execID(GetID());

				CScript* script = CScript::NewL(aPolicyID,execID);

				script->SetPermissionSet(domain->PermSet());

				iScripts.Append(script);

				TRAPD(err,iSecMgrDB->RegisterScriptL(execID,*script));

				if(KErrNone==err)

				{

					return execID;

				}

				else

				{

					return ret;

				}

			}			

		}

	return ret;	

}

TExecutableID CScriptManager::RegisterScript(TPolicyID aPolicyID,const TDesC& aHashValue/*,const CTrustInfo& aTrustPolicy*/)

	{	

		TInt ret(ErrPolicyIDDoesNotExist);

		CPolicy* policy = iPolicyMgr->Policy(aPolicyID);

		if(policy)

			{

			CProtectionDomain* domain = policy->ProtectionDomain(KDefaultTrustPolicy);

			if(domain)

				{

					TExecutableID execID(GetID());

					CScript* script = CScript::NewLC(aPolicyID,execID);

					script->SetHashMarkL(aHashValue);

					script->SetPermissionSet(domain->PermSet());

					iScripts.Append(script);

					TRAPD(err,iSecMgrDB->RegisterScriptL(execID, *script));

					if(KErrNone==err)

					{

					ret=execID;

					}

					CleanupStack::Pop(script);

				}			

			}

		return ret;	

	}

void CScriptManager::UnRegisterScriptL(TExecutableID aExecID,TPolicyID aPolicyID)

{	

	TInt ret;

	CScript *scriptInfo = CScript::NewL(KInvalidPolicyID ,aExecID);

	ret = ScriptInfo(*scriptInfo);

	if( (KErrNone<=ret) && (scriptInfo->PolicyID() != aPolicyID))

	{

		ret = ErrInvalidPolicyID;	

	}

	delete scriptInfo;

	scriptInfo = NULL;

	if(ret == KErrNone)

	{

		for(TInt i=0; i!=iScripts.Count();++i)

		{

			if(iScripts[i]->ScriptID()==aExecID)

			{

				delete iScripts[i];

				iScripts.Remove(i);	//remove from the cache first		

				break;

			}		 

		}		

		iSecMgrDB->UnRegisterScriptL(aExecID);

	}

	else

		User::Leave(ret);

}

TInt CScriptManager::ScriptInfo(CScript& aScriptInfo)

{	

	//Check if script info is already present in the cache (iScriptStore)

	//If yes, 	

	for(TInt i=0; i!=iScripts.Count();++i)

		{

		if(iScripts[i]->ScriptID()==aScriptInfo.ScriptID())

			{

				{

				aScriptInfo = *iScripts[i];				

				return KErrNone;	

				}	

			}

		}

	//Read the script file

	TInt ret = iSecMgrDB->ReadScriptInfo(aScriptInfo.ScriptID(),aScriptInfo);

	if(KErrNone==ret)

	{

		iScripts.AppendL(&aScriptInfo);		

	}

	return ret;

}

TInt CScriptManager::GetScriptFile(TExecutableID aExecID,TDes& aScriptFile)

{

	return iSecMgrDB->GetScriptFile(aExecID,aScriptFile);

}

void CScriptManager::UpdatePermGrantL(TExecutableID aExecID,TPermGrant aPermGrant,TPermGrant aPermDenied)

{	

	for(TInt i(0);i!=iScripts.Count();++i)

	{

	if(iScripts[i]->ScriptID()==aExecID)

		{

			iScripts[i]->SetPermGranted(aPermGrant);

			iScripts[i]->SetPermDenied(aPermDenied);

			iSecMgrDB->WriteScriptInfoL(aExecID,*iScripts[i]);			

			return;

		}	

	}

}