/*
* Copyright (c) 2005-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 "dumpswicertstoretool.h"
#include "options.h"
#include "datalocator.h"
#include "osinterface.h"

#ifdef _MSC_VER
#include <new>
#include <new.h>
#endif /* _MSC_VER */

#ifdef _MSC_VER
// House-keeping to allow ::new to throw rather than return NULL in MSVC++
int throwingHandler(size_t)
	{
	throw std::bad_alloc();
	return 0;
	}
#endif /* _MSC_VER */

// entry point
int main(int argc,char *argv[])
	{
#ifdef _MSC_VER
	_set_new_handler (throwingHandler);	// Make ::new throw rather than return NULL
#endif /* _MSC_VER */

	Options* options = 0;
	//if no arguments are specified,display help msg and exit.
	if(argc < 2)
		{
		Options::DisplayUsage();
		return 0;
		}
	try
		{
		options = new Options(argc, argv) ;
		//Required in order to delete the der files which are present in the current directory.
		string curDirectory = ".";
		string extension = "der";
		OSInterface::DeleteFilesFromDirectory(curDirectory , extension);
		Tool.Run(*options);
		}

	catch (Exceptions aErr)
		{
		Options::DisplayError (aErr);
		delete options;
		return aErr;
		}

	catch (bad_alloc&)
		{
		cout << " Error Allocating Memory " << endl ;
		}

	delete options;
	return 0;
	}

// Class DumpSWICertstoreTool
class DumpSWICertstoreTool Tool;

DumpSWICertstoreTool::DumpSWICertstoreTool()
	:iTocRevision (0),
	iErrors (0)
	{
	}

void DumpSWICertstoreTool::Run(const Options& aOptions)
	{
	if(!aOptions.CreateCCIFile() && !aOptions.CreateDetailCCIFile())
		{
		ofstream iniFile("swicertstore.txt",ios::out);
		iniFile.close();
		}

	//if both ROM and C:\Based certstore are specified.
	if((aOptions.CBasedWritableCertstore()) && (aOptions.ROMCertstore()))
		{
		if(!aOptions.CreateCCIFile() && !aOptions.CreateDetailCCIFile())
			{
			cout << "Merged ROM Based and C:BASED Writable Certstore " << endl << endl;
			}

		//calls C:\Based certstore first
		bool isCBasedValid = InitializeCBasedCertstore(aOptions);
		if(!isCBasedValid)
			{
			cout << "C: Based Writable Certstore is not a Permanent File Store " << endl << endl;
			}
		
		//ROM Based certstore called next.
		bool isROMValid = InitializeROMCerstore(aOptions);
		if(!isROMValid)
			{
			cout << (aOptions.ROMCertstoreFileName()).c_str() << " : Not a Permanent File Store " <<endl <<endl;
			}
		}

	//when only ROM Based dat file specified.
	else if (!(aOptions.CBasedWritableCertstore()) && (aOptions.ROMCertstore()))
		{
		if(!aOptions.CreateCCIFile() && !aOptions.CreateDetailCCIFile())
			{
			cout <<"ROM Based SwiCertStore :" << endl << endl;
			}
		bool isROMValid = InitializeROMCerstore(aOptions);
		if(!isROMValid)
			{
			throw ENotAPermanentFileStore;;
			}
		}

	//when only C:\Based Certstore is specified
	else if((aOptions.CBasedWritableCertstore()) && !(aOptions.ROMCertstore()))
		{
		if(!aOptions.CreateCCIFile() && !aOptions.CreateDetailCCIFile())
			{
			cout << " C:\\ Based Writable SwiCertstore :" << endl << endl;
			}
		bool isCBasedValid = InitializeCBasedCertstore(aOptions);
		if(!isCBasedValid)
			{
			throw ENotAPermanentFileStore;
			}
		}

	if(aOptions.CreateCCIFile() || aOptions.CreateDetailCCIFile())
		{
		cout << "[CAPABILITIES]" <<endl;
		cout << iIssuer <<endl;
		cout << "[MANDATORY]" <<endl;
		cout << iMandatory <<endl;
		}
	}

bool DumpSWICertstoreTool::InitializeCBasedCertstore(const Options& aOptions)
	{
	StoreFile* store = 0;
	int size = aOptions.GetDirectoryFiles().size();
	if(size == 0)
		{
		if(!aOptions.CreateCCIFile() && !aOptions.CreateDetailCCIFile())
			{
			cout << "No files in the specified Writable Certstore" << endl << endl;
			}
		}

	for(int i = size-1; i >= 0; i--)
		{
		string fileName = aOptions.GetDirectoryFiles().at(i);
		store = new StoreFile();
		bool isFileValid = store->CheckFileValidity(fileName.c_str(),aOptions);
		//supposing the file is corrupt,the previous file is considered,else there will be only 1 file in the directory.
		if(isFileValid)
			{
			Initialize(aOptions,*store);
			delete store;
			return true;
			}
		delete store;
		}
	
	return false;
	}

bool DumpSWICertstoreTool::InitializeROMCerstore(const Options& aOptions)
	{
	StoreFile* store = 0;
	store = new StoreFile();
	bool isFileValid = store->CheckFileValidity((aOptions.ROMCertstoreFileName()).c_str(),aOptions);
	if(isFileValid)
		{
		Initialize(aOptions,*store);
		delete store;
		return true;
		}

	delete store;
	return false;
	}

void DumpSWICertstoreTool::Initialize(const Options& aOptions , StoreFile& aStore)
	{
	Stream root = aStore.RootStream();
	DataLocator dataLoc(root, aStore.File());
	StreamReader reader(dataLoc.GetStream());
	reader.Initialise();
	Handle metadataHandle(reader.ReadInt32());

	Stream metadataStream = aStore.FindStream(metadataHandle);
	SwiCertStoreMetaData  metadata(metadataStream, aStore);
	metadata.Read(aOptions);
	iIssuer.append(metadata.GetIssuer());
	iMandatory.append(metadata.GetMandatoryStatusOfAllCertstoreCerts());
	}

ostream& DumpSWICertstoreTool::Error()
	{
	++Tool.iErrors;
	return cerr << "error: ";
	}

ostream& DumpSWICertstoreTool::Warning()
	{
	++Tool.iErrors;
	return cerr << "warning: ";
	}

DumpSWICertstoreTool::~DumpSWICertstoreTool()
	{
	}