/*

* Copyright (c) 2009 Nokia Corporation and/or its subsidiary(-ies).

* All rights reserved.

*

* This program is free software: you can redistribute it and/or modify

* it under the terms of the GNU Lesser General Public License as published by

* the Free Software Foundation, either version 3 of the License, or

* (at your option) any later version.

*

* This program is distributed in the hope that it will be useful,

* but WITHOUT ANY WARRANTY; without even the implied warranty of

* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

* GNU Lesser General Public License for more details.

* 

* You should have received a copy of the GNU Lesser General Public License

* along with this program.  If not, see <http://www.gnu.org/licenses/>.

*/

#ifndef ELFSECTION_H_

#define ELFSECTION_H_

#include <libelf.h>

#include "defs.h" // for String

#include "filefragment.h"

#include "inputfile.h"

class ElfSectionHeader {

public:

	ElfSectionHeader()

		{}

	ElfSectionHeader(Elf32_Shdr & aShdr) :

		iElf32Shdr(aShdr)

		{}

	Elf32_Shdr iElf32Shdr;

};

class ElfSectionData  : public FileFragmentOwner {

public:

	ElfSectionData()

		{}

	ElfSectionData(size_t aOffset):

		FileFragmentOwner(aOffset)

		{}

	virtual ElfSectionData * Clone() = 0;

	// The FileFragmentOwner protocol

	virtual void GetFileFragmentData(FileFragmentData & aFileFragmentData ) = 0;

	virtual size_t Size() = 0;

	virtual void DeleteFileFragmentData() = 0;

};

class ElfSectionRomData : public ElfSectionData {

public:

	ElfSectionRomData(size_t aOffset, size_t aSize) :

		ElfSectionData(aOffset), 

		iSize(aSize)

		{}

	ElfSectionRomData(const ElfSectionRomData & aData);

	// ElfSection protocol

	virtual ElfSectionRomData * Clone();

	// The FileFragmentOwner protocol

	virtual void GetFileFragmentData(FileFragmentData & aFileFragmentData ){}

	virtual size_t Size(){ return iSize; }

	// Nothing to delete

	virtual void DeleteFileFragmentData(){};

	// Dont add data;

	virtual void AddData(OutputFile & aOutputFile) {};

private:

	size_t iSize;

};

class ElfSectionElfData : public ElfSectionData {

public:

	ElfSectionElfData(FileFragmentOwner & aSource) :

		iSource(aSource)

		{}

	ElfSectionElfData(const ElfSectionElfData & aData);

	// ElfSection protocol

	virtual ElfSectionElfData * Clone();

	// The FileFragmentOwner protocol

	virtual void GetFileFragmentData(FileFragmentData & aFileFragmentData );

	virtual size_t Size();

	virtual void DeleteFileFragmentData();

	virtual void AddData(OutputFile & aOutputFile);

	virtual size_t GetOffset();

private:

	FileFragmentOwner & iSource;

};

class ElfSectionFileData : public ElfSectionData {

public:

	ElfSectionFileData(InputFile * aInputFile) :

		iInputFile(aInputFile), iData(NULL)

		{}

	ElfSectionFileData(const ElfSectionFileData & aData);

	// ElfSection protocol

	virtual ElfSectionFileData * Clone();

	// The FileFragmentOwner protocol

	virtual void GetFileFragmentData(FileFragmentData & aFileFragmentData );

	virtual size_t Size();

	virtual void DeleteFileFragmentData();

private:

	InputFile * iInputFile;

	char * iData;

};

class ElfSectionNoData : public ElfSectionData {

public:

	ElfSectionNoData(){}

	ElfSectionNoData(const ElfSectionNoData & aData);

	// ElfSection protocol

	virtual ElfSectionNoData * Clone();

	// The FileFragmentOwner protocol

	virtual void GetFileFragmentData(FileFragmentData & aFileFragmentData );

	virtual size_t Size() { return 0; }

	// Nothing to delete

	virtual void DeleteFileFragmentData(){};

	// Dont add data;

	virtual void AddData(OutputFile & aOutputFile) {};

private:

	InputFile * iInputFile;

};

class ElfSection {

public:

	ElfSection(ElfSectionData *  aData) :

		iSectionName(""), iSectionData(aData), iIndex(0)

		{}

	ElfSection(ElfSectionData *  aData, String aName, Elf32_Shdr & aShdr) :

		iSectionName(aName), iSectionHdr(aShdr), iSectionData(aData)

		{}

	ElfSection(const ElfSection & aData);

	ElfSection & operator=(const ElfSection & aSection);

	virtual ~ElfSection();

	String & GetName() { return iSectionName ; }

	void SetNameOffset(size_t nameOffset) { iSectionHdr.iElf32Shdr.sh_name = nameOffset; }

	ElfSectionHeader & GetSectionHeader() { return iSectionHdr; }

	void SetSize(size_t aSize) { iSectionHdr.iElf32Shdr.sh_size = aSize; }

	void SetOffset(size_t aOffset) { iSectionHdr.iElf32Shdr.sh_offset = aOffset; }

	virtual void AddData(OutputFile & aOutputFile);

	unsigned int GetIndex() { return iIndex; };

	void SetIndex(unsigned int aIndex) { iIndex = aIndex; };

private:

	String iSectionName;

	ElfSectionHeader iSectionHdr;

	ElfSectionData * iSectionData;

	unsigned int	iIndex;

};

#endif /*ELFSECTION_H_*/