/*

* 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: 

* tpkcs5main.cpp

*

*/

#include <pkcs5kdf.h>

#include <pkcs12kdf.h>

#include "t_testhandler.h"

#include "t_testsetup.h"

#include "tscripttests.h"

#include "tactionderivekey.h"

LOCAL_D void callExampleL() // initialize and call example code under cleanup stack

    {

    START_SCRIPT_LIST

	SCRIPT_ITEM(CActionDeriveKey,_L8("DeriveKey"))

	END_SCRIPT_LIST

	TDriveUnit sysDrive (RFs::GetSystemDrive());

	TDriveName sysDriveName (sysDrive.Name());

	TBuf<64> scriptFile (sysDriveName);

	scriptFile.Append(_L("\\tpkcs5kdf\\tpkcs12kdftests.txt"));

	TBuf<64> logFile (sysDriveName);

	logFile.Append(_L("\\tpkcs5kdf\\tpkcs12kdftests.log"));

	CTestSetup::CreateAndRunTestsL(theTestTypes, scriptFile, logFile);

    }

GLDEF_C TInt E32Main() // main function called by E32

    {

	__UHEAP_MARK;

	CTrapCleanup* cleanup = CTrapCleanup::New(); // get clean-up stack

	TRAPD(error, callExampleL());

	__ASSERT_ALWAYS(!error,User::Panic(_L("tpkcs5"),error));

	delete cleanup; // destroy clean-up stack

	__UHEAP_MARKEND;

	return 0;

    }

// The following code was used to generate test data.  It is left in

// the source file because it provides an easy way to generate a large

// number of PKCS#12 derived keys if required.

#if 0

static void TestPkcs12KdfL();

static void TestPasswordL(

	TInt aKeyLenInBits, TInt aIterCount, const TDesC8& aSalt,

	const TDesC8& aPassword, const TDesC& aPasswordText);

static void PrintKey(const TDesC8& aKey);

static void TestPkcs12KdfL()

	{

	__UHEAP_MARK;

	const TInt KKeyLenCount = 4;

	const TInt KKeyLens[KKeyLenCount] = {40, 128, 168, 368};

	const TInt KIterCount = 8;

	const TInt KIterCounts[KIterCount] = {1, 2, 4, 8, 128, 1024, 1536, 2048};

	const TInt KSaltCount = 3;

	TPtrC8 salts[KSaltCount];

	const TUint8 KSalt4[4] = {0x53, 0x41, 0x4c, 0x54};

	salts[0].Set(KSalt4, 4);

	const TUint8 KSalt20[20] =

		{

		0x1d, 0x56, 0x50, 0x78,		0xc3, 0x50, 0x6f, 0x89,

		0xbd, 0xa7, 0x3b, 0xb6,		0xe3, 0xe5, 0xb8, 0xa3,

		0x68, 0x3d, 0xd3, 0x62

		};

	salts[1].Set(KSalt20, 20);

	const TUint8 KSalt25[25] =

		{

		0xe2, 0x2c, 0x7b, 0x03,		0x16, 0x3a, 0xe5, 0x47,

		0xf8, 0x23, 0x9d, 0xa4,		0x0d, 0x6f, 0x46, 0xd7,

		0x9e, 0xa3, 0xc6, 0xff,		0xb3, 0xf0, 0x4e, 0xbe,

		0x61

		};

	salts[2].Set(KSalt25, 25);

	const TInt KPasswordCount = 5;

	HBufC8* passwords[KPasswordCount];

	TPtrC passwords0[KPasswordCount] =

		{

		_L("0000"), _L("0001"), _L("PSWD"),

		_L("password"), _L("abcdefghijklmnopqrstuvwxyz")

		};

	for (TInt i = 0; i < KPasswordCount; ++i)

		passwords[i] = PKCS12KDF::GeneratePasswordLC(passwords0[i]);

	for (TInt klenIdx = 0; klenIdx < KKeyLenCount; ++klenIdx)

		{

		for (TInt iterIdx = 0; iterIdx < KIterCount; ++iterIdx)

			{

			for (TInt saltIdx = 0; saltIdx < KSaltCount; ++saltIdx)

				{

				for (TInt pwdIdx = 0; pwdIdx < KPasswordCount; ++pwdIdx)

					{

					TestPasswordL(KKeyLens[klenIdx], KIterCounts[iterIdx], salts[saltIdx], *passwords[pwdIdx], passwords0[pwdIdx]);

					}

				}	// for (saltIdx = 0; saltIdx < KSaltCount; ++saltIdx)

			}	// for (int iterIdx = 0; iterIdx < KIterCount; ++iterIdx)

		}	// for (TInt klenIdx = 0; klenIdx < KKeyLenCount; ++klenIdx)

	CleanupStack::PopAndDestroy(KPasswordCount, passwords[0]);

	__UHEAP_MARKEND;

	}

static void TestPasswordL(

	TInt aKeyLenInBits, TInt aIterCount, const TDesC8& aSalt,

	const TDesC8& aPassword, const TDesC& aPasswordText)

	{

	__UHEAP_MARK;

	TBuf8<512> key;

	key.SetLength(aKeyLenInBits / 8);

	RDebug::Print(

		_L("\nkey len = %d, iter count = %d, password = \"%S\", salt len = %d"),

		aKeyLenInBits, aIterCount, &aPasswordText, aSalt.Length());

	PKCS12KDF::DeriveKeyL(key, PKCS12KDF::EIDByteEncryptKey, aPassword, aSalt, aIterCount);

	PrintKey(key);

	__UHEAP_MARKEND;

	}

static void PrintKey(const TDesC8& aKey)

/**

	Print the supplied key in hex byte format, with

	16 bytes on each row.

	@param	aKey			Key to print.

 */

	{

	const TInt KBlockSize = 16;

	TInt keyLen = aKey.Length();

	TInt rowCount = keyLen / KBlockSize;

	if ((keyLen % KBlockSize) != 0)

		++rowCount;

	for (TInt row = 0; row < rowCount; ++row)

		{

		TInt start = row * KBlockSize;

		TInt end = Min(start + KBlockSize, keyLen);

		TBuf<64> line;

		line.Format(_L("[%04x]"), start);

		for (TInt i = start; i < end; ++i)

			line.AppendFormat(_L(" %02x"), aKey[i]);

		line.Append('\n');

		RDebug::Print(line);

		}

	}

#endif	// #if 0