--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/cryptoservices/filebasedcertificateandkeystores/source/keystore/Server/CCreateKey.cpp	Wed Jul 08 11:25:26 2009 +0100
@@ -0,0 +1,243 @@
+/*
+* 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 "CCreateKey.h"
+#include "tokenserverdebug.h"
+#include <bigint.h>
+
+//	Wrapper class because we differ from crypto on what a DH key is
+class CSimpleDHKey : public CBase
+{
+public:
+	static CSimpleDHKey* NewL(TInt aSize);
+	~CSimpleDHKey();
+public:
+	inline RInteger& DHKey() {return (iKey);};
+private:
+	CSimpleDHKey() {};
+	void ConstructL(TInt aSize);
+private:
+	RInteger iKey;
+};
+
+CSimpleDHKey* CSimpleDHKey::NewL(TInt aSize)
+{
+	CSimpleDHKey* me = new (ELeave) CSimpleDHKey();
+	CleanupStack::PushL(me);
+	me->ConstructL(aSize);
+	CleanupStack::Pop(me);
+	return (me);
+}
+
+void CSimpleDHKey::ConstructL(TInt aSize)
+{
+	iKey = RInteger::NewRandomL(aSize - 1);							
+}
+
+CSimpleDHKey::~CSimpleDHKey()
+{
+	iKey.Close();
+}
+
+CKeyCreator::CKeyCreator()
+: CActive(EPriorityStandard), iAction(EIdle)
+{
+	CActiveScheduler::Add(this);
+}
+
+
+CKeyCreator::~CKeyCreator()
+{
+	Cancel(); 
+
+	iCreatorThread.LogonCancel(iStatus);
+	iCreatorThread.Close();
+
+	delete iCreateData;
+}
+
+//	Spin a thread to create an appropriate key, if successful, left on CleanupStack
+void CKeyCreator::DoCreateKeyAsyncL(CKeyInfo::EKeyAlgorithm aAlgorithm, TInt aSize, TRequestStatus& aStatus)
+{
+	if ( (aSize <= 0) ||
+		 (aAlgorithm==CKeyInfo::EInvalidAlgorithm) || 
+		 ((aAlgorithm!=CKeyInfo::ERSA) && (aAlgorithm!=CKeyInfo::EDSA) && (aAlgorithm!=CKeyInfo::EDH)) )
+		 User::Leave(KErrArgument);
+
+	iClientStatus = &aStatus;
+	*iClientStatus = KRequestPending;
+	
+	iCreateData = new (ELeave) CKeyCreatorData(aAlgorithm, aSize);
+	
+//	OK, ready to start the async operation...do it in RunL
+	iAction = EReadyToCreateKey;
+	iStatus = KRequestPending;
+	SetActive();
+	TRequestStatus* stat = &iStatus;
+	User::RequestComplete(stat, KErrNone);
+}
+
+//	HERE'S THE THREAD TO CREATE THE KEY
+//	Code cannot leave in here, but not as many traps as it looks
+/*static*/ TInt CKeyCreator::CreatorThreadEntryPoint(TAny* aParameters)
+{
+	CTrapCleanup* cleanup = CTrapCleanup::New();
+	if (!cleanup)
+		User::Exit(KErrNoMemory);
+
+#ifdef _DEBUG
+	TokenServerDebug::PauseOOMTest();
+#endif
+	
+	ASSERT(aParameters);
+	TInt result = KErrNone;
+	CKeyCreatorData* createData = static_cast<CKeyCreatorData*>(aParameters);
+	switch (createData->iKeyAlgorithm)
+	{
+		case(CKeyInfo::ERSA):
+			{//	Currently, CRT signing is not supported, in case the key is to be used
+			//	for such, create a standard (private) key as part of the pair
+				TRAP(result, createData->iKey.iRSAKey = CRSAKeyPair::NewL(createData->iSize));
+			}
+			break;
+		case (CKeyInfo::EDSA):
+			{
+				TRAP(result, createData->iKey.iDSAKey = CDSAKeyPair::NewL(createData->iSize));
+			}
+			break;
+		case (CKeyInfo::EDH):
+			{//	Generate a number that's less than N. The snag is that
+			//	we don't know what N is. We do know that it'll be of a
+			//	particular size, so we can safely generate any number
+			//	with less than iSize digits
+				TRAP(result, createData->iKey.iDHKey = CSimpleDHKey::NewL(createData->iSize));				
+			}
+			break;
+		default:
+			ASSERT(EFalse);
+			result = KErrArgument;
+	}
+
+#ifdef _DEBUG
+	TokenServerDebug::ResumeOOMTest();
+#endif
+	delete cleanup;
+	User::Exit(result);
+	return (KErrNone);
+}
+
+CRSAKeyPair* CKeyCreator::GetCreatedRSAKey()
+{//	Check algorithm is as expected, return NULL if no key or wrong type
+	if ( (!iCreateData) || (CKeyInfo::ERSA!=iCreateData->iKeyAlgorithm) )
+		return (NULL);
+	else
+		return (iCreateData->iKey.iRSAKey);
+}
+
+CDSAKeyPair* CKeyCreator::GetCreatedDSAKey()
+{//	Check algorithm is as expected, return NULL if no key or wrong type
+	if ( (!iCreateData) || (CKeyInfo::EDSA!=iCreateData->iKeyAlgorithm) )
+		return (NULL);
+	else
+		return (iCreateData->iKey.iDSAKey);
+}
+
+void CKeyCreator::GetCreatedDHKey(RInteger& aDHKey)
+{
+	ASSERT(iCreateData);
+	ASSERT(CKeyInfo::EDH==iCreateData->iKeyAlgorithm);		
+	aDHKey = iCreateData->iKey.iDHKey->DHKey();
+}
+
+void CKeyCreator::DoCancel()
+{//	Only do the cancel if in the middle of creating a key.  Kill the thread.
+	if (iAction!=EIdle)
+	{
+		TExitType exitType = iCreatorThread.ExitType();
+		if (EExitPending==exitType)	//	Still alive, so kill it
+			iCreatorThread.Kill(KErrCancel);
+
+		iAction = EIdle;
+	}
+
+	ASSERT(iClientStatus);
+	User::RequestComplete(iClientStatus, KErrCancel);
+}
+
+void CKeyCreator::RunL()
+{
+	ASSERT(iClientStatus);
+	
+	switch (iAction)
+	{
+		case (EReadyToCreateKey):
+		{	// Spin off the thread and pass it the parameter data, then stand by
+			// INC118634
+			// To be safe, we should use anonymous threads because naming a thread means anybody could have opened a handle on the thread,
+			// most likely system applications which want to know about panicing threads. So next thread creation will fail with KErrAlreadyExist(-11).
+			User::LeaveIfError(iCreatorThread.Create(KNullDesC, CreatorThreadEntryPoint, KDefaultStackSize, NULL, (TAny*)iCreateData));
+			iStatus = KRequestPending;
+			iCreatorThread.Logon(iStatus);
+			iAction = ECreatedKey;
+			SetActive();
+			iCreatorThread.Resume();
+		}
+		break;
+		case (ECreatedKey):
+		{//	Notify the caller
+			ASSERT(iClientStatus);
+			// May be OOM creating logon, in which case we should kill thread
+			if (iStatus.Int() == KErrNoMemory)
+				{
+				TExitType exitType = iCreatorThread.ExitType();
+				if (EExitPending==exitType)	//	Still alive, so kill it
+					iCreatorThread.Kill(KErrNone);
+				}
+			
+			User::RequestComplete(iClientStatus, iStatus.Int());
+			iAction = EIdle;
+		}
+		break;
+		default:
+			ASSERT(EFalse);
+	}
+}
+
+TInt CKeyCreator::RunError(TInt anError)
+{
+	if (iClientStatus)
+		User::RequestComplete(iClientStatus, anError);
+
+	return (KErrNone);
+}
+
+
+
+CKeyCreator::CKeyCreatorData::CKeyCreatorData(CKeyInfo::EKeyAlgorithm aAlgorithm, TInt aSize)
+:	iSize(aSize), iKeyAlgorithm(aAlgorithm)
+{}
+
+CKeyCreator::CKeyCreatorData::~CKeyCreatorData()
+{
+	if (iKeyAlgorithm==CKeyInfo::ERSA)
+		delete iKey.iRSAKey;
+	else if (iKeyAlgorithm==CKeyInfo::EDSA)
+		delete iKey.iDSAKey;
+	else if (iKeyAlgorithm==CKeyInfo::EDH)
+		delete iKey.iDHKey;
+}