/*

* Copyright (c) 2006-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: 

* Implementation of the CKeyGenerator class

* INCLUDES

*

*/

#include "KeyGenerator.h"

// ===========================================================================

// Construction/Destruction

// ===========================================================================

CKeyGenerator::CKeyGenerator()

	{

	m_nKeyLength = DEFAULT_KEY_LENGTH;

	m_Cipher = EDSACipher;

	m_bPassword = FALSE;

	m_bAsk = FALSE;

	m_bVerbose = FALSE;

	memset(m_PrivateKeyFile,0, MAXLEN);

	memset(m_sPassword, 0, MAXLEN);

	}

CKeyGenerator::~CKeyGenerator()

	{

	}

const int CKeyGenerator::GetKeyLength() const

	{

	return m_nKeyLength;

	}

void CKeyGenerator::SetKeyLength(int aKeyLen)

	{

	m_nKeyLength = aKeyLen;

	}

const TCHAR* CKeyGenerator::GetPrivateKeyFile() const

	{

	return m_PrivateKeyFile;

	}

void CKeyGenerator::SetPrivateKeyFile(_TCHAR* aPk)

	{

	_tcscpy(m_PrivateKeyFile, aPk);

	}

void CKeyGenerator::SetPassword(_TCHAR* aPw)

	{

	_tcscpy(m_sPassword, aPw);

	}

void CKeyGenerator::SetPasswordRequirement(bool aPwReq)

	{

	m_bPassword = 	aPwReq;

	}

int CKeyGenerator::GenerateSeed(int aLength, unsigned char** aBuf)

//Generate a seed data for DSA key generation

	{

	HCRYPTPROV hProv = 0;

	*aBuf = (unsigned char *)malloc(aLength * sizeof(unsigned char));

	// Get handle to default key context

	if(!CryptAcquireContext(&hProv, NULL, MS_ENHANCED_PROV, PROV_RSA_FULL, 0))

		{

		if(!CryptAcquireContext(&hProv, NULL, MS_ENHANCED_PROV, PROV_RSA_FULL, CRYPT_NEWKEYSET))

			{

			if(!CryptAcquireContext(&hProv, NULL, MS_DEF_PROV, PROV_RSA_FULL, 0))

				{

				if(!CryptAcquireContext(&hProv, NULL, MS_DEF_PROV, PROV_RSA_FULL, CRYPT_NEWKEYSET))

					{	

					PrintErrorInfo("Error acquiring context!", EMSCrypto, constparams);

					return FAIL;

					}

				}

			}

		}

	//Generate a random number

	if(!CryptGenRandom(hProv, aLength, *aBuf))

		{

		PrintErrorInfo("Error generating random number!", EMSCrypto, constparams);

		return FAIL;

		}

	return SUCCESS;

	}

const _TCHAR* CKeyGenerator::GetPassword()

	{

	return m_sPassword;

	}

void CKeyGenerator::SetAskPassword(bool aAsk)

	{

	m_bAsk = aAsk;

	}

void CKeyGenerator::SetVerbose(bool aVerbose)

	{

	m_bVerbose = aVerbose;

	}

void CKeyGenerator::PrintErrorInfo(char* aMsg, TErrType aEntType, const char* aFileName, int aLineNum)

	{

	PrintError(aMsg, aEntType, m_bVerbose, aFileName, aLineNum);

	}