/*
* Copyright (c) 2005-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 <s32file.h>
#include "tauthkeysstep.h"

using namespace AuthServer;

CTStepEncryptKey::CTStepEncryptKey()
	{
	SetTestStepName(KTStepEncryptKey);
	}

TVerdict CTStepEncryptKey::doTestStepL()
	{
	if (TestStepResult() != EPass)
		{
		return TestStepResult();
		}
__UHEAP_MARK;		// Check for memory leaks

    TBool res = ETrue; 
    TBool tmp = ETrue; 
	TInt  err = KErrNone;
 

	_LIT8(KPluginData, "ABCDEFGHIJ");
	_LIT8(KBadPluginData, "KLMNOPQRST");
	_LIT(KPluginName, "PasswordPlugin");

    HBufC* pluginName = HBufC::NewLC(14);
	
	*pluginName = KPluginName;

	CTransientKeyInfo* transInfo =
	  CTransientKeyInfo::NewLC(9999);
	  
	CTransientKey* transient =
		transInfo->CreateTransientKeyL(KPluginData);
	CleanupStack::PushL(transient);

	CProtectionKey* protection =
		CProtectionKey::NewL(8);
	CleanupStack::PushL(protection);
		
	CEncryptedProtectionKey* encrypted =
		transient->EncryptL(*protection);
	
	transInfo->SetEncryptedProtectionKeyL(encrypted);
	
	// try to set another key 
	TRAP(err, transInfo->SetEncryptedProtectionKeyL(encrypted));
	TEST(tmp = (err == KErrAlreadyExists));
	res = tmp && res;
	
	// check the encrypted key is different from the protection key
	TEST(tmp = (encrypted->KeyData() != protection->KeyData()));
	res = tmp && res;
	
	CProtectionKey* decrypted =
		transient->DecryptL(*encrypted);
	CleanupStack::PushL(decrypted);

	// check that the decrypted key is the same as the protection key
	TEST(tmp = (protection->KeyData() == decrypted->KeyData()));
	res = tmp && res;
	
	// check we can't use the wrong plugin data
	CTransientKey* transient2 = 0;
	TRAP(err, transient2 = transInfo->CreateTransientKeyL(KBadPluginData)); 
	TEST(tmp = (err != KErrNone));
	res = tmp && res;

	// avoid set but unused warning...
	if (transient2 == 0) transient2 = 0;
	
	CProtectionKey* client = protection->ClientKeyL(0xabababab);
	CleanupStack::PushL(client);

	CProtectionKey* client2 = protection->ClientKeyL(0xcdcdcdcd);
	CleanupStack::PushL(client2);

	// check protection key is different from the client key 
	TEST(tmp = (protection->KeyData() != client->KeyData()));
	res = tmp && res;
	TEST(tmp = (protection->KeyData().Length() == client->KeyData().Length()));
	res = tmp && res;

	// check two client keys are different 
	TEST(tmp = (client->KeyData() != client2->KeyData()));
	res = tmp && res;
	TEST(tmp = (client->KeyData().Length() == client2->KeyData().Length()));
	res = tmp && res;

	// check key lengths larger than SHA1 output is handled correctly 
	CProtectionKey* protection2 =
		CProtectionKey::NewL(512);
	CleanupStack::PushL(protection2);

	CProtectionKey* client3 = protection2->ClientKeyL(0xcdcdcdcd);
	CleanupStack::PushL(client3);
	
	// check protection key is different from the client key 
	TEST(tmp = (protection2->KeyData() != client3->KeyData()));
	res = tmp && res;
	TEST(tmp = (protection2->KeyData().Length() ==
				client3->KeyData().Length()));
	res = tmp && res;

	// cleanup
	CleanupStack::PopAndDestroy(8, transInfo);
	CleanupStack::PopAndDestroy(pluginName);

	SetTestStepResult(res ? EPass : EFail);
	
__UHEAP_MARKEND;	
	return TestStepResult();
	}