/*

 * 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);

     }