// Copyright (c) 1995-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:

// e32\memmodel\epoc\multiple\mcodeseg.cpp

// 

//

#include "memmodel.h"

#include <mmubase.inl>

#include "cache_maintenance.h"

#include <demand_paging.h>

DCodeSeg* M::NewCodeSeg(TCodeSegCreateInfo&)

//

// Create a new instance of this class.

//

	{

	__KTRACE_OPT(KDLL,Kern::Printf("M::NewCodeSeg"));

	return new DMemModelCodeSeg;

	}

//

// DMemModelCodeSegMemory

//

DEpocCodeSegMemory* DEpocCodeSegMemory::New(DEpocCodeSeg* aCodeSeg)

	{

	return new DMemModelCodeSegMemory(aCodeSeg);

	}

DMemModelCodeSegMemory::DMemModelCodeSegMemory(DEpocCodeSeg* aCodeSeg)

	: DMmuCodeSegMemory(aCodeSeg)

	{

	}

TInt DMemModelCodeSegMemory::Create(TCodeSegCreateInfo& aInfo)

	{

	TInt r = DMmuCodeSegMemory::Create(aInfo);

	if(r!=KErrNone)

		return r;

	Mmu& m=Mmu::Get();

	iOsAsids = TBitMapAllocator::New(m.iNumOsAsids, EFalse);

	if(!iOsAsids)

		return KErrNoMemory;

	TInt totalPages = iPageCount+iDataPageCount;

	iPages = (TPhysAddr*)Kern::Alloc(totalPages*sizeof(TPhysAddr));

	if(!iPages)

		return KErrNoMemory;

	TInt i;

	for (i=0; i<totalPages; ++i)

		iPages[i] = KPhysAddrInvalid;

	MmuBase::Wait();

	// allocate RAM pages...

	__KTRACE_OPT(KDLL,Kern::Printf("Alloc DLL pages %x,%x", iPageCount,iDataPageCount));

	TInt startPage = iIsDemandPaged ? iPageCount : 0; 	// if demand paged, skip pages for code

	TInt endPage = iPageCount+iDataPageCount;

	r=m.AllocRamPages(iPages+startPage, endPage-startPage, EPageMovable);

	// initialise SPageInfo objects for allocated pages...

	if (r==KErrNone)

		{

		NKern::LockSystem();

		for (i=startPage; i<endPage; ++i)

			{

			SPageInfo* info = SPageInfo::FromPhysAddr(iPages[i]);

			info->SetCodeSegMemory(this,i);

			if((i&15)==15)

				NKern::FlashSystem();

			}

		NKern::UnlockSystem();

		}

	MmuBase::Signal();

	if (r!=KErrNone)

		return r;

#ifdef BTRACE_CODESEGS

	BTrace8(BTrace::ECodeSegs,BTrace::ECodeSegMemoryAllocated,iCodeSeg,iPageCount<<m.iPageShift);

#endif

	DCodeSeg::Wait();

	TInt code_alloc=((totalPages<<m.iPageShift)+m.iAliasMask)>>m.iAliasShift;

	r=MM::UserCodeAllocator->AllocConsecutive(code_alloc, ETrue);

	if (r<0)

		r = KErrNoMemory;

	else

		{

		MM::UserCodeAllocator->Alloc(r, code_alloc);

		iCodeAllocBase=r;

		iRamInfo.iCodeRunAddr=m.iUserCodeBase+(r<<m.iAliasShift);

		iRamInfo.iCodeLoadAddr=iRamInfo.iCodeRunAddr;

		if (iRamInfo.iDataSize)

			{

			if(iDataPageCount)

				iRamInfo.iDataLoadAddr=iRamInfo.iCodeLoadAddr+Mmu::RoundToPageSize(iRamInfo.iCodeSize);

			else

				iRamInfo.iDataLoadAddr=iRamInfo.iCodeLoadAddr+iRamInfo.iCodeSize;

			}

		DMemModelProcess* pP=(DMemModelProcess*)TheCurrentThread->iOwningProcess;

		r=pP->MapUserRamCode(this, ETrue);

		if (r==KErrNone)

			iCreator=pP;

		}

	DCodeSeg::Signal();

	return r;

	}

void DMemModelCodeSegMemory::Substitute(TInt aOffset, TPhysAddr aOld, TPhysAddr aNew)

	{

	__KTRACE_OPT(KMMU,Kern::Printf("DMemModelCodeSegMemory::Substitute %x %08x %08x",aOffset,aOld,aNew));

	Mmu& m=Mmu::Get();

	if (iPages[aOffset>>KPageShift] != aOld)

		MM::Panic(MM::ECodeSegRemapWrongPage);

	iPages[aOffset>>KPageShift] = aNew;

	m.RemapPageByAsid(iOsAsids, iRamInfo.iCodeRunAddr+aOffset, aOld, aNew, m.PtePermissions(EUserCode));

	}

TInt DMemModelCodeSegMemory::Loaded(TCodeSegCreateInfo& aInfo)

	{

	__NK_ASSERT_DEBUG(iPages);

	TInt r = DMmuCodeSegMemory::Loaded(aInfo);

	if(r!=KErrNone)

		return r;

	Mmu& m=Mmu::Get();

	if(!iIsDemandPaged)

		{

		UNLOCK_USER_MEMORY();

		CacheMaintenance::CodeChanged(iRamInfo.iCodeLoadAddr, iRamInfo.iCodeSize);

		LOCK_USER_MEMORY();

		}

	else

		{

		// apply code fixups to pages which have already been loaded...

		TInt pageShift = m.iPageShift;

		for (TInt i = 0 ; i < iPageCount ; ++i)

			{

			if (iPages[i] != KPhysAddrInvalid)

				{

				r = ApplyCodeFixupsOnLoad((TUint32*)(iRamInfo.iCodeLoadAddr+(i<<pageShift)),iRamInfo.iCodeRunAddr+(i<<pageShift));

				if(r!=KErrNone)

					return r;

				}

			}

		// copy export directory (this will now have fixups applied)...

		TInt exportDirSize = iRamInfo.iExportDirCount * sizeof(TLinAddr);

		if (exportDirSize > 0 || (exportDirSize==0 && (iCodeSeg->iAttr&ECodeSegAttNmdExpData)) )

			{

			exportDirSize += sizeof(TLinAddr);

			TLinAddr* expDir = (TLinAddr*)Kern::Alloc(exportDirSize);

			if (!expDir)

				return KErrNoMemory;

			iCopyOfExportDir = expDir;

			UNLOCK_USER_MEMORY();

			memcpy(expDir,(TAny*)(iRamInfo.iExportDir-sizeof(TLinAddr)),exportDirSize);

			LOCK_USER_MEMORY();

			}

		}

	// unmap code from loading process...

	DMemModelProcess* pP=(DMemModelProcess*)TheCurrentThread->iOwningProcess;

	__ASSERT_ALWAYS(iCreator==pP, MM::Panic(MM::ECodeSegLoadedNotCreator));

	pP->UnmapUserRamCode(this, ETrue);

	iCreator=NULL;

	// discard any temporary pages used to store loaded data section...

	if(iDataPageCount)

		{

		MmuBase::Wait();

		TPhysAddr* pages = iPages+iPageCount;

		m.FreePages(pages,iDataPageCount, EPageMovable);

		for (TInt i = 0 ; i < iDataPageCount ; ++i)

			pages[i] = KPhysAddrInvalid;

		MmuBase::Signal();

		// see if we can free any virtual address space now we don't need any for loading data

		TInt data_start = ((iPageCount << m.iPageShift) + m.iAliasMask) >> m.iAliasShift;

		TInt data_end = (((iPageCount + iDataPageCount) << m.iPageShift) + m.iAliasMask) >> m.iAliasShift;

		if (data_end != data_start)

			{

			DCodeSeg::Wait();

			MM::UserCodeAllocator->Free(iCodeAllocBase + data_start, data_end - data_start);

			DCodeSeg::Signal();

			}

		iDataPageCount = 0;

		//Reduce the size of the DCodeSeg now the data section has been moved 

		iCodeSeg->iSize = iPageCount << m.iPageShift;

		}

	return KErrNone;

	}

void DMemModelCodeSegMemory::Destroy()

	{

	if(iCreator)

		iCreator->UnmapUserRamCode(this, ETrue);	// remove from creating process if not fully loaded

	}

DMemModelCodeSegMemory::~DMemModelCodeSegMemory()

	{

	__KTRACE_OPT(KDLL,Kern::Printf("DMemModelCodeSegMemory::~DMemModelCodeSegMemory %x", this));

	__NK_ASSERT_DEBUG(iAccessCount==0);

	__NK_ASSERT_DEBUG(iOsAsids==0 || iOsAsids->Avail()==0); // check not mapped (inverted logic!)

	Mmu& m=Mmu::Get();

	if(iCodeAllocBase>=0)

		{

		// free allocated virtual memory space...

		TInt size = (iPageCount+iDataPageCount)<<KPageShift;

		TInt code_alloc=(size+m.iAliasMask)>>m.iAliasShift;

		DCodeSeg::Wait();

		MM::UserCodeAllocator->Free(iCodeAllocBase, code_alloc);

		DCodeSeg::Signal();

		}

	if(iPages)

		{

#ifdef __DEMAND_PAGING__

		if (iIsDemandPaged)

			{

			// Return any paged memory to the paging system

			MmuBase::Wait();

			NKern::LockSystem();

			DemandPaging& p = *DemandPaging::ThePager;

			for (TInt i = 0 ; i < iPageCount ; ++i)

				{

				if (iPages[i] != KPhysAddrInvalid)

					p.NotifyPageFree(iPages[i]);

				}

			NKern::UnlockSystem();

			MmuBase::Signal();

			Kern::Free(iCopyOfExportDir);

			iCopyOfExportDir = NULL;

			}

#endif

		MmuBase::Wait();

		m.FreePages(iPages,iPageCount+iDataPageCount, EPageMovable);

		MmuBase::Signal();

		Kern::Free(iPages);

		iPages = NULL;

#ifdef BTRACE_CODESEGS

		BTrace8(BTrace::ECodeSegs,BTrace::ECodeSegMemoryDeallocated,this,iPageCount<<m.iPageShift);

#endif

		}

	delete iOsAsids;

	}

DMemModelCodeSeg::DMemModelCodeSeg()

//

// Constructor

//

	:	iCodeAllocBase(-1),

		iDataAllocBase(-1)

	{

	}

DMemModelCodeSeg::~DMemModelCodeSeg()

//

// Destructor

//

	{

	__KTRACE_OPT(KDLL,Kern::Printf("DMemModelCodeSeg::Destruct %C", this));

	Mmu& m=Mmu::Get();

	DCodeSeg::Wait();

	if (iCodeAllocBase>=0)

		{

		TBool kernel=( (iAttr&(ECodeSegAttKernel|ECodeSegAttGlobal)) == ECodeSegAttKernel );

		TBool global=( (iAttr&(ECodeSegAttKernel|ECodeSegAttGlobal)) == ECodeSegAttGlobal );

		TInt r=KErrNone;

		if (kernel)

			{

			DMemModelProcess& kproc=*(DMemModelProcess*)K::TheKernelProcess;

			r=kproc.iCodeChunk->Decommit(iCodeAllocBase, iSize);

			}

		else if (global)

			{

			r=m.iGlobalCode->Decommit(iCodeAllocBase, iSize);

			}

		__ASSERT_DEBUG(r==KErrNone, MM::Panic(MM::EDecommitFailed));

		r=r; // stop compiler warning

		}

	if(Memory())

		Memory()->Destroy();

	if (iDataAllocBase>=0 && !iXIP)

		{

		SRamCodeInfo& ri=RamInfo();

		TInt data_alloc=(ri.iDataSize+ri.iBssSize+m.iPageMask)>>m.iPageShift;

		MM::DllDataAllocator->Free(iDataAllocBase, data_alloc);

		}

	DCodeSeg::Signal();

	Kern::Free(iKernelData);

	DEpocCodeSeg::Destruct();

	}

TInt DMemModelCodeSeg::DoCreateRam(TCodeSegCreateInfo& aInfo, DProcess*)

	{

	__KTRACE_OPT(KDLL,Kern::Printf("DMemModelCodeSeg::DoCreateRam %C", this));

	TBool kernel=( (iAttr&(ECodeSegAttKernel|ECodeSegAttGlobal)) == ECodeSegAttKernel );

	TBool global=( (iAttr&(ECodeSegAttKernel|ECodeSegAttGlobal)) == ECodeSegAttGlobal );

	Mmu& m=Mmu::Get();

	SRamCodeInfo& ri=RamInfo();

	iSize = Mmu::RoundToPageSize(ri.iCodeSize+ri.iDataSize);

	if (iSize==0)

		return KErrCorrupt;

	TInt total_data_size=ri.iDataSize+ri.iBssSize;

	TInt r=KErrNone;

	if (kernel)

		{

		DMemModelProcess& kproc=*(DMemModelProcess*)K::TheKernelProcess;

		if (!kproc.iCodeChunk)

			r=kproc.CreateCodeChunk();

		if (r!=KErrNone)

			return r;

		r=kproc.iCodeChunk->Allocate(iSize, 0, m.iAliasShift);

		if (r<0)

			return r;

		iCodeAllocBase=r;

		ri.iCodeRunAddr=(TUint32)kproc.iCodeChunk->Base();

		ri.iCodeRunAddr+=r;

		ri.iCodeLoadAddr=ri.iCodeRunAddr;

		if (ri.iDataSize)

			ri.iDataLoadAddr=ri.iCodeLoadAddr+ri.iCodeSize;

		if (total_data_size)

			{

			iKernelData=Kern::Alloc(total_data_size);

			if (!iKernelData)

				return KErrNoMemory;

			ri.iDataRunAddr=(TLinAddr)iKernelData;

			}

		return KErrNone;

		}

	if (global)

		{

		if (!m.iGlobalCode)

			r=m.CreateGlobalCodeChunk();

		if (r==KErrNone)

			r=m.iGlobalCode->Allocate(iSize, 0, m.iAliasShift);

		if (r<0)

			return r;

		iCodeAllocBase=r;

		ri.iCodeRunAddr=(TUint32)m.iGlobalCode->Base();

		ri.iCodeRunAddr+=r;

		ri.iCodeLoadAddr=ri.iCodeRunAddr;

		ri.iDataLoadAddr=0;	// we don't allow static data in global code

		ri.iDataRunAddr=0;

		TInt loadSize = ri.iCodeSize+ri.iDataSize;

		memset((TAny*)(ri.iCodeRunAddr+loadSize), 0x03, iSize-loadSize);

		return KErrNone;

		}

	DCodeSeg::Wait();

	if (total_data_size && !IsExe())

		{

		TInt data_alloc=(total_data_size+m.iPageMask)>>m.iPageShift;

		__KTRACE_OPT(KDLL,Kern::Printf("Alloc DLL data %x", data_alloc));

		r=MM::DllDataAllocator->AllocConsecutive(data_alloc, ETrue);

		if (r<0)

			r = KErrNoMemory;

		else

			{

			MM::DllDataAllocator->Alloc(r, data_alloc);

			iDataAllocBase=r;

			ri.iDataRunAddr=m.iDllDataBase+m.iMaxDllDataSize-((r+data_alloc)<<m.iPageShift);

			r = KErrNone;

			}

		}

	DCodeSeg::Signal();

	if(r==KErrNone)

		r = Memory()->Create(aInfo);

	return r;

	}

TInt DMemModelCodeSeg::DoCreateXIP(DProcess* aProcess)

	{

//	__KTRACE_OPT(KDLL,Kern::Printf("DMemModelCodeSeg::DoCreateXIP %C proc %O", this, aProcess));

	return KErrNone;

	}

TInt DMemModelCodeSeg::Loaded(TCodeSegCreateInfo& aInfo)

	{

	if(iXIP)

		return DEpocCodeSeg::Loaded(aInfo);

	TBool kernel=( (iAttr&(ECodeSegAttKernel|ECodeSegAttGlobal)) == ECodeSegAttKernel );

	TBool global=( (iAttr&(ECodeSegAttKernel|ECodeSegAttGlobal)) == ECodeSegAttGlobal );

	if (Pages())

		{

		TInt r = Memory()->Loaded(aInfo);

		if(r!=KErrNone)

			return r;

		}

	else if (kernel && iExeCodeSeg!=this)

		{

		Mmu& m=Mmu::Get();

		DMemModelProcess& kproc=*(DMemModelProcess*)K::TheKernelProcess;

		SRamCodeInfo& ri=RamInfo();

		// NOTE: Must do IMB before changing permissions since ARMv6 is very pedantic and

		// doesn't let you clean a cache line which is marked as read only.

		CacheMaintenance::CodeChanged(ri.iCodeRunAddr, ri.iCodeSize);

		TInt offset=ri.iCodeRunAddr-TLinAddr(kproc.iCodeChunk->iBase);

		kproc.iCodeChunk->ApplyPermissions(offset, iSize, m.iKernelCodePtePerm);

		}

	else if (global)

		{

		Mmu& m=Mmu::Get();

		SRamCodeInfo& ri=RamInfo();

		CacheMaintenance::CodeChanged(ri.iCodeRunAddr, ri.iCodeSize);

		TInt offset=ri.iCodeRunAddr-TLinAddr(m.iGlobalCode->iBase);

		m.iGlobalCode->ApplyPermissions(offset, iSize, m.iGlobalCodePtePerm);

		}

	return DEpocCodeSeg::Loaded(aInfo);

	}

void DMemModelCodeSeg::ReadExportDir(TUint32* aDest)

	{

	__KTRACE_OPT(KDLL,Kern::Printf("DMemModelCodeSeg::ReadExportDir %C %08x",this, aDest));

	if (!iXIP)

		{

		SRamCodeInfo& ri=RamInfo();

		TInt size=(ri.iExportDirCount+1)*sizeof(TLinAddr);

		if (Memory()->iCopyOfExportDir)

			{

			kumemput(aDest, Memory()->iCopyOfExportDir, size);

			return;

			}

		NKern::ThreadEnterCS();

		Mmu& m=Mmu::Get();

		TLinAddr src=ri.iExportDir-sizeof(TLinAddr);

		MmuBase::Wait();

		TInt offset=src-ri.iCodeRunAddr;

		TPhysAddr* physArray = Pages();

		TPhysAddr* ppa=physArray+(offset>>m.iPageShift);

		while(size)

			{

			TInt pageOffset = src&m.iPageMask;

			TInt l=Min(size, m.iPageSize-pageOffset);

			TLinAddr alias_src = m.MapTemp(*ppa++,src-pageOffset)+pageOffset;

			// Note, the following memory access isn't XTRAP'ed, because...

			// a) This function is only called by the loader thread, so even if

			//    exceptions were trapped the system is doomed anyway

			// b) Any exception will cause the crash debugger/logger to be called

			//    which will provide more information than if trapped exceptions

			//    and returned an error code.

			kumemput32(aDest, (const TAny*)alias_src, l);

			m.UnmapTemp();

			size-=l;

			src+=l;

			aDest+=l/sizeof(TUint32);

			}

		MmuBase::Signal();

		NKern::ThreadLeaveCS();

		}

	}

TBool DMemModelCodeSeg::OpenCheck(DProcess* aProcess)

	{

	return FindCheck(aProcess);

	}

TBool DMemModelCodeSeg::FindCheck(DProcess* aProcess)

	{

	__KTRACE_OPT(KDLL,Kern::Printf("CSEG:%08x Compat? proc=%O",this,aProcess));

	if (aProcess)

		{

		DMemModelProcess& p=*(DMemModelProcess*)aProcess;

		DCodeSeg* pPSeg=p.CodeSeg();

		if (iAttachProcess && iAttachProcess!=aProcess)

			return EFalse;

		if (iExeCodeSeg && iExeCodeSeg!=pPSeg)

			return EFalse;

		}

	return ETrue;

	}

void DMemModelCodeSeg::BTracePrime(TInt aCategory)

	{

#ifdef BTRACE_CODESEGS

	if (aCategory == BTrace::ECodeSegs || aCategory == -1)

		{