/*

* Copyright (c) 2008-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 <e32def.h>

#include <e32def_private.h>

#include <e32rom.h>

#include "h_utl.h"

#include "r_rom.h"

#include "memmap.h"

#include "r_coreimage.h"

#include "r_global.h"

#include "utf16string.h"

#define ROM_PTR(base_ptr) ((TInt8*)iRomHdr + (base_ptr - iRomHdr->iRomBase))

void InflateUnCompress(unsigned char* source, int sourcesize, unsigned char* dest, int destsize);

const TUint KEntryAttDir=0x0010;

const TUint KEntryAttXIP=0x0080;

const TUint KEntryAttHidden=0x0002;

// CoreRomImageReader

// 

CoreRomImageReader::CoreRomImageReader(const char* aFileName, TBool aUseMemMap) : iImgFileName(aFileName), 

iData(0), iLoaderHdr(0), iRomHdr(0), iRootDirList(0), iRootDirectory(0), iUseMemMap(aUseMemMap), iImageMap(0)

{

}

CoreRomImageReader::~CoreRomImageReader() 

{

	if(iData)

	{

		if(iUseMemMap)

		{

			iImageMap->CloseMemoryMap(ETrue);

			delete iImageMap;

		}

		else

			delete iData;

	}

	if(iRootDirectory)

	{

		DeleteAll(iRootDirectory);

	}

}

TBool CoreRomImageReader::IsCoreROM()

{

	if(iData)

	{

		if(iData[0] == 'E' && iData[1] == 'P' && iData[2] == 'O' && iData[3] == 'C' &&

			iData[8]  == 'R' && iData[9]  == 'O' && iData[10] == 'M')

		{

			return ETrue;

		}

	}

	return EFalse;

}

TUint CoreRomImageReader::GetHdrSize()

{ 

	return (sizeof(TRomLoaderHeader) + sizeof(TRomHeader)); 

}

TBool CoreRomImageReader::AllocateImageMap(Memmap*& aImageMap, TUint8*& aData, TUint aLen)

{

	aImageMap = new Memmap(EFalse);

	if(aImageMap == NULL)

	{

		return EFalse;

	}

	else

	{

		aImageMap->SetMaxMapSize(aLen);

		if(aImageMap->CreateMemoryMap() == EFalse)

		{

			aImageMap->CloseMemoryMap(ETrue);

			delete aImageMap;

			aImageMap = NULL;

			return EFalse;

		}

		else

		{

			aData = (TUint8*)aImageMap->GetMemoryMapPointer();

		}

	}

	return ETrue;

}

TBool CoreRomImageReader::OpenImage()

{

	TUint aLen = 0;

	ifstream aIf(iImgFileName.c_str(), ios_base::binary | ios_base::in);

	if( !aIf.is_open() )

	{

		Print(EError, "Cannot open file %s", iImgFileName.c_str());

		return EFalse;

	}

	aIf.seekg(0, ios_base::end);

	aLen = aIf.tellg();

	if(iUseMemMap)

	{

		if(!AllocateImageMap(iImageMap, iData, aLen))

		{

			aLen = 0;

			Print(EError, "Failed to create image map object");

			return EFalse;

		}

	}

	else

	{

		iData = new unsigned char[aLen];

		if(iData == NULL)

		{

			aLen = 0;

			Print(EError, "Out of memory.\n");

			return EFalse;

		}

		memset(iData, 0, aLen);

	}

	aIf.seekg(0, ios_base::beg);

	aIf.read((char*)iData, aLen);

	if(!IsCoreROM() || !StoreImageHeader())

	{

		Print(EError, "Invalid Core ROM image %s", iImgFileName.c_str());

		aIf.close();

		return EFalse;

	}

	aIf.close();

	return ETrue;

}

TBool CoreRomImageReader::StoreImageHeader()

{

	iLoaderHdr = (TRomLoaderHeader*)iData;

	iRomHdr = (TRomHeader*)(iData + sizeof(TRomLoaderHeader));

	if(!iLoaderHdr || !iRomHdr)

		return EFalse;

	return ETrue;

}

TInt CoreRomImageReader::CreateRootDirectory()

{

	iRootDirectory = new TRomNode("", (TRomBuilderEntry*)0);

	if (iRootDirectory == 0 )

		return KErrNoMemory;

	return KErrNone;

}

TBool CoreRomImageReader::ProcessImage()

{

	Memmap *aImageMap = 0;

	if(iRomHdr->iCompressionType == KUidCompressionDeflate)

	{

		TUint aDataStart = GetHdrSize();

		TUint8* aData = 0;

		if(iUseMemMap)

		{

			if(!AllocateImageMap(aImageMap, aData, (iRomHdr->iUncompressedSize + aDataStart)))

			{

				Print(EError, "Failed to create image map object");

				return EFalse;

			}

		}

		else

		{

			aData = new unsigned char[iRomHdr->iUncompressedSize + aDataStart];

		}

		InflateUnCompress((unsigned char*)(iData + aDataStart), iRomHdr->iCompressedSize, (unsigned char*)(aData + aDataStart), iRomHdr->iUncompressedSize);

		memcpy(aData, iData, aDataStart);

		if(iUseMemMap)

		{

			iImageMap->CloseMemoryMap(ETrue);

			delete iImageMap;

			iImageMap = aImageMap;

		}

		else

		{

			delete [] iData;

		}

		iData = aData;

		//update the header fields...

		if(!StoreImageHeader())

		{

			return EFalse;

		}

	}

	else if (iRomHdr && iRomHdr->iRomPageIndex) // paged ROM

	{

		const TInt KPageSize = 0x1000;

		TUint8* aData = 0;

		TRomHeader *pRomHdr = iRomHdr;

		TUint headerSize = GetHdrSize();

		TInt numPages = (pRomHdr->iPageableRomStart + pRomHdr->iPageableRomSize+KPageSize-1)/KPageSize;

		if(iUseMemMap)

		{

			if(!AllocateImageMap(aImageMap, aData, (pRomHdr->iUncompressedSize + headerSize)))

			{

				Print(EError, "Failed to create image map object");

				return EFalse;

			}

		}

		else

		{

			aData = new TUint8[pRomHdr->iUncompressedSize + headerSize];

		}

		TUint8* dest = (aData + sizeof(TRomLoaderHeader) + pRomHdr->iPageableRomStart);

		SRomPageInfo* pi = (SRomPageInfo*)((TUint8*)pRomHdr + pRomHdr->iRomPageIndex);

                CBytePair bpe;

		for(TInt i=0; i<numPages; i++,pi++)

		{

			if (pi->iPagingAttributes != SRomPageInfo::EPageable) // skip uncompressed part at the beginning of ROM image

				continue;

			switch(pi->iCompressionType)

			{

			case SRomPageInfo::ENoCompression:

				{

					memcpy(dest, (TUint8*)pRomHdr + pi->iDataStart, pi->iDataSize);

					dest += pi->iDataSize;

				}

				break;

			case SRomPageInfo::EBytePair:

				{

					TUint8* srcNext=0;

					TInt unpacked = bpe.Decompress((unsigned char*)dest, KPageSize, (TUint8*)pRomHdr + pi->iDataStart, pi->iDataSize, srcNext);

					if (unpacked < 0)

					{

						if(iUseMemMap)

						{

							aImageMap->CloseMemoryMap(ETrue);

							delete aImageMap;

						}

						else

						{

							delete [] aData;

						}

						Print(EError, "Corrupted BytePair compressed ROM image %s", iImgFileName.c_str());

						return EFalse;

					}

					dest += unpacked;

				}

				break;

			default:

				{

					if(iUseMemMap)

					{

						aImageMap->CloseMemoryMap(ETrue);

						delete aImageMap;

					}

					else

					{

						delete [] aData;

					}

					Print(EError, "Undefined compression type in %s", iImgFileName.c_str());

					return EFalse;

				}

			}

		}

		memcpy(aData, iData, sizeof(TRomLoaderHeader) + pRomHdr->iPageableRomStart);

		if(iUseMemMap)

		{

			iImageMap->CloseMemoryMap(ETrue);

			delete iImageMap;

			iImageMap = aImageMap;

		}

		else

		{

			delete [] iData;

		}

		iData = aData;

		//update the header fields...

		if(!StoreImageHeader())

		{

			return EFalse;

		}

	}

	if(CreateRootDirectory() != KErrNone)

	{

		return EFalse;

	}

	if(GetDirectoryStructures() != KErrNone)

	{

		return EFalse;

	}

	return ETrue;

}

TInt CoreRomImageReader::GetDirectoryStructures()

{

	int aDirs = 0;

	TRomDir	*aRomDir = 0;

	iRootDirList = (TRomRootDirectoryList*)ROM_PTR(iRomHdr->iRomRootDirectoryList);

	while( aDirs < iRootDirList->iNumRootDirs )

	{

		aRomDir = (TRomDir*)ROM_PTR(iRootDirList->iRootDir[aDirs].iAddressLin);

		if(BuildDir(aRomDir, iRootDirectory) != KErrNone)

		{

			return KErrNoMemory;

		}

		aDirs++;

	}

	return KErrNone;

}

TInt OffsetCompare(const void *a, const void *b)

{

	return ( *(TInt16*)a - *(TInt16*)b );

}

TInt CoreRomImageReader::BuildDir(TRomDir* aDir, TRomNode* aPaFSEntry)

{

	TInt16			*aFileCnt = 0;

	TInt16			*aBase, aCount;

	TInt16			*aSubDirCnt = 0;

	//Sub directories in this directories

	aSubDirCnt = (TInt16*)((char*)aDir + aDir->iSize + sizeof(aDir->iSize));

	//Files within this directory

	aFileCnt = aSubDirCnt+1;

	aCount = (*aFileCnt + *aSubDirCnt);

	aBase = aFileCnt+1;

	qsort((void*)aBase, aCount, sizeof(TInt16), &OffsetCompare);

	return BuildDir(aBase, aCount, aDir, aPaFSEntry);

}

TInt CoreRomImageReader::BuildDir(TInt16 *aOffsetTbl, TInt16 aOffsetTblCount, 

								  TRomDir *aPaRomDir, TRomNode* aPaFSEntry)

{

	TRomNode		*aNewFSEntry;

	TRomDir			*aNewDir;

	TRomEntry		*aRomEntry;

	TUint32			aOffsetFromBase;

	string		aName;

	//char	*aPtr;

	while( aOffsetTblCount )

	{

		aOffsetFromBase = *aOffsetTbl;

		aOffsetFromBase <<= 2;

		aRomEntry = (TRomEntry*)((char*)aPaRomDir + sizeof(int) + aOffsetFromBase);

		UTF16String unistr(reinterpret_cast<const TUint16*>(aRomEntry->iName),aRomEntry->iNameLength);

		if(!unistr.ToUTF8(aName))

			aName.assign(reinterpret_cast<const char*>(aRomEntry->iName),aRomEntry->iNameLength);

		if( aRomEntry->iAtt & KEntryAttDir )

		{

			// add directory

			aNewFSEntry = aPaFSEntry->NewSubDir(aName.c_str());

			if(aRomEntry->iAtt & KEntryAttHidden)

				aNewFSEntry->iHidden = ETrue;

			else

				aNewFSEntry->iHidden = EFalse;

			aNewDir = (TRomDir*)ROM_PTR(aRomEntry->iAddressLin);

			if(BuildDir(aNewDir, aNewFSEntry) != KErrNone)

			{

				return KErrNoMemory;

			}

		}

		else

		{

			// add file

			if(AddFile(aPaFSEntry, aName.c_str(), aRomEntry) != KErrNone)

			{

				return KErrNoMemory;

			}

		}

		aOffsetTblCount--;

		aOffsetTbl++;

	}

	return KErrNone;

}

TInt CoreRomImageReader::AddFile(TRomNode *aPa, const char* entryName, TRomEntry* aRomEntry)

{

	TRomImageHeader*	aImgHdr = 0;

	TRomBuilderEntry*	aFile = 0;

	TRomNode*			aNewFSEntry = 0;

	static TRomNode*	aLastExecutable = GetRootDirectory();

	TUint8				aUid1[4];

	aImgHdr = (TRomImageHeader*)ROM_PTR(aRomEntry->iAddressLin);

	aFile = new TRomBuilderEntry(0, entryName);

	if(!aFile)

	{

		return KErrNoMemory;

	}

	size_t len = strlen(aFile->iName) + 1;	

	aFile->iBareName = new char[len];

	memcpy(aFile->iBareName,aFile->iName,len);

	aFile->iUid1 = aImgHdr->iUid1;

	aFile->iUid2 = aImgHdr->iUid2;

	aFile->iUid3 = aImgHdr->iUid3;

	aFile->iRomImageFlags = aImgHdr->iFlags;

	aFile->iHardwareVariant = aImgHdr->iHardwareVariant;

	memcpy(aUid1, &(aImgHdr->iUid1), 4);

	aFile->iResource = !IsExecutable(aUid1);

	aNewFSEntry = new TRomNode(entryName, aFile);

	if(!aNewFSEntry)

	{

		return KErrNoMemory;

	}

	if(aRomEntry->iAtt & KEntryAttHidden)

		aNewFSEntry->iHidden = ETrue;

	else

		aNewFSEntry->iHidden = EFalse;

	// RomEntry Update

	aNewFSEntry->iRomFile->SetRomEntry(aRomEntry);

	aNewFSEntry->iRomFile->iFinal = ETrue;

	// E32 Image Pointer Update

	aNewFSEntry->iRomFile->iAddresses.iImageAddr = aRomEntry->iAddressLin;

	aNewFSEntry->iRomFile->iAddresses.iRunAddr = aRomEntry->iAddressLin;

	aNewFSEntry->iRomFile->iAddresses.iSize = aRomEntry->iSize;

	aNewFSEntry->iRomFile->iAddresses.iImagePtr = 0;

	aNewFSEntry->iRomFile->iExportDir.iImagePtr = 0;

	if(!aFile->iResource  && !aFile->HCRDataFile())

	{

		// Hardware Variant Update

		aNewFSEntry->iRomFile->iHwvd = aImgHdr->iHardwareVariant;

		// Security Info Update

		aNewFSEntry->iRomFile->iRbEntry->iS = aImgHdr->iS;