/*

* 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 "dumpswiregistrytool.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 */

class DumpRegistryTool tool;

//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 */

	if(argc < 2)

		{

		Options::DisplayUsage();

		return 0;

		}

	try

		{

		auto_ptr<Options> options(new Options(argc , argv));

		tool.Run(*options);

		}

	catch (Exceptions aErr)

		{

		Options::DisplayError(aErr);

		return aErr;

		}

	catch (bad_alloc&)

		{

		cout << "Error Allocating Memory " << endl ;

		}

	return 0; 

	}

DumpRegistryTool::DumpRegistryTool()

	:iRegistryEntryValid(false) ,

	iFile("\\0") ,

	iFileMajorVersion (0) ,

	iFileMinorVersion (0) ,

	iPkgVendorLocalizedName (0) ,

	iPkgInstallType (0) ,

	iRomBasedPkg (0) ,

	iPkgPreinstalledStatus (0) ,	

	iPkgSignedStatus (0) ,

	iPkgTrust (0) ,	 

	iNotUsed (0) ,

	iTimeStampl (0) ,

	iTimeStamph (0)

	{

	iInstallType[0] = "Installation" ;

	iInstallType[1] = "Augmentation" ;

	iInstallType[2] = "PartialUpgrade" ;

	iInstallType[3] = "PralledApp" ;

	iInstallType[4] = "PralledPatch" ;

	iPackageTrust[0] = "SisPackageUnsignedOrSelfSigned" ;

	iPackageTrust[50] = "SisPackageValidationFailed" ;

	iPackageTrust[100] = "SisPackageCertificateChainNoTrustAnchor" ;

	iPackageTrust[200] = "SisPackageCertificateChainValidatedToTrustAnchor" ;

	iPackageTrust[300] = "SisPackageChainValidatedToTrustAnchorOCSPTransientError" ;

	iPackageTrust[400] = "SisPackageChainValidatedToTrustAnchorAndOCSPValid" ;

	iPackageTrust[500] = "SisPackageBuiltIntoRom" ;

	iValidationStatus[0] = "Unknown" ;

	iValidationStatus[10] = "Expired" ;

	iValidationStatus[20] = "Invalid" ;

	iValidationStatus[30] = "Unsigned" ;

	iValidationStatus[40] = "Validated" ;

	iValidationStatus[50] = "ValidatedToAnchor" ;

	iValidationStatus[60] = "PackageInRom" ;

	iRevocationStatus[0] = "Unknown2" ;

	iRevocationStatus[10] = "OcspNotPerformed" ;

	iRevocationStatus[20] = "OcspRevoked" ;

	iRevocationStatus[30] = "OcspUnknown" ;

	iRevocationStatus[40] = "OcspTransient" ;

	iRevocationStatus[50] = "OcspGood" ;

	}

/**

  Calls the ListFilesInDirectory(),which provides a vector containing all the .reg files present in the 

  package directory specified,and iterates this vector one by one and calls ExtractRegistry() ,to extract the information

  from that particular .reg file.

  */

void DumpRegistryTool::Run(const Options& aOptions)

	{

	std::vector<std::string> regFiles;

	//-p option specified

	if(aOptions.IsPkgUIDSpecified())

		{

		regFiles = OSInterface::ExtractFilesFromDirectory(aOptions.PkgUID(), "reg");

		}

	//-r option specified

	if(aOptions.IsRegistryEntrySpecified())

		{

		regFiles = OSInterface::ExtractFilesFromDirectory(aOptions.RegistryFilePath(), "reg");

		}

	for(vector<string>::const_iterator regIterator = regFiles.begin(); regIterator != regFiles.end(); regIterator++)

		{

		string file = *regIterator;

		const char* fileName = file.c_str();

		bool success = ExtractRegistry(fileName , aOptions);

		if(!success)

			{

			cout << "Package Information:" << endl << endl;

			cout << "Registry Entry :" << fileName <<" IS EMPTY " << endl;

			}

		}

	//if -r option is specified and the registry entry(either pkg name,pkg uid or vendor name is not specified)

	//display usage.

	if(!iRegistryEntryValid && (aOptions.IsRegistryEntrySpecified()))

		{

		cout << "Invalid Registry Entry Specified ." << endl << endl; 

		}

	}

/**

  Extracts the particular registry contents.

  @param registry file

 */

bool DumpRegistryTool::ExtractRegistry(const char* aRegFile , const Options& aOptions)	

	{

	StreamReader reader(aRegFile);

	//check if file is empty

	bool isFileEmpty = reader.Initialise();

	if(isFileEmpty)

		{

		return false;

		}

	RegistryPackage package;

	RegistryToken token;

	token.ExtractRegistryToken(reader,package);

	ExtractPackageInformation(reader);

	unsigned int uid = package.PkgUID();

	char uidBuf[KLength];

	itoa(uid,uidBuf,KRadix);

	//if -p option is specified

	if(aOptions.IsPkgUIDSpecified())

		{

		cout << "Package Information:" << endl << endl;

		cout << "Registry File :" << aRegFile << endl << endl;

		package.DisplayPackage();

		token.DisplayRegistryToken();

		DisplayPackageInformation();

		}

	//if -r option sis specified.

	if(aOptions.IsRegistryEntrySpecified()) 

		{

		int packageUid = CompareUserAndFileSpecifiedData(aOptions.PkgUID(), uidBuf);

		int packageName = CompareUserAndFileSpecifiedData(aOptions.PkgName(), package.PackageName());

		int vendorName = CompareUserAndFileSpecifiedData(aOptions.VendorName(), package.Vendor());

		if(packageUid && packageName && vendorName)

			{

			iRegistryEntryValid = true;

			cout <<"Registry Controller Information:" << endl << endl;

			cout << "Registry Entry :" << aRegFile << endl << endl;

			package.DisplayPackage();

			unsigned int count = token.GetControllerCount();

			for(int i = 0 ; i < count ; i++)

				{

				token.ControllerInformation()->DisplayControllerInfo(i);

				}

			ExtractControllerInformation(reader);

			}

		}

	iFile.close();

	return true;

	}

/**

  Extracts the information about sis dependencies,sis embeds, sis file description,sis properties , 

  validation status and revocation status .

 */

void DumpRegistryTool::ExtractControllerInformation(StreamReader& aReader)

	{

	//list of depenedent SISX packages 

	unsigned int dependencyCount = aReader.ReadInt32(); 

	cout << "Number of Dependencies : " << dependencyCount << endl << endl;

	if(dependencyCount)

		{

		cout << "Dependencies Information : " << endl << endl;

		SISDependency* dependency = new SISDependency[dependencyCount];	

		for(int j = 0; j < dependencyCount; j++)

			{

			dependency[j].ExtractDependency(aReader);

			dependency[j].DisplayDependencyInfo();

			}

		delete [] dependency;

		}

	//list of embedded SISX files

	unsigned int embeddedCount = aReader.ReadInt32(); 

	cout<<"Number of Embedded Packages : " << embeddedCount << endl << endl;

	if(embeddedCount)

		{

		cout << "List Of Embedded  Packages:" << endl << endl;

		SISEmbeds* embed = new SISEmbeds[embeddedCount];	

		for(int i = 0; i < embeddedCount ; i++)

			{

			embed[i].ExtractEmbeds(aReader);

			embed[i].DisplayEmbeddedInfo();

			}

		delete [] embed;

		}

	//properties for the SISX package,is a key, value pair associated with a SISX package.

	unsigned int sisPropertyCount = aReader.ReadInt32();

	cout << "SISProperty Count:"<< sisPropertyCount << endl << endl;

	if(sisPropertyCount)

		{

		cout << "SIS Properties :" << endl << endl;

		SISProperty* property = new SISProperty[sisPropertyCount];

		for(int k = 0; k < sisPropertyCount; k++)

			{

			property[k].ExtractSISProperty(aReader);

			property[k].DisplaySISProperty();

			}

		delete [] property;

		}

	//information about files to be installed .

	unsigned int sisFileDescriptionCount = aReader.ReadInt32();

	cout << "Number Of File Descriptions :" << sisFileDescriptionCount << endl << endl;

	if(sisFileDescriptionCount)

		{

		cout << "SIS File Description:" << endl << endl;

		SISFileDescription* description = new SISFileDescription[sisFileDescriptionCount];	

		for(int l = 0; l < sisFileDescriptionCount; l++)

			{

			description[l].ExtractSISFileDescription(aReader);

			description[l].DisplaySISFileDescription(l);

			}

		delete [] description;

		}

	unsigned int statusOfValidation = aReader.ReadInt32();

	if(tool.iValidationStatus.find(statusOfValidation) !=tool.iValidationStatus.end())

		{

		cout<<"Validation Status : "<< iValidationStatus[statusOfValidation] << endl;

		}

	else

		{

		cout << "Validation Status : " << "Unknown" << endl;

		}

	unsigned int revocationStatus = aReader.ReadInt32();

	if(tool.iRevocationStatus.find(revocationStatus) != tool.iRevocationStatus.end())

		{

		cout << "Revocation Status : " << iRevocationStatus[revocationStatus] << endl;

		}

	else

		{

		cout << "Revocation Status : " << "Unknown" << endl;

		}

	}

//this function is to check whether the user specified values for registry entries matches the one specified in the .reg file.

int DumpRegistryTool::CompareUserAndFileSpecifiedData(const string& aUserSpecified , const string& aFileSpecified)

	{

	string::const_iterator userIter = aUserSpecified.begin();

	string::const_iterator fileIter = aFileSpecified.begin();

	int userSize = aUserSpecified.length();

	int size = aFileSpecified.length();

	//stop when either string end has reached

	while((userIter != aUserSpecified.end()) && (fileIter != aFileSpecified.end()))

		{

		if(::toupper(*userIter) != ::toupper(*fileIter))

			{

			return 0;

			}

		++userIter;

		++fileIter;

		}

	if(userSize == size)

		{

		return 1;

		}

	else

		{

		return 0;

		}

	}

/**

  This function extracts top level information about a package namely install type,whether it is rom based,trust status of package

 */

void DumpRegistryTool::ExtractPackageInformation(StreamReader& aReader)

	{

	iFileMajorVersion = aReader.ReadInt16();

	iFileMinorVersion = aReader.ReadInt16();

	iPkgVendorLocalizedName = aReader.ReadDescriptor();

	iPkgInstallType = aReader.ReadInt32(); 

	iRomBasedPkg = aReader.ReadInt32();				//is package ROM Based

	iPkgPreinstalledStatus = aReader.ReadInt32();	//is package preinstalled

	iPkgSignedStatus = aReader.ReadInt32();			//is package signed   

	iPkgTrust = aReader.ReadInt32();				//trust status of package 

	iNotUsed = aReader.ReadInt32(); 

	iTimeStampl = aReader.ReadInt32();

	iTimeStamph = aReader.ReadInt32();

	}

void DumpRegistryTool::DisplayPackageInformation()

	{

	cout << "File Major Version :" << iFileMajorVersion << endl;

	cout << "File Minor Version :" << iFileMinorVersion << endl;

	cout << "Package Vendor Localized Name :" << iPkgVendorLocalizedName << endl;

	if(tool.iInstallType.find(iPkgInstallType) != tool.iInstallType.end())

		{

		cout << "Package Install Type :" << tool.iInstallType[iPkgInstallType] << endl;

		}

	else

		{

		cout << "Package Install Type :" << "Unknown" << endl;

		}

	if(iRomBasedPkg)

		{

		cout << "Is Package in ROM :" << "Yes" << endl;

		}

	else

		{

		cout<<"Is Package in ROM :"<< "No" <<endl;

		}

	if(iPkgPreinstalledStatus)

		{

		cout << "Package Preinstalled :" << "Yes" << endl;

		}

	else

		{

		cout << "Package Preinstalled :" << "No" << endl;

		}

	if(iPkgSignedStatus)

		{

		cout << "Package Signed :" << "Yes" << endl;

		}

	else

		{

		cout << "Is Package Signed :" << "No" << endl;

		}

	if(tool.iPackageTrust.find(iPkgTrust) !=tool.iPackageTrust.end())

		{

		cout << "Package Trust :" << tool.iPackageTrust[iPkgTrust] << endl;

		}

	else

		{

		cout << "Package Trust :" << "Unknown" << endl << endl;

		}

	}

DumpRegistryTool::~DumpRegistryTool()

	{

	delete [] iPkgVendorLocalizedName;

	}