/*

* Copyright (c) 1998-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 "cryptotestaction.h"

#include <securityerr.h>

#include <cryptostrength.h>

#include "../common/inlines.h"

#include "t_output.h"

CCryptoTestAction::CCryptoTestAction(RFs& aFs, CConsoleBase& aConsole, Output& aOut)

:	CTestAction(aConsole, aOut), iFs(aFs), iEffectiveKeyLen(128)

{}

CCryptoTestAction::~CCryptoTestAction()

{

	delete iBody;

}

void CCryptoTestAction::ConstructL(const TTestActionSpec& aTestActionSpec)

{

	CTestAction::ConstructL(aTestActionSpec);

	iBody = HBufC8::NewL(aTestActionSpec.iActionBody.Length());

	iBody->Des().Copy(aTestActionSpec.iActionBody);	

}

void CCryptoTestAction::DoPerformPrerequisite(TRequestStatus& aStatus)

{

	TRequestStatus* status = &aStatus;

	TRAPD(r, DoPerformPrerequisiteL());

	if ((r != iExpectedWeakResult) && (TCrypto::Strength() == TCrypto::EWeak))

		{

		r = KErrUnknown;

		}

	User::RequestComplete(status, r);

	if (KErrNone==r)

		{

		iActionState = CTestAction::EAction;

		}

	else if (r==KErrKeyNotWeakEnough)

		{

		iResult = ETrue;

		iConsole.Printf(_L("Crypto libraries returned KErrKeyNotWeakEnough!  Passing test automatically.\n\r"));

		iOut.writeString(_L("Crypto libraries returned KErrKeyNotWeakEnough!  Passing test automatically.\r\n"));

		iActionState = CTestAction::EPostrequisite;

		}

	else if (r==KErrWeakKey)

		{

		iResult = ETrue;

		iConsole.Printf(_L("Crypto libraries returned KErrWeakKey!  Passing test automatically.\n\r"));

		iOut.writeString(_L("Crypto libraries returned KErrWeakKey!  Passing test automatically.\r\n"));

		iActionState = CTestAction::EPostrequisite;

		}

	else

		{

		iActionState = CTestAction::EPostrequisite;

		}

}

void CCryptoTestAction::DoPerformPostrequisite(TRequestStatus& aStatus)

{

	TRequestStatus* status = &aStatus;

	TRAPD(r, DoPerformPostrequisiteL());

	delete iKey;

	delete iInput;

	delete iOutput;

	delete iIV;

	delete iEResult;

	delete iDResult;

	iFinished = ETrue;

	User::RequestComplete(status, r);

}

void CCryptoTestAction::DoReportAction(void)

{}

void CCryptoTestAction::DoCheckResult(TInt)

{

	if (iResult==EFalse)

		iConsole.Printf(_L("X"));

	else

		iConsole.Printf(_L("."));

}

void CCryptoTestAction::PerformAction(TRequestStatus& aStatus)

{

	TRequestStatus* status = &aStatus;

	TRAPD(res, DoPerformActionL());

	if (res==KErrNoMemory)

		{

		User::Leave(res);	//	For OOM testing

		}

	iActionState = CTestAction::EPostrequisite;

	User::RequestComplete(status, res);

}

void CCryptoTestAction::DoInputParseL(TDesC8& aScriptBuffer)

{

	TInt err = KErrNone;

	TInt pos = 0;

	TPtrC8 keyTemp = Input::ParseElement(aScriptBuffer, KKeyStart, KKeyEnd, pos, err);

	iKey = HBufC8::NewL(keyTemp.Length());

	*iKey = keyTemp;

	Hex(*iKey);

	pos = 0;

	TPtrC8 inputTemp = Input::ParseElement(aScriptBuffer, KInputStart, KInputEnd, pos, err);

	iInput = HBufC8::NewL(inputTemp.Length());

	*iInput = inputTemp;

	Hex(*iInput);

	pos = 0;

	TPtrC8 ivTemp = Input::ParseElement(aScriptBuffer, KIVStart, KIVEnd, pos, err);

	iIV = HBufC8::NewL(ivTemp.Length());

	*iIV = ivTemp;

	Hex(*iIV);

	pos = 0;

	TPtrC8 outputTemp = Input::ParseElement(aScriptBuffer, KOutputStart, KOutputEnd, pos, err);

	iOutput = HBufC8::NewL(outputTemp.Length());

	*iOutput = outputTemp;

	Hex(*iOutput);

	pos = 0;

	TPtrC8 cipherType = Input::ParseElement(aScriptBuffer, KCipherTypeStart, KCipherTypeEnd, pos, err);

	iExpectedWeakResult = KErrNone;

	if( cipherType == _L8("DESECB") )

	{

		iCipherType = EDESECB;

	}

	else if( cipherType == _L8("DESCBC") )

	{

		iCipherType = EDESCBC;

	}

	else if( cipherType == _L8("3DESECB") )

	{

		iCipherType = E3DESECB;

		iExpectedWeakResult = KErrKeyNotWeakEnough;

	}

	else if( cipherType == _L8("3DESCBC") )

	{

		iCipherType = E3DESCBC;

		iExpectedWeakResult = KErrKeyNotWeakEnough;

	}

	else if( cipherType == _L8("AESECB") )

	{

		iCipherType = EAESECB;

		iExpectedWeakResult = KErrKeyNotWeakEnough;

	}

	else if (cipherType==_L8("AESENC_ECB") )

	{

		iCipherType = EAESMonteCarloEncryptECB;	

		iExpectedWeakResult = KErrKeyNotWeakEnough;

	}

	else if (cipherType==_L8("AESDEC_ECB") )

	{

		iCipherType = EAESMonteCarloDecryptECB;	

		iExpectedWeakResult = KErrKeyNotWeakEnough;

	}

	else if (cipherType==_L8("AESENC_CBC") )

	{

		iCipherType = EAESMonteCarloEncryptCBC;	

		iExpectedWeakResult = KErrKeyNotWeakEnough;

	}

	else if (cipherType==_L8("AESDEC_CBC") )

	{

		iCipherType = EAESMonteCarloDecryptCBC;		

		iExpectedWeakResult = KErrKeyNotWeakEnough;

	}

	else if( cipherType == _L8("RC2ECB") )

	{

		iCipherType = ERC2ECB;

		// weak enough if either aKey or aEffectiveKeyLen is weak

		TInt minKeySize = Min(iEffectiveKeyLen, BytesToBits(iKey->Size()));

		TRAP(iExpectedWeakResult, TCrypto::IsSymmetricWeakEnoughL(minKeySize));

	}

	else if( cipherType == _L8("RC2CBC") )

	{

		iCipherType = ERC2CBC;

		// weak enough if either aKey or aEffectiveKeyLen is weak

		TInt minKeySize = Min(iEffectiveKeyLen, BytesToBits(iKey->Size()));

		TRAP(iExpectedWeakResult, TCrypto::IsSymmetricWeakEnoughL(minKeySize));

	}

	else if( cipherType == _L8("RC4") )

	{

		iCipherType = ERC4;

		TRAP(iExpectedWeakResult, TCrypto::IsSymmetricWeakEnoughL(BytesToBits(iKey->Size())));

	}

	else if( cipherType == _L8("NULL") )

	{

		iCipherType = ECipherNull;

	}

	else if( cipherType == _L8("DESECB_WKT"))

	{

		iCipherType = EDESECB;	

		iExpectedWeakResult = KErrWeakKey;

	}

	else if( cipherType == _L8("DESCBC_WKT"))

	{

		iCipherType = EDESCBC;	

		iExpectedWeakResult = KErrWeakKey;

	}

	else 

	{

		TBuf<64> cipherName(0);

		cipherName.Copy(cipherType);

		TBuf<256> formattable;

		formattable.Format(_L("\nBad Cipher Type: %S"), &cipherName);

		iConsole.Printf(formattable);

		iActionState = CTestAction::EPostrequisite;

		User::Leave(KErrNotSupported);

	}

	if ( (iCipherType==ERC2ECB) || (iCipherType==ERC2CBC) )

	{

		pos = 0;

		TPtrC8 effKeyLenTemp = Input::ParseElement(aScriptBuffer, KEffKeyLenStart, KEffKeyLenEnd, pos, err);

		if (effKeyLenTemp.Length() != 0)

		{

			TLex8 lex;

			lex.Assign(effKeyLenTemp);

			User::LeaveIfError(lex.Val(iEffectiveKeyLen));

		}

	}

	pos = 0;

	TPtrC8 iterationsTemp = Input::ParseElement(aScriptBuffer, KIterationsStart, KIterationsEnd, pos, err);	

	TLex8 lex;

	lex.Assign(iterationsTemp);

	err = lex.Val(iIterationTime);

	pos = 0;

	TPtrC8 randTemp = Input::ParseElement(aScriptBuffer, KRandDataSizeStart, KRandDataSizeEnd, pos, err);	

	TLex8 randlex;

	randlex.Assign(randTemp);

	err = randlex.Val(iRandDataSize);	

	delete iBody;

	iBody = NULL;

}

void CCryptoTestAction::Hex(HBufC8& aString)

    {

    TPtr8 ptr=aString.Des();

    if (aString.Length()%2)

        {

        ptr.SetLength(0);

        return;

        }

    TInt i;

    for (i=0;i<aString.Length();i+=2)

        {

        TUint8 tmp;

        tmp=(TUint8)(aString[i]-(aString[i]>'9'?('A'-10):'0'));

        tmp*=16;

        tmp|=(TUint8)(aString[i+1]-(aString[i+1]>'9'?('A'-10):'0'));

        ptr[i/2]=tmp;

        }

    ptr.SetLength(aString.Length()/2);

    }