/*

* 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 "keystreamutils.h"

#include "asymmetrickeys.h"

#ifdef SYMBIAN_KEYSTORE_USE_AUTH_SERVER

#include <s32mem.h>

#include <pbe.h>

#include <pbedata.h>

#endif // SYMBIAN_KEYSTORE_USE_AUTH_SERVER

#include <e32debug.h>

void ExternalizeL(const CRSAPublicKey& aKey, RWriteStream& aStream)

	{

	aStream << aKey.N() << aKey.E();

	}

void ExternalizeL(const CRSAPrivateKey& aData, RWriteStream& aStream)

	{

	aStream << aData.N();

	// Check the incoming RSA private key (standard or CRT)	

	TRSAPrivateKeyType keyType = aData.PrivateKeyType();

	aStream.WriteInt32L((TInt32)keyType);

	if (EStandard==keyType)

		{

		const CRSAPrivateKeyStandard& key = static_cast<const CRSAPrivateKeyStandard&>(aData);

		aStream << key.D();

		}

	else if (EStandardCRT==keyType)

		{

		const CRSAPrivateKeyCRT& key = static_cast<const CRSAPrivateKeyCRT&>(aData);

		aStream << key.P() << key.Q() << key.DP() << key.DQ() << key.QInv();

		}

	else

		{

		User::Leave(KErrNotSupported);

		}

	}

void ExternalizeL(const CDSAPublicKey& aKey, RWriteStream& aStream)

	{

	aStream << aKey.P() << aKey.Q() << aKey.G() << aKey.Y();

	}

void ExternalizeL(const CDSAPrivateKey& aKey, RWriteStream& aStream)

	{

	aStream << aKey.P() << aKey.Q() << aKey.G() << aKey.X();

	}

void CreateL(RReadStream& aStream, CRSAPublicKey*& aOut)

	{

	RInteger N, keyPublicExp;

	CreateLC(aStream, N);

	CreateLC(aStream, keyPublicExp);

	aOut = CRSAPublicKey::NewL(N, keyPublicExp);

	CleanupStack::Pop(2, &N); // keyPublicExp, N

	}

void CreateL(RReadStream& aStream, CRSAPrivateKey*& aOut)

	{

	RInteger privateN;

	CreateLC(aStream, privateN);

	TRSAPrivateKeyType keyType = EStandard;

	keyType = (TRSAPrivateKeyType)aStream.ReadInt32L();

	if (EStandard==keyType)

		{

		RInteger D;

		CreateLC(aStream, D);

		aOut = CRSAPrivateKeyStandard::NewL(privateN, D);

		CleanupStack::Pop(&D);

		}

	else if (EStandardCRT==keyType)

		{

		RInteger p, q, dP, dQ, qInv;

		CreateLC(aStream, p);

		CreateLC(aStream, q);

		CreateLC(aStream, dP);

		CreateLC(aStream, dQ);

		CreateLC(aStream, qInv);

		aOut = CRSAPrivateKeyCRT::NewL(privateN, p, q, dP, dQ, qInv);

		CleanupStack::Pop(5, &p);

		}

	else

		{

		User::Leave(KErrNotSupported);

		}

	CleanupStack::Pop(&privateN);

	}

void CreateL(RReadStream& aStream, CDSAPublicKey*& aOut)

	{

	RInteger P, Q, G, Y;

	CreateLC(aStream, P);

	CreateLC(aStream, Q);

	CreateLC(aStream, G);

	CreateLC(aStream, Y);

	aOut = CDSAPublicKey::NewL(P, Q, G, Y);

	CleanupStack::Pop(4, &P);

	}

void CreateL(RReadStream& aStream, CDSAPrivateKey*& aOut)

	{

	RInteger P, Q, G, X;

	CreateLC(aStream, P);

	CreateLC(aStream, Q);

	CreateLC(aStream, G);

	CreateLC(aStream, X);

	aOut = CDSAPrivateKey::NewL(P, Q, G, X);

	CleanupStack::Pop(4, &P);

	}

#ifdef SYMBIAN_KEYSTORE_USE_AUTH_SERVER

/**

 * The input stream contains data in encrypted form. This method 

 * supports pbe. In this case the key is the password. It 

 * retrieves the plaintext data by decrypting the data using the 

 * supplied key.

 */

HBufC8* DecryptFromStreamL( RReadStream& aInStream, TPtrC8& aKey )

	{

	CPBEncryptionData* data = CPBEncryptionData::NewL(aInStream);

	CleanupStack::PushL(data);

	TInt32 encKeyLength = aInStream.ReadInt32L();

	HBufC8* encKey = HBufC8::NewMaxLC(encKeyLength);

	TPtr8 encKeyPtr(encKey->Des());

	encKeyPtr.FillZ();

	aInStream.ReadL(encKeyPtr,encKeyLength);

	CPBEncryptElement* encryption = CPBEncryptElement::NewLC(*data,aKey);

	CPBDecryptor* decryptor = encryption->NewDecryptLC();

	HBufC8* plaintext = HBufC8::NewLC(decryptor->MaxOutputLength(encKeyPtr.Length())); 

	TPtr8 plaintextPtr = plaintext->Des();	

	plaintextPtr.FillZ();

	decryptor->ProcessFinalL(encKeyPtr, plaintextPtr);

	CleanupStack::Pop(plaintext);

	CleanupStack::PopAndDestroy(4,data); // encKey, encryption, decryptor 

	return plaintext;

	}

#endif // SYMBIAN_KEYSTORE_USE_AUTH_SERVER