/*

* 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 CRSAKeyGenerator class

* INCLUDES

*

*/

#include "RSAKeyGenerator.h"

#include <iostream.h>

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

// Construction/Destruction

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

CRSAKeyGenerator::CRSAKeyGenerator()

{

}

CRSAKeyGenerator::~CRSAKeyGenerator()

{

}

int CRSAKeyGenerator::Generate()

//Generate an RSA key with pre-determined length

{

	RSA*		   pRSAKey	    = NULL;

	FILE*		   fp		    = NULL;

	LPSTR		   pbPassword   = NULL;

	const _TCHAR*  pPrivKeyFile = NULL;

	int retVal  = FAIL;

	int retFunc = FAIL;

	pPrivKeyFile = GetPrivateKeyFile();

	if(!pPrivKeyFile)

		{

		PrintErrorInfo("Bad parameter error!", EGeneric, constparams);

		return 0;

		}

	OPENSSL_add_all_algorithms_conf();

	ERR_load_crypto_strings();

	int keyLength = 0;

	keyLength = GetKeyLength();

	try

		{

		//Generate RSA key

		_tprintf(_T("\nGenerating RSA key ."));

		do

			{

			pRSAKey = RSA_generate_key(keyLength, RSA_F4, RSAKeyStatus, NULL);

			}

		while((retVal = RSA_check_key(pRSAKey)) == 0); // if return is 0, the key should be regenerated!

		if(retVal != 1)

			{

			PrintErrorInfo("RSA key generation failed!", EOPENSSL, constparams);

			throw EOPENSSL;

			}

		_tprintf(_T("Generated!\n"));

		//Create a key file

		fp = _tfopen(pPrivKeyFile, _T("w"));

		if(!fp)

			{

			PrintErrorInfo("Error creating key file!", EGeneric, constparams);

			throw EGeneric;

			}

		//Write generated DSA key to the key file

		if(m_bPassword)

			{

			DWORD len = 0;

			len = _tcslen(GetPassword());

			pbPassword = MakeMBCSString(GetPassword(), CP_UTF8, len);

			retVal = PEM_write_RSAPrivateKey(fp, pRSAKey, EVP_des_ede3_cbc(), (unsigned char *) pbPassword, len, NULL, NULL);

			delete pbPassword;

			}

		else if(m_bAsk)

			{

			retVal = PEM_write_RSAPrivateKey(fp, pRSAKey, EVP_des_ede3_cbc(), NULL, 0, NULL, NULL);

			}

		if(!retVal)

			{

			PrintErrorInfo("Error writing to key file", EOPENSSL, constparams);

			throw EOPENSSL;

			}

		//Free variables

		RSA_free(pRSAKey);

		fclose(fp);	

		//Get command prompt handle

		HANDLE	hndl = 0;

		DWORD bytesWritten;

		hndl = GetStdHandle(STD_OUTPUT_HANDLE);

		_tprintf(_T("\nCreated key: "));

		WriteConsole(hndl, pPrivKeyFile, wcslen(pPrivKeyFile), &bytesWritten, 0);

		retFunc = SUCCESS;

		}

	catch (...)

		{

		//Delete rsa params

		if(pRSAKey != NULL) 

			{

			RSA_free(pRSAKey);

			}

		}

	return retFunc;

	}

//RSA key generation callback function

static void RSAKeyStatus(int aCode, int aArg, void *aCbArg)

	{

	if (aCode == 1 && aArg && !(aArg % 3))

		{

		printf(".");

		}

	return;

	}