sl@0: /*
sl@0: * Copyright (c) 2004-2009 Nokia Corporation and/or its subsidiary(-ies).
sl@0: * All rights reserved.
sl@0: * This component and the accompanying materials are made available
sl@0: * under the terms of the License "Eclipse Public License v1.0"
sl@0: * which accompanies this distribution, and is available
sl@0: * at the URL "http://www.eclipse.org/legal/epl-v10.html".
sl@0: *
sl@0: * Initial Contributors:
sl@0: * Nokia Corporation - initial contribution.
sl@0: *
sl@0: * Contributors:
sl@0: *
sl@0: * Description: 
sl@0: *
sl@0: */
sl@0: 
sl@0: 
sl@0: #include <random.h>
sl@0: #include <padding.h>
sl@0: #include <securityerr.h>
sl@0: #include "tpaddingPKCS7.h"
sl@0: #include <cryptopanic.h>
sl@0: 
sl@0: static void GenerateInput(TInt aBytes, TPtr8& in)
sl@0:  	{
sl@0:  	for (TInt j = 0; j < aBytes; j++)
sl@0:  		{
sl@0:  		TInt text('a'+j%25);
sl@0:  		in.Append(text);
sl@0:  		}
sl@0:  	}
sl@0: 
sl@0: CTestPadPKCS7::CTestPadPKCS7()
sl@0: 	{
sl@0: 	SetTestStepName(KPadPKCS7);
sl@0: 	}
sl@0: 
sl@0: CTestPadPKCS7::~CTestPadPKCS7()
sl@0: 	{
sl@0: 	}
sl@0: 
sl@0: TVerdict CTestPadPKCS7::doTestStepL()
sl@0: 	{
sl@0: 	SetTestStepResult(EPass);
sl@0: 	__UHEAP_MARK;
sl@0:    
sl@0: 	INFO_PRINTF1(_L("Test of PKCS7  padding"));
sl@0: 
sl@0: 	for (TInt i = 1; i < 255; i++)
sl@0: 		{
sl@0: 		TestPKCS7Padding(i);
sl@0: 		}
sl@0:   
sl@0: 	__UHEAP_MARKEND;
sl@0: 	return TestStepResult();
sl@0: 	}
sl@0: 
sl@0: 
sl@0: void CTestPadPKCS7::TestPKCS7Padding(TInt aBlockSize)
sl@0: 	{
sl@0:    	CPaddingPKCS7 *padding = CPaddingPKCS7::NewLC(aBlockSize);
sl@0: 	// Starts with zero input size(total block data is filled/padded with block size) 
sl@0: 	for (TInt i = 0 ; i <= aBlockSize; i++)
sl@0: 		{
sl@0: 		HBufC8 *padInData = HBufC8::NewLC(i);
sl@0: 		HBufC8 *padOutData = HBufC8::NewLC(i+(aBlockSize-i%aBlockSize));
sl@0: 		TPtr8 in(padInData->Des());
sl@0: 		TPtr8 out(padOutData->Des());
sl@0: 
sl@0: 		GenerateInput(i, in);
sl@0: 		TRAPD(err, padding->PadL(in, out));
sl@0: 		
sl@0: 		TEST(err == KErrNone);
sl@0: 		
sl@0: 		TInt totalLength = out.Length();
sl@0: 		TUint paddingLength = aBlockSize - in.Length()%aBlockSize;
sl@0: 		// Test that the total length is a multiple of blockSize
sl@0: 		TEST((totalLength % aBlockSize) == 0);
sl@0:       
sl@0: 		// Test that the padding bytes are equal in value to the paddingLength,
sl@0: 		// ie, if padding length is 5 the 5 last octets in the out array should be 0x05
sl@0: 		for (TInt j = paddingLength; j > 0 ; j--)
sl@0: 			{
sl@0: 			TEST(out[out.Length()-j] == paddingLength);
sl@0: 			}
sl@0:       
sl@0: 		// Test that the data has not been corrupted
sl@0: 		TEST(in == out.Left(out.Length() - paddingLength));
sl@0: 		
sl@0: 		CleanupStack::PopAndDestroy(2, padInData); // padInData, padOutData
sl@0: 		}
sl@0: 	CleanupStack::PopAndDestroy(padding);
sl@0: 	}
sl@0:    
sl@0: CTestUnpadPKCS7::CTestUnpadPKCS7()
sl@0: 	{
sl@0:    	SetTestStepName(KUnpadPKCS7);
sl@0: 	}
sl@0: 
sl@0: CTestUnpadPKCS7::~CTestUnpadPKCS7()
sl@0: 	{
sl@0: 	}
sl@0: 
sl@0: TVerdict CTestUnpadPKCS7::doTestStepL()
sl@0: 	{
sl@0: 	SetTestStepResult(EPass);
sl@0: 	__UHEAP_MARK;
sl@0: 
sl@0: 	INFO_PRINTF1(_L("Test of PKCS7 unpadding"));
sl@0: 	for (TInt i = 1; i < 255; i++)
sl@0: 		{
sl@0: 		TestPKCS7Unpadding(i);
sl@0: 		}
sl@0: 
sl@0: 	__UHEAP_MARKEND;
sl@0: 	return TestStepResult();
sl@0: 	}
sl@0: 
sl@0: 
sl@0: void CTestUnpadPKCS7::TestPKCS7Unpadding(TInt aBlockSize)
sl@0: 	{
sl@0: 	CPaddingPKCS7 *padding = CPaddingPKCS7::NewLC(aBlockSize);
sl@0: 
sl@0: 	// Input must be < aBlockSize otherwise this wouldn't be 
sl@0: 	// a padded block
sl@0: 	for (TInt i = 0 ; i < aBlockSize; i++)
sl@0: 		{
sl@0: 		// Input to un-padding should always be an entire block
sl@0: 		// for correctly data.
sl@0: 		HBufC8 *padInData = HBufC8::NewLC(aBlockSize);
sl@0: 		HBufC8 *padOutData = HBufC8::NewLC(i);
sl@0: 		HBufC8 *padCompareData = HBufC8::NewLC(i);
sl@0: 		TPtr8 in(padInData->Des());
sl@0: 		TPtr8 out(padOutData->Des());
sl@0: 		TPtr8 comp(padCompareData->Des());
sl@0: 
sl@0: 		GenerateInput(i, in);
sl@0: 		comp.Copy(in);		
sl@0: 
sl@0: 		in.SetLength(aBlockSize);
sl@0: 		TInt paddingBytes = aBlockSize - (i % aBlockSize);
sl@0: 		for (TInt j = 1; j <= paddingBytes; j++)
sl@0: 			{
sl@0: 			in[in.Length()-j] = (TUint8)paddingBytes;
sl@0: 			}
sl@0: 
sl@0: 		TRAPD(err, padding->UnPadL(in, out));
sl@0: 		if (err != KErrNone)
sl@0: 			{
sl@0: 			INFO_PRINTF3(_L("The Error returned for block size %d is %d"), aBlockSize,err);
sl@0: 			}
sl@0: 		TEST(err == KErrNone); // Verify UnPadL leave code 
sl@0: 		TEST(out == comp); // Verify  UnPadL output data with expected data
sl@0: 		CleanupStack::PopAndDestroy(3, padInData); // padInData, padOutData, padCompareData
sl@0: 		}
sl@0: 	CleanupStack::PopAndDestroy(padding);
sl@0: 	}
sl@0: 
sl@0:  
sl@0: CTestUnpadCorruptPKCS7::CTestUnpadCorruptPKCS7()
sl@0: 	{
sl@0: 	SetTestStepName(KUnpadCorruptPKCS7);
sl@0: 	}
sl@0:  
sl@0: CTestUnpadCorruptPKCS7::~CTestUnpadCorruptPKCS7()
sl@0: 	{
sl@0: 	}
sl@0:  
sl@0: TVerdict CTestUnpadCorruptPKCS7::doTestStepL()
sl@0:  	{
sl@0:  	SetTestStepResult(EPass);
sl@0:  	__UHEAP_MARK;
sl@0:  
sl@0:  	TInt blockSize;
sl@0:  	TInt textSize;
sl@0:  	TInt paddingNum = 0;
sl@0:  	
sl@0:  	if (GetIntFromConfig(ConfigSection(), _L("blocksize"), blockSize))
sl@0:  		{
sl@0:  		if (GetIntFromConfig(ConfigSection(), _L("textsize"), textSize))
sl@0:  			{
sl@0:  			if (GetIntFromConfig(ConfigSection(), _L("paddingbyte"), paddingNum))
sl@0:  				{
sl@0:  				INFO_PRINTF1(_L("Test of PKCS7 unpadding with corrupt data"));
sl@0:  				TUint8 paddingByte = Min(paddingNum, 255);
sl@0:    				TestCorruptPKCS7Unpadding(blockSize, textSize, paddingByte);	  					
sl@0:  				}				
sl@0:  			else
sl@0:  				{
sl@0:  				ERR_PRINTF1(_L("Missing parameter - paddingbyte"));
sl@0:  				}
sl@0:  			}
sl@0:  		else
sl@0:  			{
sl@0:  			ERR_PRINTF1(_L("Missing parameter - textsize"));
sl@0:  			}
sl@0:  		}
sl@0:  	else
sl@0: 		{
sl@0:  		ERR_PRINTF1(_L("Missing parameter - blocksize"));
sl@0:  		}
sl@0:  
sl@0:  	__UHEAP_MARKEND;
sl@0:  	return TestStepResult();
sl@0:  	}
sl@0:  
sl@0: void CTestUnpadCorruptPKCS7::TestCorruptPKCS7Unpadding(TInt aBlockSize, TInt aTextSize, TUint8 aPaddingByte)
sl@0: 	{
sl@0:  	CPaddingPKCS7 *padding = CPaddingPKCS7::NewLC(aBlockSize);
sl@0:  	TInt paddingBytes = (aBlockSize - aTextSize % aBlockSize);
sl@0:  	
sl@0: 	HBufC8 *padInData = HBufC8::NewLC(aTextSize + paddingBytes);
sl@0:  	HBufC8 *padOutData = HBufC8::NewLC(aTextSize);
sl@0:  	TPtr8 in(padInData->Des());
sl@0:  	TPtr8 out(padOutData->Des());
sl@0:  
sl@0:  	GenerateInput(aTextSize, in);
sl@0:  
sl@0:  	in.SetLength(in.Length() + paddingBytes);
sl@0:  	for (TInt j = 1; j <= paddingBytes; j++)
sl@0:  		{
sl@0:  		in[in.Length()-j] = (TUint8) aPaddingByte;
sl@0:  		}
sl@0:  
sl@0:  	TRAPD(err, padding->UnPadL(in, out));
sl@0:  	
sl@0:  	if ( err == KErrInvalidPadding )
sl@0: 		{
sl@0:   		INFO_PRINTF2(_L("The PKCS7 unpadding UnPadL method returned error is %d"), err);
sl@0:   		TEST(err == KErrInvalidPadding);
sl@0:   		SetTestStepResult(EPass);
sl@0:   		}
sl@0: 	else if ( err == KErrNone )
sl@0: 		{
sl@0: 		TEST(err == KErrNone);
sl@0: 		}
sl@0: 	CleanupStack::PopAndDestroy(3, padding); // padding, padInData, padOutData
sl@0: 	}
sl@0: 
sl@0: CTestPaddingCorruptPKCS7::CTestPaddingCorruptPKCS7()
sl@0: 	{
sl@0: 	SetTestStepName(KPaddingCorruptPKCS7);
sl@0: 	}
sl@0:  
sl@0: CTestPaddingCorruptPKCS7::~CTestPaddingCorruptPKCS7()
sl@0: 	{
sl@0: 	}
sl@0:  
sl@0: TVerdict CTestPaddingCorruptPKCS7::doTestStepL()
sl@0:  	{
sl@0:  	SetTestStepResult(EPass);
sl@0:  	__UHEAP_MARK;
sl@0:  
sl@0:  	TInt blockSize;
sl@0:  	TInt textSize;
sl@0:  	 	
sl@0:  	if (GetIntFromConfig(ConfigSection(), _L("blocksize"), blockSize))
sl@0:  		{
sl@0:  		if (GetIntFromConfig(ConfigSection(), _L("textsize"), textSize))
sl@0:  			{
sl@0:  			INFO_PRINTF1(_L("Test of PKCS7 unpadding with corrupt data"));
sl@0:    			TestCorruptPKCS7padding(blockSize, textSize);	  					
sl@0:  			}
sl@0:  		else
sl@0:  			{
sl@0:  			ERR_PRINTF1(_L("Missing parameter - textsize"));
sl@0:  			}
sl@0:  		}
sl@0:  	else
sl@0: 		{
sl@0:  		ERR_PRINTF1(_L("Missing parameter - blocksize"));
sl@0:  		}
sl@0:  
sl@0:  	__UHEAP_MARKEND;
sl@0:  	return TestStepResult();
sl@0:  	}
sl@0:  
sl@0: void CTestPaddingCorruptPKCS7::TestCorruptPKCS7padding(TInt aBlockSize, TInt aTextSize)
sl@0: 	{
sl@0: 	CPaddingPKCS7 *padding = CPaddingPKCS7::NewLC(aBlockSize);
sl@0: 
sl@0: 	TInt paddingBytes = 0;
sl@0: 	//Divide by 0 is undefined.
sl@0: 	if(aBlockSize != 0)
sl@0: 		{
sl@0: 		paddingBytes = aBlockSize - (aTextSize % aBlockSize);
sl@0: 		}
sl@0: 		
sl@0:  	HBufC8 *padOutData = HBufC8::NewLC(aTextSize + paddingBytes);
sl@0:  	HBufC8 *padInData = HBufC8::NewLC(aTextSize);
sl@0:  	TPtr8 in(padInData->Des());
sl@0:  	TPtr8 out(padOutData->Des());
sl@0:  
sl@0:  	GenerateInput(aTextSize, in);
sl@0:  	TRAPD(err, padding->PadL(in, out));
sl@0: 
sl@0: 	INFO_PRINTF2(_L("The PKCS7 padding PadL method returned error is %d"), err);
sl@0: 	
sl@0: 	//check expected result
sl@0: 	TPtrC expectedContent;
sl@0:  	if (GetStringFromConfig(ConfigSection(), _L("case"), expectedContent))
sl@0:  		{
sl@0:  		if(expectedContent.Compare(_L("InvalidPadding")) ==0)
sl@0:  			{
sl@0:  			TEST(err == KErrInvalidPadding);
sl@0:  			}
sl@0:  		else if(expectedContent.Compare(_L("Valid")) ==0)
sl@0:  			{
sl@0:  			TEST(err == KErrNone);
sl@0:  			}
sl@0:  		else if(expectedContent.Compare(_L("CorruptBlockSize")) ==0)
sl@0:  			{
sl@0:  			TEST(err == KErrArgument);
sl@0:  			}
sl@0:  		}
sl@0:  	
sl@0:  	//skip the checking on padded data if padding is unsuccessful(no padding is done), 
sl@0:  	//otherwise the erroneous operation on output descriptor will panic.
sl@0:  	if(err != KErrNone)
sl@0:  		{
sl@0:  		CleanupStack::PopAndDestroy(3, padding);
sl@0:  		return;
sl@0:  		}
sl@0: 	
sl@0: 	TInt totalLength = out.Length();
sl@0: 	TInt inLength = in.Length();
sl@0: 	
sl@0: 	TUint paddingLength = 0;
sl@0: 	//Divide by 0 is undefined.
sl@0: 	if(aBlockSize != 0)
sl@0: 		{
sl@0: 		paddingLength = aBlockSize - inLength%aBlockSize;
sl@0: 		// Test that the total length is a multiple of blockSize
sl@0: 		TEST((totalLength % aBlockSize) == 0);
sl@0: 		}			
sl@0: 	
sl@0:  	// Test that the padding bytes are equal in value to the paddingLength,
sl@0: 	// ie, if padding length is 5 the 5 last octets in the out array should be 0x05
sl@0: 	for (TInt j = paddingLength; j > 0 ; j--)
sl@0: 		{
sl@0: 		TEST(out[out.Length()-j] == paddingLength);
sl@0: 		}
sl@0: 
sl@0: 	
sl@0:  	
sl@0:  	// Test that the data has not been corrupted
sl@0: 	TEST(in == out.Left(out.Length() - paddingLength));
sl@0:  	
sl@0:  	CleanupStack::PopAndDestroy(3, padding); // padInData, padOutData, padCompareData, padding
sl@0: 	}
sl@0: