/*
* 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 "tactionderivekey.h"
#include "t_input.h"
#include <pkcs5kdf.h>
#include <pkcs12kdf.h>
#include <stdlib.h>

_LIT8(KKdfStart, "<kdf>");
_LIT8(KKdfEnd, "</kdf>");
_LIT8(KPKCS12Kdf, "pkcs#12");
_LIT8(KDeriveKeyStart, "<derivekey>");
_LIT8(KDeriveKeyEnd, "</derivekey>");
_LIT8(KKeyStart, "<key>");
_LIT8(KKeyEnd, "</key>");
_LIT8(KPasswdStart, "<passwd>");
_LIT8(KPasswdEnd, "</passwd>");
_LIT8(KSaltStart, "<salt>");
_LIT8(KSaltEnd, "</salt>");
_LIT8(KIterationsStart, "<iterations>");
_LIT8(KIterationsEnd, "</iterations>");
_LIT8(KLeaveInPerformAction, "<leaveinperformaction>");
_LIT8(KLeaveInPerformActionEnd, "</leaveinperformaction>");

CTestAction* CActionDeriveKey::NewL(RFs& aFs,
									   CConsoleBase& aConsole,
									   Output& aOut, 
									   const TTestActionSpec& aTestActionSpec)
	{
	CTestAction* self = CActionDeriveKey::NewLC(aFs, aConsole,
		aOut, aTestActionSpec);
	CleanupStack::Pop();
	return self;
	}

CTestAction* CActionDeriveKey::NewLC(RFs& aFs,
										CConsoleBase& aConsole,
										Output& aOut, 
										const TTestActionSpec& aTestActionSpec)
	{
	CActionDeriveKey* self = new(ELeave) CActionDeriveKey(aFs, aConsole, aOut);
	CleanupStack::PushL(self);
	self->ConstructL(aTestActionSpec);
	return self;
	}

CActionDeriveKey::~CActionDeriveKey()
	{
	delete iBody;
	}

CActionDeriveKey::CActionDeriveKey(RFs& aFs, 
								 CConsoleBase& aConsole,
								 Output& aOut)
								 
: CTestAction(aConsole, aOut), iFs(aFs)
	{
	}

void CActionDeriveKey::ConstructL(const TTestActionSpec& aTestActionSpec)
	{
	CTestAction::ConstructL(aTestActionSpec);
	iBody = HBufC8::NewL(aTestActionSpec.iActionBody.Length());
	iBody->Des().Copy(aTestActionSpec.iActionBody);
	}

void CActionDeriveKey::DoPerformPrerequisite(TRequestStatus& aStatus)
	{
	TRequestStatus* status = &aStatus;
	TInt err = KErrNone;
	TInt pos = 0;
	TPtrC8 deriveKey = Input::ParseElement(*iBody, KDeriveKeyStart, KDeriveKeyEnd, pos, err);

	// KDF is only explicitly specified for PKCS#12 derived keys
	pos = 0;
	TPtrC8 kdfTemp = Input::ParseElement(deriveKey, KKdfStart, KKdfEnd, pos, err);
	if (err == KErrNone)
		iKdf = kdfTemp.AllocL();
	
	pos = 0;
	TPtrC8 passwdTemp = Input::ParseElement(deriveKey, KPasswdStart, KPasswdEnd, pos, err);
	iPasswd = HBufC8::NewL(passwdTemp.Length());
	*iPasswd = passwdTemp;

	pos = 0;
	TPtrC8 iterationsTemp = Input::ParseElement(deriveKey, KIterationsStart, KIterationsEnd, pos, err);
	iIterations = HBufC8::NewL(iterationsTemp.Length() + 1); //added 1 for the null zero used later
	*iIterations = iterationsTemp;

	pos = 0;
	TPtrC8 saltTemp = Input::ParseElement(deriveKey, KSaltStart, KSaltEnd, pos, err);
	iSalt = HBufC8::NewL(saltTemp.Length());
	*iSalt = saltTemp;
	Hex(*iSalt);

	pos = 0;
	TPtrC8 keyTemp = Input::ParseElement(deriveKey, KKeyStart, KKeyEnd, pos, err);
	iKey = HBufC8::NewL(keyTemp.Length());
	*iKey = keyTemp;
	Hex(*iKey);

	iOutput = HBufC8::NewL(iKey->Length());

	pos = 0;
	iLeaveInPerformAction = Input::ParseIntElement(deriveKey, 
		KLeaveInPerformAction, KLeaveInPerformActionEnd, pos, err);
	if (err)
		{
		iLeaveInPerformAction = 0;
		}
	User::RequestComplete(status, KErrNone);
	iActionState = CTestAction::EAction;
	}

void CActionDeriveKey::DoPerformPostrequisite(TRequestStatus& aStatus)
	{
	TRequestStatus* status = &aStatus;
	delete iKey;
	delete iSalt;
	delete iIterations;
	delete iPasswd;
	delete iOutput;
	delete iKdf;
	iKdf = 0;
	
	iFinished = ETrue;
	User::RequestComplete(status, KErrNone);
	}

void CActionDeriveKey::DoReportAction(void)
	{
	}

void CActionDeriveKey::DoCheckResult(TInt)
	{

	}

void CActionDeriveKey::PerformAction(TRequestStatus& aStatus)
	{
	TRequestStatus* status = &aStatus;
	iResult = EFalse;
	
	if (iLeaveInPerformAction)
		{
		User::Leave(KErrArgument);
		}
	iOutput->Des().SetLength(iKey->Length());

	TUint8* nptr= (TUint8*)(iIterations->Des().PtrZ()); 
	TUint32 i = strtoul((char*)nptr, 0, 10); 
	
	iConsole.Printf(_L("."));
	TPtr8 outputActual = iOutput->Des();
	TPtr8 passwdActual = iPasswd->Des();
	if (iKdf != 0 && *iKdf == KPKCS12Kdf)
		{
		// convert the password to PKCS#12 password format
		HBufC* pwdNative = HBufC::NewLC(iPasswd->Length());
		pwdNative->Des().Copy(*iPasswd);
		HBufC8* pwdPKCS12 = PKCS12KDF::GeneratePasswordLC(*pwdNative);
		PKCS12KDF::DeriveKeyL(outputActual, PKCS12KDF::EIDByteEncryptKey, *pwdPKCS12, *iSalt, i);
		CleanupStack::PopAndDestroy(2, pwdNative);
		}
	else	// PKCS#5
		{
		TPtr8 saltActual = iSalt->Des();
		TPKCS5KDF::DeriveKeyL(outputActual, passwdActual, saltActual,i);
		}
		
	if(*iOutput == *iKey)
		{
		iResult = ETrue;
		}
		
	User::RequestComplete(status, KErrNone);
	iActionState = CTestAction::EPostrequisite;
	}

void CActionDeriveKey::Hex(HBufC8& aString)
/**
	Convert the supplied hex string into the binary equivalent.
	
	@param	aString			Hex string.  On entry this contains
							a sequence of hexadecimal characters,
							e.g., "3037AFC8EA".  On exit it is
							half the original length and each two-digit
							hex number is reduced to the matching
							byte value.
 */
    {
    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);
    }