--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/cryptoservices/filebasedcertificateandkeystores/test/certtool/certtool_addapps.cpp	Wed Jul 08 11:25:26 2009 +0100
@@ -0,0 +1,173 @@
+/*
+* Copyright (c) 2004-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 "certtool_commands.h"
+#include "certtool_controller.h"
+#include "keytool_utils.h"
+
+/*static*/ CCertToolAddApps* CCertToolAddApps::NewLC(CCertToolController* aController)
+	{
+	CCertToolAddApps* self = new (ELeave) CCertToolAddApps(aController);
+	CleanupStack::PushL(self);
+	self->ConstructL();
+	return self;
+	}
+	
+/*static*/ CCertToolAddApps* CCertToolAddApps::NewL(CCertToolController* aController)
+	{
+	CCertToolAddApps* self = CCertToolAddApps::NewLC(aController);
+	CleanupStack::Pop(self);
+	return self;
+	}
+
+CCertToolAddApps::CCertToolAddApps(CCertToolController* aController) : CCertToolList(aController)
+	{
+	}
+
+	
+CCertToolAddApps::~CCertToolAddApps()
+	{
+	Cancel();
+	}
+
+void CCertToolAddApps::RunL()
+	{
+	if (iStatus.Int() != KErrNone)
+		{
+		// A problem occured. Handle gracefully.
+		User::Leave(iStatus.Int());
+		}	
+		
+	switch (iState)
+		{	 
+		case EListCerts:
+			{
+			if (iCertInfos.Count()==0)
+				{
+				iController->DisplayLocalisedMsgL(R_CERTTOOL_ERR_EMPTY);
+				CActiveScheduler::Stop();				
+				}
+			else
+				{ 
+				if (iParams->iDefault)    
+					{
+					// Look for a specific certificate					
+					TInt certIndex = -1;
+					for (TInt j = 0; j < iCertInfos.Count(); j++)
+						{
+						if (iCertInfos[j]->Label() == *iParams->iDefault) 
+							{
+							certIndex = j;
+							break;
+							}
+						}	
+						
+					if (certIndex != -1)
+						{
+						iIndex = certIndex;						
+						// Certificate found set app uids!
+						if (iParams->iCertstoreIndex == -1)
+							{
+							iParams->iCertstoreIndex = 0;
+							}
+						if (iParams->iCertstoreIndex >= iCertStore->CertStoreCount())
+							{
+							iController->DisplayLocalisedMsgL(R_CERTTOOL_ERR_NOTEXIST);	
+							User::Leave(KErrArgument);			
+							}						
+						iCertStore->Applications(*iCertInfos[iIndex], iApps, iStatus);
+						iState = ESetApps;
+						SetActive();
+						}
+					else 
+						{
+						User::Leave(KErrNotFound);
+						}			
+					}
+				}
+			}
+			break;
+		case ESetApps:
+			{				
+			if (iParams->iCertstoreIndex == -1)
+				{
+				iParams->iCertstoreIndex = 0;
+				}
+			if (iParams->iCertstoreIndex >= iCertStore->CertStoreCount())
+				{
+				iController->DisplayLocalisedMsgL(R_CERTTOOL_ERR_NOTEXIST);	
+				User::Leave(KErrArgument);			
+				}				
+			// Check for duplicates
+			for (TInt k = 0; k < iParams->iUIDs.Count(); k++)
+				{
+				if (iApps.Find(iParams->iUIDs[k]) != KErrNotFound)
+					{
+					TBuf<128> msg;
+					const TDesC& tmp = KeyToolUtils::Uid2Des(iParams->iUIDs[k]);
+					msg.Format(_L("The certificate is already trusted for application %S."), &tmp);
+					iController->DisplayErrorL(msg, iParams->iPageWise);
+					User::Leave(KErrArgument);
+					}
+				}
+
+			// add the apps to the array
+			for (TInt j = 0; j < iApps.Count(); j++)
+				{
+				iParams->iUIDs.Append(iApps[j]);
+				}
+			iState = EFinished;
+			iCertStore->SetApplicability(*iCertInfos[iIndex], iParams->iUIDs, iStatus);
+			SetActive();
+			}
+			break;
+		case EFinished:
+			{
+			iCertInfos.Close(); 
+			iController->DisplayLocalisedMsgL(R_CERTTOOL_ERR_ADDAPPS_SUCCESS);				
+			CActiveScheduler::Stop();			
+			}
+			break;
+		default:
+			{
+			User::Panic(_L("ADDAPPS action: Illegal state."), 1);
+			}
+		}
+	}
+
+TInt CCertToolAddApps::RunError(TInt aError)
+	{
+	CActiveScheduler::Stop();
+
+	switch (aError)
+		{
+		case KErrNotFound :
+			{
+			TRAP_IGNORE(iController->DisplayLocalisedMsgL(R_CERTTOOL_ERR_NOTFOUND));
+			TRAP_IGNORE(iController->DisplayErrorL(_L("\n"), iParams->iPageWise));			
+			}
+			break;
+		default:
+			{
+			TRAP_IGNORE(iController->DisplayLocalisedMsgL(R_CERTTOOL_ERR_ADDAPPS_FAIL));			
+			TRAP_IGNORE(iController->DisplayErrorL(_L("\n"), iParams->iPageWise));			
+			}
+		}	
+
+	return KErrNone;	
+	}