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


/**
 @file
 @internalComponent
 @released
*/
#ifndef __CRYPTOH4AES__
#define __CRYPTOH4AES__
#include "cryptojobs.h"
#include <drivers/dma.h>

#define FAKE_DMA FakeDma

//#define TDFC_WRAPPER

#ifdef TDFC_WRAPPER
class TDfcWrapper : public TDfc
	{
public:
	TDfcWrapper(const TDfcWrapper &aOrig);
	TDfcWrapper(TDfcFn aFunction, TAny* aPtr, TInt aPriority);
	void Enque();
private:
	TDfcWrapper &operator=(const TDfcWrapper &aRHS);
	void BaseEnque();
    static void DfcWrapperFunc(TAny* aPtr);
	TDfcFn iRealFunction;
	TAny *iRealPtr;
	};
#endif

class DCryptoH4Chan;
class DLddChanAes;
class CryptoH4JobAes : public CryptoJobAes
	{
public:
	CryptoH4JobAes(DLddChanAes &aLddChanAes);
	~CryptoH4JobAes();

	void SetDfcQ(TDfcQue *aDfcQue);

	virtual TUint8 *GetKeyBuffer(); // Max 32 bytes long (256bits)
	virtual TUint8 *GetIVBuffer(); // 16 bytes long
	virtual TUint32 MaxBytes() const;
	virtual TUint8 *GetIOBuffer(); // MaxBytes() long

	virtual void GetToPddBuffer(TUint8 * &aBuf, TUint32 &aBufLen, TBool &aMore);
	virtual void BytesWrittenToPdd(TUint32 aBytes);

	virtual void GetFromPddBuffer(TUint8 * &aBuf, TUint32 &aBufLen, TBool &aMore);
	virtual void BytesReadFromPdd(TUint32 aBytes);

	// Must be called after setting the IV (if used)
	virtual TInt SetDetails(DCryptoJobScheduler *aJobScheduler, 
							MCryptoJobCallbacks *aCallbacks,
							TBool aEncrypt, 
							TInt aKeyLength,
							RCryptoDriver::TChainingMode aMode);
	virtual void NotifyReadRequestLength(TUint32 aReadRequestLength);
	virtual void HwPerfCheck();
	
private:	
	// Implementations of virtual base class methods to support DCryptoJobScheduler
	virtual void DoSlice(TBool aFirstSlice);
	virtual TBool DoSaveState();
	virtual void DoRestoreState();
	virtual void DoReleaseHw();

	void MaybeSetupWriteDmaToHw();
#ifdef FAKE_DMA
	void FakeDma();
#endif

	DLddChanAes &iLddChanAes;
	
	// Details of job
	TBool iEncrypt;
	RCryptoDriver::TChainingMode iMode;

	TUint32 iKeyLengthBytes;
	TUint32 iKey[8]; // 128, 192 or 256 bits
	TUint32 iIV[4]; // 128 bits
	TUint32 iSavedState[4]; // Used for CBC chaining

	TUint32 iReadRequestLength;

	// The AES data buffer
	//	TUint32 iAesBuffer[2*4+4]; // Temporary buffer 8 = 2x16 bytes blocks, +4 space (which never gets used)
	TUint32 iAesBuffer[400*4+4]; // Temporary buffer 8 = 2x16 bytes blocks, +4 space (which never gets used)
	// There are 4 indexes into the above buffer, and after converting
	// all to bytes indexes, and allowing for wrapping the following
	// condition should always hold iSwReadIndex <= iHwWriteIndex <=
	// iHwReadIndex <= iSwWriteByteOffset
	TInt32 iSwWriteByteOffset;
	TInt32 iHwReadIndex;
	TInt32 iHwWriteIndex;
	TInt32 iSwReadByteOffset;

	TBool iHwRunning;
	TInt iDmaToHwPending;
	TInt iDmaFromHwPending;

#ifdef FAKE_DMA
	TInt iFakeDmaToHwQueued;
	TInt iFakeDmaFromHwQueued;
#endif
	
	void SetupHw(TBool aUseSavedState);
	void SetupDma(TUint32 aPtr, TBool aToHw);
	void StopHw();
	static void DmaComplete(DDmaRequest::TResult aResult, TAny *aPtr);

    static void DmaToHwCompleteDfc(TAny* aPtr);
	void DoDmaToHwCompleteDfc();
#ifdef TDFC_WRAPPER
	TDfcWrapper iDmaToHwCompleteDfc;
#else
    TDfc iDmaToHwCompleteDfc;	
#endif

	void CheckIndexes() const;
	void CompareIndexes(const char *aName, TInt32 aP1, TInt32 aP2) const;

    static void DmaFromHwCompleteDfc(TAny* aPtr);
	void DoDmaFromHwCompleteDfc();
#ifdef TDFC_WRAPPER
    TDfcWrapper iDmaFromHwCompleteDfc;	
#else
    TDfc iDmaFromHwCompleteDfc;	
#endif
	};
#endif