crypto/weakcryptospi/test/tcryptospi/src/symmetriccipherencryptdecryptstep.cpp
changeset 8 35751d3474b7
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/crypto/weakcryptospi/test/tcryptospi/src/symmetriccipherencryptdecryptstep.cpp	Thu Sep 10 14:01:51 2009 +0300
@@ -0,0 +1,184 @@
+/*
+* Copyright (c) 2007-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: 
+* Example CTestStep derived implementation
+*
+*/
+
+
+/**
+ @file
+ @internalTechnology
+*/
+#include "symmetriccipherencryptdecryptstep.h"
+
+using namespace CryptoSpi;
+
+CSymmetricCipherEncryptDecryptStep::~CSymmetricCipherEncryptDecryptStep()
+	{
+	}
+
+
+CSymmetricCipherEncryptDecryptStep::CSymmetricCipherEncryptDecryptStep()
+	{
+	SetTestStepName(KSymmetricCipherEncryptDecryptStep);
+	}
+
+
+TVerdict CSymmetricCipherEncryptDecryptStep::doTestStepPreambleL()
+	{
+	SetTestStepResult(EPass);
+	return TestStepResult();
+	}
+
+
+TVerdict CSymmetricCipherEncryptDecryptStep::doTestStepL()
+	{
+	INFO_PRINTF1(_L("*** Symmetric Cipher - Encrypt/Decrypt ***"));
+	INFO_PRINTF2(_L("HEAP CELLS: %d"), User::CountAllocCells());
+	
+  	if (TestStepResult()==EPass)
+		{
+		
+		//Assume faliure, unless all is successful
+		SetTestStepResult(EFail);
+		
+		TVariantPtrC operationMode;
+
+		// Create a Symmetric Cipher with the values from the ini file	
+		CryptoSpi::CSymmetricCipher * impl = NULL;	
+			
+		CKey* key = NULL;
+		SetupCipherL(ETrue, EFalse, operationMode, impl, key);
+
+		INFO_PRINTF1(_L("Plugin loaded."));
+	
+		CleanupStack::PushL(key);
+		CleanupStack::PushL(impl);
+
+				
+		HBufC8* iv = NULL;
+		TInt err(0);
+				
+		if((TUid(operationMode) == KOperationModeCBCUid) || (TUid(operationMode) == KOperationModeCTRUid))
+			{
+			TInt blockSize(0);
+					
+			if (TUid(operationMode) == KOperationModeCTRUid)
+				{
+				blockSize = CtrModeCalcBlockSizeL(*impl);
+				}
+			else
+				{
+				blockSize = impl->BlockSize();
+				}
+			
+			// (blocksize is in bits so to allocate the correct number of bytes divide by 8)
+			// iv is left on the cleanup stack for the duration of the test and deleted in a conditional at the end of the outer block.
+			// If this conditional block changes, take care to update the condition for deleting this allocated IV, near the end of this function.
+			iv = HBufC8::NewLC(blockSize/8);	
+			
+			// blocksize is in bits so to allocate the correct number of 8 byte chunks divide by 64
+			for(TInt i = 0 ; i <blockSize/64 ; i++)
+				{
+				iv->Des().Append(_L8("12345678"));
+				}
+					
+			TRAP_LOG(err,impl->SetIvL(iv->Des()));
+			}
+			
+		//read from src file
+		HBufC8* srcData = ReadInPlaintextL();
+		CleanupStack::PushL(srcData);
+					
+		//Create buffer for encrypted data
+		TInt maxOutputLength = impl->MaxFinalOutputLength(srcData->Length());
+		HBufC8* encrypted =	HBufC8::NewLC(maxOutputLength);
+		TPtr8 encryptedPtr = encrypted->Des();
+
+		INFO_PRINTF1(_L("Encrypting Source Data..."));
+
+		//Perform the encryption operation
+		TRAP_LOG(err, impl->ProcessFinalL((*srcData), encryptedPtr));
+					
+		if(encryptedPtr.Compare((*srcData)) != 0)
+			{
+			//Switch to decrypt
+			TRAP_LOG(err,impl->SetCryptoModeL(KCryptoModeDecryptUid));
+
+			//If in CTR mode need to reset the keystream to the start of the sequence used for encryption
+			if(TUid(operationMode) == KOperationModeCTRUid)
+				{
+				impl->SetIvL(iv->Des());
+				}
+					
+			//Create a buffer for the decrypted data
+			maxOutputLength = encryptedPtr.Length();
+					
+			TInt bufSize = impl->MaxFinalOutputLength(maxOutputLength);
+					
+			HBufC8* output = HBufC8::NewLC(bufSize);
+			TPtr8 outputPtr = output->Des();
+					
+			INFO_PRINTF1(_L("Decrypting Data..."));
+					
+			//Perform the decryption operation
+			TRAP_LOG(err, impl->ProcessFinalL(encryptedPtr, outputPtr));
+					
+			if(err == KErrNone)
+				{
+				//Check that the source data matches the data thats 
+				//been encrypted then decrypted
+				if(	!outputPtr.Compare(*srcData) )
+					{
+					INFO_PRINTF1(_L("PASS : Decrypted Data and Source Match"));
+					SetTestStepResult(EPass);
+					}
+				else
+					{
+					INFO_PRINTF1(_L("FAIL : Decrypted Data and Source Mismatch"));	
+					SetTestStepResult(EFail);
+					}
+				}
+						
+			CleanupStack::PopAndDestroy(output); 
+			}
+		else
+			{
+			INFO_PRINTF1(_L("FAIL : Encrpyted Data and Source Data length and content is the same"));	
+			SetTestStepResult(EFail);	
+			}
+					
+		CleanupStack::PopAndDestroy(encrypted); 
+		CleanupStack::PopAndDestroy(srcData);
+		if((TUid(operationMode) == KOperationModeCBCUid) || (TUid(operationMode) == KOperationModeCTRUid))
+			{
+			// Iv is left on the cleanupstack at creation.  
+			// If it becomes possible for operationMode to be modified during
+			// the test this needs to be re-engineered.
+			CleanupStack::PopAndDestroy(iv);	
+			}
+		CleanupStack::PopAndDestroy(2, key);		
+		}
+	INFO_PRINTF2(_L("HEAP CELLS: %d"), User::CountAllocCells());
+
+	return TestStepResult();
+
+	}
+
+
+TVerdict CSymmetricCipherEncryptDecryptStep::doTestStepPostambleL()
+	{
+	return TestStepResult();
+	}