/*

* Copyright (c) 2004-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 <padding.h>

#include "tpaddingSSLv3.h"

#include <securityerr.h>

CTestPadSSLv3::CTestPadSSLv3()

	{

   SetTestStepName(KPadSSLv3);

	}

CTestPadSSLv3::~CTestPadSSLv3()

	{

	}

TVerdict CTestPadSSLv3::doTestStepL()

	{

    SetTestStepResult(EPass);

    __UHEAP_MARK;

	INFO_PRINTF1(_L("Test of padding with type SSLv3"));

	TInt blockSize;

    TInt textSize;

	if (!GetStringFromConfig(ConfigSection(), _L("TestCaseName"), iTestCaseName))

 		{

 		INFO_PRINTF1(_L("Could not find TestCaseName in tpadSSLv3.ini"));

      	return EFail;

 		}

    if (!GetIntFromConfig(ConfigSection(), _L("BlockSize"), blockSize))

 		{

 		INFO_PRINTF1(_L("Could not find blocksize in tpadSSLv3.ini"));

      	return EFail;

 		}

  	if(iTestCaseName.Compare(_L("CipherAES_CBC"))==0 || iTestCaseName.Compare(_L("CipherDES_CBC"))==0 || iTestCaseName.Compare(_L("CipherRC2_CBC"))==0)

 		{

 		TestSSLv3Padding(blockSize);

 		}

 	else

 		{

 		if (!GetIntFromConfig(ConfigSection(), _L("TextSize"), textSize))

   			{

      		INFO_PRINTF1(_L("Could not find TextSize in tpadSSLv3.ini"));

      		return EFail;

   			}

   		if (!GetStringFromConfig(ConfigSection(), _L("ExpectedResult"), iExpectedResult))

 			{

 			INFO_PRINTF1(_L("Could not find TextSize in tpadSSLv3.ini"));

      		return EFail;

 			}

 		TestSSLv3CorruptPadding(blockSize, textSize);	

 		}

	__UHEAP_MARKEND;

	return TestStepResult();

	}

void CTestPadSSLv3::TestSSLv3Padding(TInt aBlockSize)

	{

	CPaddingSSLv3 *padding = CPaddingSSLv3::NewLC(aBlockSize);

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

		{

		HBufC8 *padInData = HBufC8::NewLC(i);

		HBufC8 *padOutData = HBufC8::NewLC(i+(aBlockSize-i%aBlockSize));

		TPtr8 in(padInData->Des());

		TPtr8 out(padOutData->Des());

		TInt j;

		for (j = 0; j < i; j++)

			{

			TInt text('a'+j%25);

			in.Append(text);

			}

		TRAPD(err, padding->DoPadL(in, out));

		INFO_PRINTF3(_L("The error returned for input size %d is %d"), i, err);

		TEST(err == KErrNone);

		TInt totalLength = out.Length();

		TInt paddingLength = aBlockSize - in.Length()%aBlockSize;

		// Test that the total length is a multiple of blockSize

		TEST((totalLength % aBlockSize) == 0);

		// Test that the padding bytes are equal in value to the paddingLength,

		// ie, if padding length is 5 the 5 last octets in the out array should be 0x05

		// This is according to RFC2246 (TLS1.0). The padding content in SSL3.0 is arbitrary.

		for (TInt i = paddingLength; i > 0 ; i--)

			{

			TEST(out[out.Length()-i] == paddingLength - 1);

			}

		// Test that the data has not been corrupted

		TEST(in == out.Left(out.Length() - paddingLength));

		CleanupStack::PopAndDestroy(2, padInData); // padInData, padOutData

		}

	CleanupStack::PopAndDestroy(padding);

	}

void CTestPadSSLv3::TestSSLv3CorruptPadding(TInt aBlockSize, TInt aTextSize)

	{

	CPaddingSSLv3 *padding = CPaddingSSLv3::NewLC(aBlockSize);

	HBufC8 *padInData = HBufC8::NewLC(aTextSize);

	TInt paddingBytes = 0;

	//If BlockSize is 0, Divide by 0 is undefined

	if(aBlockSize != 0)

		{

		paddingBytes = (aBlockSize - aTextSize % aBlockSize);

		}

	HBufC8 *padOutData = HBufC8::NewLC(aTextSize + paddingBytes);

	TPtr8 in(padInData->Des());

	TPtr8 out(padOutData->Des());

	TInt j;

	for (j = 0; j < aTextSize; j++)

   		{

	 	TInt text('a'+j%25);

	 	in.Append(text);

      	}

	TRAPD(err, padding->DoPadL(in, out));

	if(iExpectedResult.Compare(_L("CorruptBlockSize")) ==0)

 		{

 		TEST(err == KErrArgument);

 		}

	else if(iExpectedResult.Compare(_L("Valid")) ==0)

 		{

 		TEST(err == KErrNone);

 		}

	else if(iExpectedResult.Compare(_L("InvalidPadding")) == 0)

		{

		TEST(err == KErrInvalidPadding);

		}	

	TInt totalLength = out.Length();

	TInt paddingLength = 0;

	//If BlockSize is 0, Divide by 0 is undefined

	if(aBlockSize != 0)

		{

      	paddingLength = aBlockSize - in.Length()%aBlockSize;

		}

	//If BlockSize is 0, Divide by 0 is undefined

	if(aBlockSize != 0)

		{

   		// Test that the total length is a multiple of blockSize

		TEST((totalLength % aBlockSize) == 0);

		}

	// Test that the padding bytes are equal in value to the paddingLength,

	// ie, if padding length is 5 the 5 last octets in the out array should be 0x05

	// This is according to RFC2246 (TLS1.0). The padding content in SSL3.0 is arbitrary.

	for (TInt i = paddingLength; i > 0 ; i--)

		{

	 	TEST(out[out.Length()-i] == paddingLength - 1);

      	}

	if(aBlockSize > 0)

   		{

   		// Test that the data has not been corrupted

   		TEST(in == out.Left(out.Length() - paddingLength));

   		}

	CleanupStack::PopAndDestroy(2, padInData); // padInData, padOutData

	CleanupStack::PopAndDestroy(padding);

	}

CTestUnpadSSLv3::CTestUnpadSSLv3()

	{

	SetTestStepName(KUnpadSSLv3);

	}

CTestUnpadSSLv3::~CTestUnpadSSLv3()

	{

	}

TVerdict CTestUnpadSSLv3::doTestStepL()

	{

	SetTestStepResult(EPass);

	__UHEAP_MARK;

	INFO_PRINTF1(_L("Test of unpadding with type SSLv3"));

	TInt blockSize;

	TInt textSize;

	if (!GetIntFromConfig(ConfigSection(), _L("BlockSize"), blockSize))

 		{

 		INFO_PRINTF1(_L("Could not find blocksize in tpadSSLv3.ini"));

      	return EFail;

 		}

 	if (!GetStringFromConfig(ConfigSection(), _L("TestCaseName"), iTestCaseName))

 		{

 		INFO_PRINTF1(_L("Could not find TestCaseName in tpadSSLv3.ini"));

      	return EFail;

 		}	

 	if(iTestCaseName.Compare(_L("CipherAES_CBC"))==0 || iTestCaseName.Compare(_L("CipherDES_CBC"))==0 || iTestCaseName.Compare(_L("CipherRC2_CBC"))==0)

 		{

 		TestSSLv3Unpadding(blockSize);

 		}

 	else

 		{

 		if (!GetIntFromConfig(ConfigSection(), _L("TextSize"), textSize))

   			{

      		INFO_PRINTF1(_L("Could not find TextSize in tpadSSLv3.ini"));

      		return EFail;

   			}

   		if (!GetStringFromConfig(ConfigSection(), _L("ExpectedResult"), iExpectedResult))

 			{

 			INFO_PRINTF1(_L("Could not find TextSize in tpadSSLv3.ini"));

      		return EFail;

 			}

 		TestSSLv3CorruptUnpadding(blockSize, textSize);	

 		}

	__UHEAP_MARKEND;

	return TestStepResult();

	}

void CTestUnpadSSLv3::TestSSLv3Unpadding(TInt aBlockSize)

	{

	CPaddingSSLv3 *padding = CPaddingSSLv3::NewLC(aBlockSize);

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

		{

		HBufC8 *padInData = HBufC8::NewLC(i+(aBlockSize - i%aBlockSize));

		HBufC8 *padOutData = HBufC8::NewLC(i);

		HBufC8 *padCompareData = HBufC8::NewLC(i);

		TPtr8 in(padInData->Des());

		TPtr8 out(padOutData->Des());

		TPtr8 comp(padCompareData->Des());

		TInt j;

		// build up a padded string here

		for (j = 0; j < i; j++)

			{

			TInt text('a'+j%25);

			in.Append(text);

			}

		comp.Append(in);

		TInt paddingBytes = aBlockSize - i%aBlockSize;

		in.SetLength(in.Length() + paddingBytes);

		// pad with arbitary data, to test unpadding of SSL 3.0 padded data

		in[in.Length()-1] = (TUint8) (paddingBytes - 1);

		for (j = 2; j <= paddingBytes; j++)

			{

			in[in.Length()-j] = (TUint8) ('a' + j%25);

			}

		TRAPD(err, padding->UnPadL(in, out));

		INFO_PRINTF3(_L("The error returned for input size %d is %d"), i, err);

		TEST(err == KErrNone);

		// test if the unpadding was correct.

		TEST(out == comp);

		CleanupStack::PopAndDestroy(3, padInData); // padInData, padOutData, padCompareData

		}

	CleanupStack::PopAndDestroy(padding);

	}

void CTestUnpadSSLv3::TestSSLv3CorruptUnpadding(TInt aBlockSize, TInt aTextSize)

	{

	CPaddingSSLv3 *padding = CPaddingSSLv3::NewLC(aBlockSize);

	TInt paddingBytes = 0;

	//If BlockSize is 0, Divide by 0 is undefined

	if(aBlockSize != 0)

   		{

   		paddingBytes = aBlockSize - aTextSize%aBlockSize;

   		}

	HBufC8 *padInData = HBufC8::NewLC(aTextSize+ paddingBytes);

	HBufC8 *padOutData = HBufC8::NewLC(aTextSize);

	HBufC8 *padCompareData = HBufC8::NewLC(aTextSize);

	TPtr8 in(padInData->Des());

	TPtr8 out(padOutData->Des());

	TPtr8 comp(padCompareData->Des());

	TInt j;

	if(in.Length() < aTextSize)

		{

		for (j = 0; j < in.Length(); j++)

   			{

	 		TInt text('a'+j%25);

	 		in.Append(text);

      		}

		}

	else

		{

		// build up a padded string here

		for (j = 0; j < aTextSize; j++)

   			{

	 		TInt text('a'+j%25);

	 		in.Append(text);

      		}

		}

	comp.Append(in);

	in.SetLength(in.Length() + paddingBytes);

	if(aBlockSize > 0)

  		{

   		// pad with arbitary data, to test unpadding of SSL 3.0 padded data

   		in[in.Length()-1] = (TUint8) (paddingBytes - 1);

   		for (j = 2; j <= paddingBytes; j++)

   			{

	 		in[in.Length()-j] = (TUint8) ('a' + j%25);

      		}

  		 }  

	TRAPD(err, padding->UnPadL(in, out));

    if(iExpectedResult.Compare(_L("CorruptBlockSize")) ==0)

 		{

 		TEST(err == KErrArgument);

 		}

	else if(iExpectedResult.Compare(_L("Valid")) ==0)

 		{

 		TEST(err == KErrNone);

 		}

	else if(iExpectedResult.Compare(_L("InvalidPadding")) == 0)

		{

		TEST(err == KErrInvalidPadding);

		}

	// test if the unpadding was correct.

	TEST(out == comp);

	CleanupStack::PopAndDestroy(3, padInData); // padInData, padOutData, padCompareData

	CleanupStack::PopAndDestroy(padding);

	}