// Copyright (c) 2007-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 <x86_mem.h>

GLDEF_C __NAKED__ void DoProcessSwitch()

	{

	// Enter with kernel locked

	// EBX = pointer to new thread (NThread)

	// EDI = pointer to TScheduler

	// ESI = pointer to TSubScheduler (SMP only)

	// Preserve ESI, EDI

	// eax = new process (DX86PlatProcess)

	asm("mov eax, [ebx+%0]": : "i"_FOFF(NThreadBase,iAddressSpace)); 

#ifdef __SMP__

	asm("cmp eax, [esi+%0]": :"i"_FOFF(TSubScheduler,iAddressSpace));

#else

	asm("cmp eax, [edi+%0]": : "i"_FOFF(TScheduler,iAddressSpace));

#endif

	asm("jz no_as_switch");

	asm("mov ecx, [eax+%0]": : "i"_FOFF(DMemModelProcess,iPageDir));

	asm("mov cr3, ecx");

#ifdef __SMP__

	asm("mov [esi+%0], eax": :"i"_FOFF(TSubScheduler,iAddressSpace));

#else

	asm("mov [edi+%0], eax": : "i"_FOFF(TScheduler,iAddressSpace));

#endif

	asm("no_as_switch:");

	// Get DThread* from NThread* (former contains latter so we go backwards in memory to get there)

	asm("mov ecx, 0");

	asm("lea ecx, [ecx+%0]": : "i"_FOFF(DThread,iNThread));

	asm("sub ebx, ecx");

	// check for memory being aliased...

	asm("mov ecx, [ebx+%0]": : "i"_FOFF(DMemModelThread,iAliasLinAddr));

	asm("cmp ecx, 0");

	asm("jz done");

	// restore alias...

	// Disable interrupts so setting the alias pde is must complete before

	// Mmu::RemoveAliasesForPageTable() can execute the alias IPI and vice 

	// versa.

	asm("cli ");					// turn off interrupts

	asm("mov edx, [ebx+%0]": : "i"_FOFF(DMemModelThread,iAliasPdePtr));

	asm("mov eax, [ebx+%0]": : "i"_FOFF(DMemModelThread,iAliasPde));

	asm("mov [edx], eax");

	asm("invlpg [ecx]");

	asm("sti ");					// turn on interrupts

	asm("done:");

	asm("ret");

	};

__NAKED__ void DMemModelThread::RestoreAddressSpace()

	{

#ifndef __SMP__

	//SMP FIXME

	asm("mov eax, [%a0]": : "i"(&TheScheduler.iCurrentThread));

	// edx = current thread owning process...

	asm("mov edx, 0");

	asm("lea edx, [edx+%0]": : "i"_FOFF(DThread,iNThread));

	asm("neg edx");

	asm("mov edx, [eax+edx+%0]": : "i"_FOFF(DThread,iOwningProcess));

	// update page directory and address space values...

	asm("cli");

	asm("mov [%a0], edx": :"i"(&TheScheduler.iAddressSpace));

	asm("mov [eax+%0], edx": : "i"_FOFF(NThreadBase,iAddressSpace));

	asm("mov edx, [edx+%0]": : "i"_FOFF(DMemModelProcess,iPageDir));

	asm("mov cr3, edx");

	asm("sti");

#endif

	asm("ret");

	}