--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/crypto/weakcryptospi/test/tpbe/tactionsetreadpfs.cpp	Thu Sep 10 14:01:51 2009 +0300
@@ -0,0 +1,239 @@
+/*
+* 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 "tactionsetreadpfs.h"
+#include "t_input.h"
+#include <cryptostrength.h>
+#include <securityerr.h>
+#include <pbedata.h>
+#include <f32file.h>
+#include <s32file.h>
+#include <stdlib.h>
+#include <s32mem.h>
+#include <s32std.h>
+
+_LIT8(KReadPFSStart, "<readpfs>");
+_LIT8(KReadPFSEnd, "</readpfs>");
+_LIT8(KStrengthStart, "<strength>");
+_LIT8(KStrengthEnd, "</strength>");
+_LIT8(KInputStart, "<input>");
+_LIT8(KInputEnd, "</input>");
+_LIT8(KPasswdStart, "<passwd>");
+_LIT8(KPasswdEnd, "</passwd>");
+_LIT8(KStrong, "strong");
+_LIT16(KWeakFileName, "\\tpbe\\weak.dat");
+_LIT16(KStrongFileName, "\\tpbe\\strong.dat");
+
+CTestAction* CActionSetReadPFS::NewL(RFs& aFs,
+									   CConsoleBase& aConsole,
+									   Output& aOut, 
+									   const TTestActionSpec& aTestActionSpec)
+	{
+	CTestAction* self = CActionSetReadPFS::NewLC(aFs, aConsole,
+		aOut, aTestActionSpec);
+	CleanupStack::Pop();
+	return self;
+	}
+
+CTestAction* CActionSetReadPFS::NewLC(RFs& aFs,
+										CConsoleBase& aConsole,
+										Output& aOut, 
+										const TTestActionSpec& aTestActionSpec)
+	{
+	CActionSetReadPFS* self = new(ELeave) CActionSetReadPFS(aFs, aConsole, aOut);
+	CleanupStack::PushL(self);
+	self->ConstructL(aTestActionSpec);
+	return self;
+	}
+
+CActionSetReadPFS::~CActionSetReadPFS()
+	{
+	delete iBody;
+	}
+
+CActionSetReadPFS::CActionSetReadPFS(RFs& aFs, 
+								 CConsoleBase& aConsole,
+								 Output& aOut)
+								 
+: CTestAction(aConsole, aOut), iFs(aFs)
+	{
+	}
+
+void CActionSetReadPFS::ConstructL(const TTestActionSpec& aTestActionSpec)
+	{
+	CTestAction::ConstructL(aTestActionSpec);
+	iBody = HBufC8::NewL(aTestActionSpec.iActionBody.Length());
+	iBody->Des().Copy(aTestActionSpec.iActionBody);
+	
+	}
+
+void CActionSetReadPFS::DoPerformPrerequisite(TRequestStatus& aStatus)
+	{
+	TRequestStatus* status = &aStatus;
+	TInt err = KErrNone;
+	TInt pos = 0;
+	TPtrC8 encryptElement = Input::ParseElement(*iBody, KReadPFSStart,
+		KReadPFSEnd, pos, err);
+
+	TPtrC8 strengthTemp = Input::ParseElement(encryptElement, KStrengthStart, 
+		KStrengthEnd, pos=0, err);
+		
+	TDriveUnit sysDrive (RFs::GetSystemDrive());
+	if (strengthTemp.CompareF(KStrong))
+		{
+		iFileName = sysDrive.Name();
+		iFileName.Append(KStrongFileName);
+		}
+	
+	else
+		{
+		iFileName = sysDrive.Name();
+		iFileName.Append(KWeakFileName);
+		}
+		
+	TPtrC8 passwdTemp = Input::ParseElement(encryptElement, KPasswdStart, 
+		KPasswdEnd, pos=0, err);
+	iPasswd = HBufC::NewL(passwdTemp.Length());
+	TPtr16 passwdTemp3( iPasswd->Des());
+	passwdTemp3.Copy(passwdTemp);
+
+	TPtrC8 inputTemp = Input::ParseElement(encryptElement, KInputStart, 
+		KInputEnd, pos=0, err);
+	iInput = HBufC8::NewL(inputTemp.Length());
+	*iInput = inputTemp;
+
+	User::RequestComplete(status, KErrNone);
+	iActionState = CTestAction::EAction;
+	}
+
+void CActionSetReadPFS::DoPerformPostrequisite(TRequestStatus& aStatus)
+	{
+	TRequestStatus* status = &aStatus;
+	delete iPasswd;
+	delete iInput;
+
+	iFinished = ETrue;
+	User::RequestComplete(status, KErrNone);
+	}
+
+void CActionSetReadPFS::DoReportAction(void)
+	{
+	}
+
+void CActionSetReadPFS::DoCheckResult(TInt)
+	{
+
+	}
+
+void CActionSetReadPFS::PerformAction(TRequestStatus& aStatus)
+	{
+	__UHEAP_MARK;
+	TRequestStatus* status = &aStatus;
+	iResult = EFalse;
+	
+	//Change the password by appending the letter 'a' to it
+	HBufC* newPasswordTemp = HBufC::NewMaxLC(iPasswd->Length()+1);
+	TPtr newPassword = newPasswordTemp->Des();
+	newPassword.Copy(*iPasswd);
+	newPassword.Append('a');
+
+	//prepare to read the streams back in, creating a new TPBEncryptionData
+	RStoreReadStream read;
+	// open the next PFS
+	CFileStore *store = CPermanentFileStore::OpenLC(iFs, iFileName, EFileRead | EFileWrite);
+	TStreamId dataStreamId(2); // we know it was the second stream written
+	read.OpenLC(*store, dataStreamId);
+	CleanupStack::Pop();
+	//read in Encryption data
+	CPBEncryptionData* data = CPBEncryptionData::NewL(read);
+	read.Close();
+	CleanupStack::PushL(data);
+
+	//read in encrypted master key
+	TStreamId keyStreamId(1); // we know it was the first stream written
+	read.OpenLC(*store, keyStreamId);
+	CleanupStack::Pop();
+	HBufC8* encryptedMasterKey = HBufC8::NewLC(read, 10000); //some large number
+	read.Close();
+	//create a new set encryption class
+	CPBEncryptSet* set = CPBEncryptSet::NewLC(*data, *encryptedMasterKey, newPassword);
+
+	//read in ciphertext key
+	TStreamId cipherId(3); // we know it was the third stream written
+	read.OpenLC(*store, cipherId);
+	CleanupStack::Pop();
+	HBufC8* ciphertextTemp = HBufC8::NewLC(read, 10000); //some large number
+	read.Close();
+	TPtr8 ciphertext = ciphertextTemp->Des();
+	
+	HBufC8* plaintextTemp = HBufC8::NewLC(ciphertext.Length());
+	TPtr8 plaintext = plaintextTemp->Des();	
+
+	// weak crypto should fail if trying to decrypt strong
+	TRAPD(err, 
+		{
+		CPBDecryptor* decryptor = set->NewDecryptLC();
+		decryptor->Process(ciphertext, plaintext);
+
+		//this Mid call is due to get rid of the decrypted padding at the end
+		if ((plaintext.Mid(0,iInput->Length()) == *iInput) &&
+		!((TCrypto::Strength() == TCrypto::EWeak) && (iFileName == KStrongFileName)))
+			{
+			iResult = ETrue;
+			}
+		CleanupStack::PopAndDestroy(decryptor);
+		});
+
+	if ((err == KErrKeyNotWeakEnough) && 
+	(TCrypto::Strength() == TCrypto::EWeak) && (iFileName == KStrongFileName))
+		{
+		iResult = ETrue;
+		}
+
+	CleanupStack::PopAndDestroy(plaintextTemp);
+	CleanupStack::PopAndDestroy(ciphertextTemp);
+	CleanupStack::PopAndDestroy(set);
+	CleanupStack::PopAndDestroy(encryptedMasterKey);
+	CleanupStack::PopAndDestroy(data);
+	CleanupStack::PopAndDestroy(store);
+	CleanupStack::PopAndDestroy(newPasswordTemp);
+
+	User::RequestComplete(status, KErrNone);
+	iActionState = CTestAction::EPostrequisite;
+	__UHEAP_MARKEND;
+	}
+
+void CActionSetReadPFS::Hex(HBufC8& aString)
+    {
+    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);
+    }