/*

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

*

*/

#pragma warning (disable: 4786)

#ifdef SYMBIAN_UNIVERSAL_INSTALL_FRAMEWORK

#include "dbhelper.h"

#include "dblayer.h"

#include "dbconstants.h"

#include "is_utils.h"

#include "xmlgenerator.h"

#include "xmlparser.h"

#include "util.h"

#include <algorithm>

#include <iostream>

#include <stdexcept>

#include <iostream>

#endif //SYMBIAN_UNIVERSAL_INSTALL_FRAMEWORK

// constants 

#define IGNORE_FORMATTING -1

#define ENABLE 1

#define DISABLE 0

// System Includes

#include <list>

#include <fstream>

#include <algorithm>

#include <sys/stat.h>

#include <sstream>

#ifndef __TOOLS2_LINUX__

#include <windows.h>

#endif 

// SisX Library Includes

#include "sisfiledescription.h"

#include "sisstring.h"

// User Includes

#include "sisregistry.h"

#include "errors.h"

#include "rommanager.h"

#include "stringutils.h"

#include "is_utils.h"

#include "sisfile.h"

#include "sisregistrypackage.h"

#include "sisregistryobject.h"

#include "configmanager.h"

// Constants

const std::wstring SisRegistry::KPathToRegistry(L"\\sys\\install\\sisregistry/");

const std::wstring SisRegistry::KPathToRomStubs(L"\\system\\install\\");

const std::wstring SisRegistry::KPathToRegistryVersionMarker(L"\\system\\data\\");

const std::wstring KRegistryEntry(L"00000000.reg");

const std::wstring KControllerEntry(L"00000000_0000.ctl");

const std::wstring KBackupLst(L"backup.lst");

const int SisRegistry::KSisRegistryMajorVersion = 5;

const int SisRegistry::KSisRegistryMinorVersion = 1;

// Supported SisRegistry versions

const std::wstring SisRegistry::KRegistryFormat(L"sisregistry_?.?.txt");

const std::wstring SisRegistry::KRegistryV40(L"sisregistry_4.0.txt");

const std::wstring SisRegistry::KRegistryV51(L"sisregistry_5.1.txt");

const std::wstring SisRegistry::KRegistryV52(L"sisregistry_5.2.txt");

const std::wstring SisRegistry::KRegistryV53(L"sisregistry_5.3.txt");

const std::wstring SisRegistry::KRegistryV54(L"sisregistry_5.4.txt");

const std::wstring SisRegistry::KRomStubDir(L"z:\\system\\install\\");

const std::wstring SisRegistry::KRomRegistryVersionMarker(L"z:\\system\\data\\");

const std::string SisRegistry::KRegistryV40string("4.0");

const std::string SisRegistry::KRegistryV50string("5.0");

const std::string SisRegistry::KRegistryV51string("5.1");

const std::string SisRegistry::KRegistryV52string("5.2");

const std::string SisRegistry::KRegistryV53string("5.3");

const std::string SisRegistry::KRegistryV54string("5.4");

#ifdef SYMBIAN_UNIVERSAL_INSTALL_FRAMEWORK

const std::wstring SisRegistry::NativeSoftwareType(L"native");

#ifdef _MSC_VER

// "." and ".ctl"

// VC does not have ".." entry

const int KDirFileCount = 2;

#else

// "." ".." and ".ctl"

const int KDirFileCount = 3;

#endif

#endif //SYMBIAN_UNIVERSAL_INSTALL_FRAMEWORK

// KRegDirFileCount is the number of directory entries present

// in the reg directory when there is no upgrade is made to the

// package. i.e. one .reg and one .ctl file

//

#ifdef _MSC_VER

// "." ".reg" and ".ctl"

// VC does not have ".." entry

const int KRegDirFileCount = 3;

#else

// "." ".." ".reg" and ".ctl"

const int KRegDirFileCount = 4;

#endif

// function object for determining if all characters are valid hex chars

struct AllHexDigits

    {

	bool result;

	AllHexDigits() : result(true) {}

	void operator()(char a)

	    {

		if (!((a >= '0' && a <= '9') ||

			(a >= 'a' && a <= 'f') ||

			(a >= 'A' && a <= 'F')))

		    {

			result = false;

		    }

	    }

    };

// function object for retreiving keys from the sisregistry entry map

struct CopyUids

    {

	CopyUids(std::vector<TUint32>& uids) : iUids(uids) {}

	std::vector<TUint32>& iUids;

    void operator()(SisRegistry::EntryMap::value_type& a)

	{ iUids.push_back(a.first); }

    };

struct FindFile

    {

	FindFile(const std::wstring& aFile) : iFile(aFile), iFileDescr(0) {}

	bool operator()(const SisRegistry::EntryMap::value_type&  val)

	    {

	    const SisRegistryObject& obj = *val.second;

	    return (iFileDescr = obj.GetFile(iFile)) != 0;

	    }

    std::wstring iFile;

	const FileDescription* iFileDescr;

    };

struct FindFile2

    {

	FindFile2(TUint32 aSid, const FileDescription*& aFd) :

		iSid(aSid), iFileDescr(aFd) {}

	bool operator()(const SisRegistry::EntryMap::value_type& val)

	    {

		const SisRegistryObject& obj = *val.second;

		if (iFileDescr = obj.GetFile(iSid))

		    {

			return true;

		    }

		return false;

	    }

	TUint32 iSid;

	const FileDescription*& iFileDescr;

    };

struct Contains

    {

	const std::wstring& iFile;

	Contains(const std::wstring& aFilename) : iFile(aFilename)

	{}

	SisRegistryObject* operator()(const SisRegistry::EntryMap::value_type& aObj)

	    {

		if (aObj.second->GetFile(iFile) != 0)

			{

			return aObj.second;

			}

    	}

    };

SisRegistry::SisRegistry( CParameterList& aOptions, RomManager& aRomManager, ConfigManager& aConfigManager)

:   iParamList( aOptions ), iRomManager( aRomManager ), iConfigManager( aConfigManager )

    {

	// Determine which version of the SisRegistry to generate

	if (iParamList.RegistryVersionExists() == false)

		{

		ReadSisRegistryVersion();

		}

	if(iParamList.RegistryVersionExists())

		{

		LINFO(L"Using SIS Registry v" << iParamList.RegistryMajorVersion() << L"." << iParamList.RegistryMinorVersion());

		std::wstring registryPathDriveC = KPathToRegistry;

		ConvertToLocalPath( registryPathDriveC, iParamList.SystemDrivePath() );

		ReadRegistry( registryPathDriveC );

		}

	#ifdef SYMBIAN_UNIVERSAL_INSTALL_FRAMEWORK

	// this code is for handling the database registry implementation

	else

		{

		LINFO(L"Using database implementation of SIS Registry");

		int val = iParamList.IsFlagSet(CParameterList::EFlagsDisableZDriveChecksSet);

		if	( 

			(iParamList.RomLogFileNames().size() != 0 && iParamList.RomDrivePath().size() == 0 ) ||

			iParamList.IsFlagSet( CParameterList::EFlagsDisableZDriveChecksSet) 

			)

			{

			iDbHelper = new DbHelper(iParamList.SystemDrivePath(), true);	

			}

		else

			{

			iDbHelper = new DbHelper(iParamList.RomDrivePath(), false);			

			}		

		}

	#endif // SYMBIAN_UNIVERSAL_INSTALL_FRAMEWORK

	GenerateStubRegistry();

    }

SisRegistry::~SisRegistry()

    {

	if(iParamList.RegistryVersionExists())

		{

		EntryMap::const_iterator end = iEntries.end();

		for (EntryMap::const_iterator curr = iEntries.begin() ;

			curr != end; ++curr)

			{

			delete 	curr->second;

			}

		}

	#ifdef SYMBIAN_UNIVERSAL_INSTALL_FRAMEWORK

	else

		{

		delete iDbHelper;

		}

	#endif //SYMBIAN_UNIVERSAL_INSTALL_FRAMEWORK

    }

/**

 * If the ROM drive has been defined, find the marker file in the stub directory 

 * e.g. z:\system\data\. If using the ROM/ROFS logs, find the marker file in the

 * logs. If not found, then check to see if the marker file exists in the ROM stub directory

 * specified using the -t option. If still not found, use the default version of the Sis Registry.

 */

void SisRegistry::ReadSisRegistryVersion()

	{

	// ROM drive defined

	if (iParamList.IsFlagSet(CParameterList::EFlagsZDriveSet)) 

		{

		// ROM SisRegistry marker file

		std::wstring stubPathZ = KPathToRegistryVersionMarker;

		ConvertToLocalPath( stubPathZ, iParamList.RomDrivePath() );

		FindMarkerAndSetVersion(stubPathZ);

		}

	// ROM/ROFS logs defined

	else if (iParamList.IsFlagSet(CParameterList::EFlagsRomRofsLogFilesSet))

		{

			// Check the stub dir

			if (iParamList.IsFlagSet(CParameterList::EFlagsStubDirectorySet))

				{

				// ROM SisRegistry marker file

				std::wstring customStubPath = iParamList.SisStubPath();

				FindMarkerAndSetVersion(customStubPath);

				}

			// Search in the ROM/ROFS logs. This takes precedence over the stub dir

			if(iRomManager.RomFileExists(KPathToRegistryVersionMarker+KRegistryV40))

			{

				iParamList.SetRegistryMajorVersion(4);

				iParamList.SetRegistryMinorVersion(0);

			}

			else if(iRomManager.RomFileExists(KRomRegistryVersionMarker+KRegistryV51))

			{

				iParamList.SetRegistryMajorVersion(5);

				iParamList.SetRegistryMinorVersion(1);

			}

			else if(iRomManager.RomFileExists(KRomRegistryVersionMarker+KRegistryV52))

			{

				iParamList.SetRegistryMajorVersion(5);

				iParamList.SetRegistryMinorVersion(2);

			}

			else if(iRomManager.RomFileExists(KRomRegistryVersionMarker+KRegistryV53))

			{

				iParamList.SetRegistryMajorVersion(5);

				iParamList.SetRegistryMinorVersion(3);

			}

			else if(iRomManager.RomFileExists(KRomRegistryVersionMarker+KRegistryV54))

			{

				iParamList.SetRegistryMajorVersion(5);

				iParamList.SetRegistryMinorVersion(4);

			}

		}

	#ifndef SYMBIAN_UNIVERSAL_INSTALL_FRAMEWORK

	// No valid ROM stub marker (sisregistry_x.x.txt) found!

	// Revert to the default (5.1) SisRegistry version as defined 

	// by KSisRegistryMajorVersion and KSisRegistryMinorVersion

	if (iParamList.RegistryMajorVersion() == 0 && iParamList.RegistryMinorVersion() == 0)

		{

		iParamList.SetRegistryMajorVersion(KSisRegistryMajorVersion);

		iParamList.SetRegistryMinorVersion(KSisRegistryMinorVersion);

		}

	#endif // SYMBIAN_UNIVERSAL_INSTALL_FRAMEWORK

	}

void SisRegistry::FindMarkerAndSetVersion(const std::wstring& aPath)

	{

	// Make sure the system\data directory name is properly terminated.

	std::wstring basePath = StringUtils::EnsureDirectoryTerminated( aPath );

	std::list<std::wstring> systemDir;

	GetDirContents( basePath, systemDir );

	// Loop on the system\data files until we find a SisRegistry marker file

	std::list<std::wstring>::iterator curr = systemDir.begin();

	while (curr != systemDir.end())

		{

		std::wstring fileFound ( *curr );

		// Found the SisRegistry marker file

		if (fileFound.compare(KRegistryV40) == 0)

			{

			iParamList.SetRegistryMajorVersion(4);

			iParamList.SetRegistryMinorVersion(0);

			}

		else if (fileFound.compare(KRegistryV51) == 0) // introduced isRemovable flag

			{

			iParamList.SetRegistryMajorVersion(5);

			iParamList.SetRegistryMinorVersion(1);

			}

		else if (fileFound.compare(KRegistryV52) == 0) // fixed DEF114705 - SIS controller incorrect

			{

			iParamList.SetRegistryMajorVersion(5);

			iParamList.SetRegistryMinorVersion(2);

			}

		else if (fileFound.compare(KRegistryV53) == 0) // introduced SU cert flag

			{

			iParamList.SetRegistryMajorVersion(5);

			iParamList.SetRegistryMinorVersion(3);

			}

		else if (fileFound.compare(KRegistryV54) == 0) // Support for exact & equivalent device supported languages

			{

			iParamList.SetRegistryMajorVersion(5);

			iParamList.SetRegistryMinorVersion(4);

			}

			++curr;

		}

	}

bool SisRegistry::IsPackageDir(const std::wstring& aString)

    {

	AllHexDigits digits;

	std::for_each(aString.begin(), aString.end(), digits);

	return aString.length() == 8 && digits.result;

    }

TBool SisRegistry::IsInstalled(TUint32 aUid) const

    {

	if(iParamList.RegistryVersionExists())

		{

		return iEntries.find(aUid) != iEntries.end();;

		}

	#ifdef SYMBIAN_UNIVERSAL_INSTALL_FRAMEWORK

	else

		{

		return iDbHelper->IsInstalled(aUid);

		}

	#endif //SYMBIAN_UNIVERSAL_INSTALL_FRAMEWORK

	return false;

    }

const SisRegistryObject& SisRegistry::GetRegistryObject(TUint32 aUid) const

    {

	if(iParamList.RegistryVersionExists())

		{

		EntryMap::const_iterator end = iEntries.end();

		for (EntryMap::const_iterator curr = iEntries.begin() ;

			 curr != end; ++curr)

			{

			if ((curr->first == aUid) && ((*curr->second).GetIndex() == 0))

				{

				return *curr->second;

				}

			}

		}

	#ifdef SYMBIAN_UNIVERSAL_INSTALL_FRAMEWORK

	else

		{

		SisRegistryObject* sisRegistryObject = new SisRegistryObject(iConfigManager);

		if(iDbHelper->InternalizeIfExists(*sisRegistryObject, aUid))

			{

			return *sisRegistryObject;

			}

		}

	#endif //SYMBIAN_UNIVERSAL_INSTALL_FRAMEWORK

	throw std::runtime_error("Couldn't find specified package");

    }

SisRegistryObject* SisRegistry::GetRegistryObjectPtr(TUint32 aUid)

    {

	if(iParamList.RegistryVersionExists())

		{

		EntryMap::const_iterator end = iEntries.end();

		for (EntryMap::const_iterator curr = iEntries.begin() ;

			 curr != end; ++curr)

			{

			if ((curr->first == aUid) && ((curr->second)->GetIndex() == 0))

				{

				return curr->second;

				}

			}

		}

	#ifdef SYMBIAN_UNIVERSAL_INSTALL_FRAMEWORK

	else

		{

		SisRegistryObject* sisRegistryObject = new SisRegistryObject(iConfigManager);

		if(iDbHelper->InternalizeIfExists(*sisRegistryObject, aUid))

			{

			return sisRegistryObject;

			}

		}

	#endif //SYMBIAN_UNIVERSAL_INSTALL_FRAMEWORK

	throw std::runtime_error("Couldn't find specified package");

    }

const SisRegistryPackage& SisRegistry::SidToPackage(TUint32 aSid)

    {

	const FileDescription* fileDesc = 0;

	FindFile2 fileFinder(aSid, fileDesc);

  	EntryMap::iterator x = std::find_if(iEntries.begin(),

										iEntries.end(),

										fileFinder);

	if (x != iEntries.end())

	    {

		return *x->second;

	    }

	else

	    {

		throw std::runtime_error("Couldn't find specified file");

	    }

    }

void SisRegistry::SidToFileName(TUint32 aSid, std::wstring& aFileName)

    {

	if(iParamList.RegistryVersionExists())

		{

		const FileDescription* fileDesc = 0;

		FindFile2 fileFinder(aSid, fileDesc);

  		EntryMap::iterator x = std::find_if(iEntries.begin(),

											iEntries.end(),

											fileFinder);

		if (x != iEntries.end())

			{

			aFileName = fileFinder.iFileDescr->GetTarget();

			}

		else

			{

			throw std::runtime_error("Couldn't find specified file");

			}

		}

	#ifdef SYMBIAN_UNIVERSAL_INSTALL_FRAMEWORK

	else

		{

		aFileName = iDbHelper->SidToFileName(aSid);	

		if(aFileName.empty())

			{

			throw std::runtime_error("Couldn't find specified file");

			}

		}

	#endif //SYMBIAN_UNIVERSAL_INSTALL_FRAMEWORK

    }