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