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


#ifndef	__RIJNDAELIMPL_H__
#define	__RIJNDAELIMPL_H__

/**
@file 
@internalComponent
@released
*/

#include <e32base.h>
#include <e32cmn.h>
#include <cryptospi/keys.h>
#include "symmetriccipherimpl.h"

/**
Plug-in class for AES (Rijndael) block cipher
*/
namespace SoftwareCrypto
	{
	NONSHARABLE_CLASS(CRijndaelImpl) : public CSymmetricBlockCipherImpl
		{
	public:
	
		/**
		Creates an instance of an AES (Rijndael) symmetric cipher plug-in.
		@param aKey The key
		@param aCryptoMode Whether to encrypt or decrypt
		@param aOperationMode The block cipher mode ECB, CBC, CTR etc
		@param aPadding The padding scheme to use None, SSLv3, PKCS#7
		@return A pointer to a CRijndaelImpl instance
		*/
		static CRijndaelImpl* NewL(const CryptoSpi::CKey& aKey, 
			TUid aCryptoMode, TUid aOperationMode, TUid aPadding, TUid aImplementationUid);

		/**
		Creates an instance of an AES (Rijndael) symmetric cipher plug-in.
		A pointer to the plug-in instance is placed on the cleanup stack.
		@param aKey The key
		@param aCryptoMode Whether to encrypt or decrypt
		@param aOperationMode The block cipher mode ECB, CBC, CTR etc
		@param aPadding The padding scheme to use None, SSLv3, PKCS#7
		@return A pointer to a CRijndaelImpl instance
		*/
		static CRijndaelImpl* NewLC(const CryptoSpi::CKey& aKey, 
			TUid aCryptoMode,	TUid aOperationMode, TUid aPadding, TUid aImplementationUid);

		// From CSymmetricCipherImpl
		TBool IsValidKeyLength(TInt aKeyBytes) const;
		TUid ImplementationUid() const;
		
		const CExtendedCharacteristics* GetExtendedCharacteristicsL();
		static CExtendedCharacteristics* CreateExtendedCharacteristicsL();

		/// Destructor
		~CRijndaelImpl();

	private:
		/**
		Constructor
		@param aOperationMode The mode of operation e.g. CBC
		@param aCryptoMode Whether to encrypt or decrypt
		@param aPaddingMode The padding mode to use. None, SSL, PKCS#7
		*/
		CRijndaelImpl(TUid aOperationMode, TUid aCryptoMode, TUid aPaddingMode, 
						TUid aImplementationUid);
			
		///	second phase of construction	
		void ConstructL(const CryptoSpi::CKey& aKey);
					
		/**
		Creates the key schedule iK from CBlockTransformationImpl::iKey.
		This should be called from ConstructL and Reset
		*/
		void SetKeySchedule();
		
		/**
		Setup the key schedule for encryption
		@param aKey The input key
		@param aKeySchedule The pointer to the key schedule buffer
		*/
		void SetEncryptKeySchedule(const TDesC8& aKey, TUint32* aKeySchedule);
		
		/**
		Setup the key schedule for decryption
		@param aKey The input key
		@param aKeySchedule The pointer to the key schedule buffer		
		*/
		void SetDecryptKeySchedule(const TDesC8& aKey, TUint32* aKeySchedule);

		// From CSymmetricBlockCipherImpl
		void TransformEncrypt(TUint8* aBuffer, TUint aNumBlocks);
		void TransformDecrypt(TUint8* aBuffer, TUint aNumBlocks);
				
	private:
	
		/** 
		The key-schedule size in WORDS
		
	 	The maximum size is (((KAESMaxBlockSize/4)+6)+1)*4
	 	*/
	 	static const TUint32 KKeyScheduleSize = 60;
	 	
	 	/// the key schedule
		TUint32 iK[KKeyScheduleSize];
		
		TInt iRounds;
		
		TUid iImplementationUid;
		};
	}

#endif //__RIJNDAELIMPL_H__