/*

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

* makekeys main function

* INCLUDES

*

*/

#include <windows.h>

#include <iostream> 

#include <tchar.h>

#include "utils.h"

#include "makekeys.h"

#define NORSA	//to keep DTI sweet!

#ifdef KEYS	//define KEYS for standalone keygen support (ie -KEYS works)

#undef KEYS

#endif

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

// CONSTANTS

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

// Set up the cout stream so that we can use it with either narrow or wide

// chars at build time

using namespace std;

#undef OUT

#ifdef _UNICODE

#define OUT wcout

#else

#define OUT cout

#endif

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

// CMakeKeys

// The main application object. Interfaces to Wincrypto.lib

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

CMakeKeys::CMakeKeys()

:m_fVerbose(FALSE)

	{ ; }

int CMakeKeys::Run(const int argc, const _TCHAR *argv[], const _TCHAR *envp[])

// Inputs   : argc, argv, envp - The command line passed to the process

	{	

	BOOL bValid = TRUE;

	m_KeyGen = NULL;

	pMakeCert = NULL;

	pReq = NULL;

	try

		{

		m_CmdOptions.ParseCommandLine(argc, argv, &m_dwMode);

		// We now have valid parameters

		SetVerbose(m_CmdOptions.GetVerbose());

		}

	catch(TCommandLineException)

		{

		// Show the title and help

		ShowBanner();

		ShowUsage();

		return 1;

		}

	catch(TUtilsException err)

		{

		ShowUtilsError(err);

		}

	catch(TAppError apperr)

		{

		OUT << endl << _T("Internal application error" << apperr) << endl;

		}

#ifdef KEYS

	//Key maker, -keys is not in the spec but its handy to have for testing!

	DoVerbage(_T("Generating private key..."));

	if (m_dwMode == EKeys)

		{ 

		if (m_CmdOptions.GetKeyType() == EDSACipher)

			{

			m_KeyGen = new CDSAKeyGenerator;

			}

		else

			{

			m_KeyGen = new CRSAKeyGenerator;

			}

		if (!m_KeyGen) throw ErrOutOfMemory;

		m_KeyGen->SetKeyLength(m_CmdOptions.GetCMDKeyLen());

		m_KeyGen->SetPasswordRequirement(m_CmdOptions.GetPasswordEnabled());

		m_KeyGen->SetPrivateKeyFile(m_CmdOptions.GetPrivateName());

		m_KeyGen->SetVerbose(m_CmdOptions.GetVerbose());

		//If user did not supply password to encrypt private key before writing, get it

		if (!m_CmdOptions.GetPasswordEnabled())

			{

			char Warn[128];

			OUT << _T("Warning: the private key should be encrypted with the -password option") << endl;

			OUT << _T("Do you want to use a password (y/n)? ");

			cin >> Warn;

			if (stricmp(Warn, "n") != 0)

				{

				m_KeyGen->SetAskPassword(TRUE);

				}

			} 

		else

			{

			m_KeyGen->SetPassword(m_CmdOptions.GetPassword());

			}

		m_KeyGen->Generate();

		if (m_KeyGen) delete m_KeyGen;

		}

#endif

	//create a Self Signed Cert

	if (m_dwMode == ECert)

		{

		DoVerbage(_T("Generating private key and self-signed certificate..."));

		pMakeCert = new CCertificateGenerator;

		if (!pMakeCert) throw ErrOutOfMemory;

		pMakeCert->SetKeyLength(m_CmdOptions.GetCMDKeyLen());

		pMakeCert->SetPasswordRequirement(m_CmdOptions.GetPasswordEnabled());

		if(m_CmdOptions.GetPasswordEnabled())

		pMakeCert->SetPassword(m_CmdOptions.GetPassword());

		if(m_CmdOptions.GetDNameString())

			pMakeCert->SetDName(m_CmdOptions.GetDNameString());

		pMakeCert->SetPrivKeyName(m_CmdOptions.GetPrivateName());

		pMakeCert->SetCertificateName(m_CmdOptions.GetPublicName());

		pMakeCert->SetVerbose(m_CmdOptions.GetVerbose());

#ifndef NORSA

		pMakeCert->SetKeyType(m_CmdOptions.GetKeyType());

#else

			pMakeCert->SetKeyType(EDSACipher);

#endif

			if ((!m_CmdOptions.GetPasswordEnabled()) && (!pMakeCert->Exists(m_CmdOptions.GetPrivateName()) ))

				{

				char Warn[128];

				OUT << _T("Warning: the private key should be encrypted with the -password option") << endl;

				OUT << _T("Do you want to use a password (y/n)? ");

				cin >> Warn;

				if (stricmp(Warn, "n") != 0)

					{

					pMakeCert->SetAskPassword(TRUE);

					}

				}

			if (!pMakeCert->Exists(m_CmdOptions.GetPrivateName()))

				{

				//OUT << _T("Generating private key...") << endl;

				}

			pMakeCert->MakeSelfSigned(_T(""),m_CmdOptions.GetDNameString(),

			m_CmdOptions.GetPrivateName(), m_CmdOptions.GetPublicName(),

			m_CmdOptions.GetCMDKeyLen(),0,m_CmdOptions.GetCMDCertExpiryInDays());

			if (pMakeCert) delete pMakeCert;

			}

		//Create a CSR from command line parameters

		if (m_dwMode == EReq)

			{

			DoVerbage(_T("Generating certificate request...\n"));

			pReq = new CCertificateRequestGenerator();

			if (!pReq) throw ErrOutOfMemory;

			pReq->SetVerbose(m_CmdOptions.GetVerbose());	//enable additional reporting within Wincrypto

			pReq->SetPasswordRequirement(m_CmdOptions.GetPasswordEnabled());

			pReq->SetPassword(m_CmdOptions.GetPassword());

			pReq->SetDName(m_CmdOptions.GetDNameString());

			pReq->SetPrivKeyName(m_CmdOptions.GetPrivateName());

			pReq->SetCertificateName(m_CmdOptions.GetPublicName());

			pReq->SetRequestName(m_CmdOptions.GetRequestName());

			pReq->Generate();

			if (pReq) delete pReq;

			}

			return 0;

		}

	void CMakeKeys::ShowBanner()

		// Displays the copyright gumph...

		{

		short	major = KMakeKeysVersion / 100,

				minor = KMakeKeysVersion % 100;

		// Show the title

		OUT << endl << _T("MakeKeys, version ") << major << _T(".") << minor << endl;

		OUT << _T("Creates public/private key pairs and certificate requests for use with makesis.") << endl;

#ifdef _DEBUG

		OUT << _T("Development Version") << endl;

#endif

	OUT << _T("Copyright (c) 2007 Symbian Ltd. All rights reserved.") << endl;

	OUT << endl;

	}

	void CMakeKeys::ShowUtilsError(TUtilsException err)

		// Purpose  : Not currently used

		// Inputs   : err - the error ID

		{

		switch(err)

			{

			case ErrNotEnoughMemory:

				DoErrMsg(_T("not enough memory"));

				break;

			}

		}

	void CMakeKeys::ShowUsage()

		// Write command line help to STDOUT (cout/wcout)

		{

		OUT << _T("Usage:") << endl;

	OUT << _T(" makekeys -cert [-expdays <cert-expiry-in-days>] [-password <password> <At least 4 characters>] [-len <key-length>] -dname <distinguised-name-string> <private-key-file> <public-key-cert>") << endl;	

	OUT << _T(" makekeys -req [-password <password> <At least 4 characters>] -dname <distinguised-name-string> <private-key-file> <public-key-cert> <cert-request-file>") << endl;	

	OUT << _T(" makekeys -view <certificate-file>") << endl << endl;

	OUT << _T(" (use -v on the command line for verbose mode)") << endl;

	OUT << _T(" (use -i on the command line for license information)") << endl;

	OUT << endl;

		OUT << _T("Notes:") << endl;

		OUT << _T(" For -cert if the private-key-file does not exist it will be created") << endl;	

		OUT << _T(" Distinguished name string format:") << endl;	

		OUT << _T("    CN - Common Name, e.g. CN=Joe Bloggs") << endl;

		OUT << _T("    C - Country, e.g. C=GB") << endl;

		OUT << _T("    O - Organisation, e.g. O=Acme Ltd") << endl;

		OUT << _T("    OU - Organisational Unit, e.g. OU=Development") << endl;

		OUT << _T("    EM - E-Mail address, e.g. EM=noone@nowhere.com") << endl;

		OUT << _T(" A distinguished name strings needs at least two attributes") << endl;

		OUT << _T(" Certificate expiry date(in days) can be specified by the user (optional) , else defaults to a year.") << endl;

		OUT << endl;

		OUT << _T("Example Usage:") << endl;

		OUT << _T(" makekeys -cert -expdays 3650 -password yourpassword -len 2048 -dname \"CN=Joe Bloggs OU=Development O=Acme Ltd C=GB EM=noone@nowhere.com\" mykey.key mycert.cer") << endl;

		OUT << _T(" makekeys -req -password yourpassword -dname \"CN=Joe Bloggs OU=Development O=Acme Ltd C=GB EM=noone@nowhere.com\" mykey.key mycert.cer myreq.p10") << endl; 

		}

	void CMakeKeys::DoVerbage(const _TCHAR* pszText) const

		// Purpose  : Displays 'verbose' output text - if appropriate

		// Inputs   : pszText - The text to display

		{

		if(Verbose())

			DoMsg(_T(""),pszText);

		}

	void CMakeKeys::DoErrMsg(const _TCHAR* pszText) const

		{

		DoMsg(_T("error: "),pszText);

		}	

	void CMakeKeys::DoMsg(const _TCHAR* pszText1, const _TCHAR* pszText2) const

	// Purpose  : Displays 'verbose' output text - if appropriate

	// Inputs   : pszText - The text to display

		{

		OUT << pszText1 << pszText2 << endl;

		}

	// main function - constructs a CMakeKeys object and calls it's run method

	extern "C"

		{

		int _tmain(int argc, const _TCHAR *argv[], const _TCHAR *envp[])

			{

			CMakeKeys app;

			return app.Run(argc, argv, envp);

			}

		}