/*

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

*/

#include <string.h>

#include "elfsectionmanager.h"

void ElfSectionManager::GetFileFragmentData(FileFragmentData & aFileFragmentData ){

	int nSections = iSections.size();

	if (nSections > 0){

		iData = new Elf32_Shdr[nSections];

		Elf32_Shdr * p = iData;

		SectionList::iterator aSection = iSections.begin();

		SectionList::iterator end = iSections.end();

		while (aSection != end) {

			*p = aSection->GetSectionHeader().iElf32Shdr;

			p++;

			aSection++;

		}

		SetFileFragmentData(aFileFragmentData, Size(), reinterpret_cast<char *>(iData));

	}

}

size_t ElfSectionManager::Size(){

	return SectionHeaderSize();

}

void ElfSectionManager::DeleteFileFragmentData(){

	if (iData) {

		Elf32_Shdr * d = iData;

		iData = NULL;

		delete[] d;

	}

}

void ElfSectionManager::AddData(){

	AddData(iOutputFile);

}

void ElfSectionManager::AddData(OutputFile & aOutputFile){

	if (iSections.size() > 0) {

		SectionList::iterator aSection = iSections.begin();

		SectionList::iterator end = iSections.end();

		while (aSection != end) {

			aSection->AddData(aOutputFile);

			aSection++;

		}

		AddSectionTable();

	}

}

void ElfSectionManager::EnsureSectionStringTableSectionAdded(){

	if (iSectionStringTableSectionAdded) return;

	// set iSectionStringTableSectionAdded true now so we don't do the next bit twice;

	iSectionStringTableSectionAdded = true;

	// first create the UNDEF section

	ElfSectionNoData * aUndefData = new ElfSectionNoData();

	Elf32_Shdr undef;

	undef.sh_name = 0; 

	undef.sh_type = SHT_NULL;

	undef.sh_flags = 0;

	undef.sh_addr = 0;

	undef.sh_offset = 0; 

	undef.sh_size = 0; 

	undef.sh_link = 0;

	undef.sh_info = 0;

	undef.sh_addralign = 0;

	undef.sh_entsize = 0;

	ElfSection aUndefSection(aUndefData, "", undef);

	AddSection(aUndefSection);

	ElfSectionElfData * aSectionStringTableSectionData = new ElfSectionElfData(iStringTable);

	Elf32_Shdr shdr;

	shdr.sh_name = 0; // for now.

	shdr.sh_type = SHT_STRTAB;

	shdr.sh_flags = 0;

	shdr.sh_addr = 0;

	shdr.sh_offset = 0; // for now

	shdr.sh_size = 0; // for now.

	shdr.sh_link = SHN_UNDEF;

	shdr.sh_info = 0;

	shdr.sh_addralign = 0;

	shdr.sh_entsize = 0;

	ElfSection aSectionStringTableSection(aSectionStringTableSectionData, ".shstrtab", shdr);

	AddSection(aSectionStringTableSection);

	iElf32Header.SetSectionStringNdx(iSections.size() - 1);

}

void ElfSectionManager::AddSection(ElfSection & aSection){

	EnsureSectionStringTableSectionAdded();

	if (aSection.GetName().size() > 0){

		// rename sections for GDB (yuk!)

		String sectionName(aSection.GetName());

		String ro("ER_RO");

		String text(".text");

		if (sectionName == ro){

			sectionName = text;

		} else {

			String rw("ER_RW");

			String data(".data");

			if (sectionName == rw){

				sectionName = data;

			} else {

				String zi("ER_ZI");

				String bss(".bss");

				if (sectionName == zi)

					sectionName = bss;

			}

		}

		size_t nameOffset = iStringTable.AddName(sectionName);

		aSection.SetNameOffset(nameOffset);

	} else {

		// use the initial Null String.

		size_t nameOffset = iStringTable.AllocateInitialNullString();

		aSection.SetNameOffset(nameOffset);

	}

	aSection.SetIndex(iSections.size());

	iSections.push_back(aSection);

	iElf32Header.AddSectionHdr();

}

size_t ElfSectionManager::SectionHeaderSize(){

	return iSections.size() * sizeof(Elf32_Shdr);

}

void ElfSectionManager::AddSectionStringTable(){

	// Assume the section header already setup and we've got hold of it. We need to say where the strings ended up.

	const FileFragment & aSectionTableFrag = iOutputFile.GetFileFragment(this);	

	SetOffset(aSectionTableFrag.GetOffset());

}

void ElfSectionManager::AddSectionTable(){

	const FileFragment & aSectionTableFrag = iOutputFile.GetFileFragment(this);	

	SetOffset(aSectionTableFrag.GetOffset());

	iElf32Header.SetSectionHdrOffset(GetOffset());

}