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

// tcertuils.h

// 

//

/**

 @file 

 @internalTechnology

*/

#ifndef __TCERTUTILS_H__

#define __TCERTUTILS_H__

#include <e32base.h>

#include <x509cert.h>

#include <unifiedcertstore.h>

#include <pkixcertchain.h>

#include "t_output.h"

/**

CCertUtils provides various functions often used in tests.

This class has changed because the functions are asynchronous now. 

The function are no longer static as we need an instance to serve as

an active object.

*/

class CCertUtils : public CActive

	{

public:

	/**

	 * This enumeration enumerates all the states used for the

	 * RunL function.

	 */

	enum TState

		{

		EAddCACerts,

		EAddCert,

		EAddCACertGetCAEntry,

		EAddCACertSetApplications,

		EAddCACertSetTrust,

		EAddCACertFinished,

		ERemoveCertsGetCACerts,

		ERemoveCertsCACertsRetrieved,

		ERemoveCertsRemoveCACerts,

		ERemoveCertsGetUserCerts,

		ERemoveCertsUserCertsRetrieved,

		ERemoveCertsRemoveUserCerts,

		ERemoveCertsFinished

		};	

public:

	IMPORT_C static CCertUtils* NewL(RFs& aFs);

	IMPORT_C static CCertUtils* NewLC(RFs& aFs);

	/**

	The destructor destroys all the members of CCertUtils but only destroys

	iStoreManager if the CCertStoreManager instance was created by this CCertUtils

	instance. It doesn't destroy it if it was passed as an argument in the 

	constructor.

	*/

	IMPORT_C virtual ~CCertUtils();

	/**

	Certificate handling functions

	*/

	IMPORT_C void AddCertL(const TDesC& aLabel, 

							TCertificateFormat aFormat,

							TCertificateOwnerType aCertificateOwnerType, 

							TInt aTrustedUsage, 

							const TDesC& aCertificatePath, 

							const TDesC& aCertificateFileName, 

							TRequestStatus& aStatus);

	IMPORT_C void AddCert(const TDesC& aLabel, 

							TCertificateFormat aFormat,

							TCertificateOwnerType aCertificateOwnerType, 

							TInt aTrustedUsage, 

							const TDesC& aCertificatePath, 

							const TDesC& aCertificateFileName, 

							CUnifiedCertStore& aUnifiedCertStore, 

							TRequestStatus& aStatus);

	IMPORT_C void AddCACertsL(const CDesCArray& aRoots, 

							const CDesCArray& aLabels,

							TCertificateFormat aFormat, 

							TInt aTrustedUsage, 

							const TDesC& aPath, 

							TRequestStatus& aStatus);

	IMPORT_C void AddCACertsL(const CDesCArray& aRoots, 

							const CDesCArray& aLabels,

							TCertificateFormat aFormat, 

							TInt aTrustedUsage, 

							const TDesC& aPath, 

							CUnifiedCertStore& aUnifiedCertStore, 

							TRequestStatus& aStatus);

	IMPORT_C void RemoveCertsL(TRequestStatus& aStatus);

	IMPORT_C void RemoveCertsL(CUnifiedCertStore& aUnifiedCertStore, 

							TRequestStatus& aStatus);

	IMPORT_C static CCertificate* CertFromFileLC(const TDesC& aFilename, 

							const TDesC& aPathname, 

							RFs& aFs, 

							TCertificateFormat aFormat);

	IMPORT_C static CCertificate* CertFromFileL(const TDesC& aFilename,

							const TDesC& aPathname, 

							RFs& aFs, 

							TCertificateFormat aFormat);

	IMPORT_C static void WriteError(TValidationError aError, Output& aOut);

	IMPORT_C static TPtrC MapError(TValidationError aError);

	IMPORT_C static TValidationError MapError(const TDesC& aError);

	IMPORT_C static TPtrC ParseElement(const TDesC& aBuf, const TDesC& aStart, const TDesC& aEnd, TInt& aPos, TInt& aError);

	IMPORT_C HBufC* DiagnosticLC() const;

	IMPORT_C void AddApplicationL(const TDesC& aName, TUid aUid) const;

	IMPORT_C void RemoveApplicationL(TUid aUid) const;

private:

	/**

	This constructor constructs a CCertUtils instance without initializing

	iStoreManager. A store manager will be created if one is needed and

	will be destroyed by the destructor.

	@param aFs A file server session. The file server session must have opened.

	*/

	CCertUtils(RFs& aFs);

	void ConstructL();

	//Virtual from CActive

	void RunL();

	void DoCancel();

    TInt RunError(TInt aError);

private:

	// The following functions handle the different RunL states

	void HandleEAddCACertsL();

	void HandleEAddCACertL();

	void HandleEAddCACertGetCAEntry();

	void HandleEAddCACertSetApplicationsL();

	void HandleEAddCACertSetTrust();

	void HandleEAddCACertFinishedL();

	void HandleERemoveCertsGetCACerts();

	void HandleERemoveCertsCACertsRetrieved();

	void HandleERemoveCertsRemoveCACerts();

	void HandleERemoveCertsGetUserCerts();

	void HandleERemoveCertsUserCertsRetrieved();

	void HandleERemoveCertsRemoveUserCerts();

	void HandleERemoveCertsFinished();

private:

	/**

	File Server Sessioin

	*/

	RFs& iFs;

	/**

	The state used to know what RunL should do

	*/

	TState iState;

	/**

	The store managers will be used for certificate store operations,

	*/

	CUnifiedCertStore* iUnifiedCertStore;

	CUnifiedCertStore* iCreatedUnifiedCertStore;

	/**

	These members are used to store the arguments of the functions

	because most of the things are handled in a RunL.

	*/

	const CDesCArray* iRoots;					// Used by AddCertsL

	const CDesCArray* iLabels;					// Used by AddCertsL

	TCertificateFormat iFormat;

	TInt iTrustedUsage;							// Used by AddCertsL, AddCert

	RArray<TUid> iTrusters;						// Used by AddCertsL, AddCert

	const TDesC* iPath;							// Used by AddCertsL, AddCert

	TRequestStatus* iOriginalRequestStatus;		// Used by AddCertsL, AddCert

	TInt iIndex;								// Used by AddCertsL to know 

												// which certificate to add next

												// Used by RemoveCerts

	CCertUtils* iSecondCertUtils;				// Used by AddCertsL to add each individual

												// certificate

	const TDesC* iLabel;						// Used by AddCert

	TCertificateOwnerType iCertificateOwnerType;

	const TDesC* iCertificateFileName;			// Used by AddCert 

	CCertificate* iCertificate;					// Used by AddCACert

	MCTWritableCertStore *iStore;				// Used by AddCACert

	TPtrC8 iEncoding;							// Used by AddCACert

	HBufC* iCertData;

	HBufC* iLabelData;

	/**

	This is a filter used by the GetCACerts function. We use the filter constructed

	by the constructor with no argument, the filter doesn't filter anything.

	*/

	CCertAttributeFilter* iCAFilter;				// Used by RemoveCerts

	RMPointerArray<CCTCertInfo> iCACertStoreEntries;	// Used by RemoveCerts

	/**

	This is a filter used by the GetUserCerts function. We use the filter constructed

	by the constructot with no argument, the filter doesn't filter anything.

	*/

	CCertAttributeFilter* iUserFilter;			// Used by RemoveCerts

	RMPointerArray<CCTCertInfo> iUserCertStoreEntries; // Used by RemoveCerts

public:

	/**

	A diagnostic variable for when an error occurs. It stores the state at which

	the error occurred.

	*/

	TState iDiagnosticState;

	/** 

	A diagnostic variable for when an error occurs. It is an info message set

	to whatever is relevant for the error.

	*/

	TBuf<400> iDiagnosticMessage;

	};

#endif