// Copyright (c) 2002-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:
// e32test\debug\t_schedhook.cpp
// Overview:
// Test the device hook functionality.
// API Information:
// RBusLogicalChannel, DLogicalDevice
// Details:	
// - Load the specified logical driver dll and check the return value 
// is as expected. Create the logical channel for the current thread 
// and check the return value as KErrNone.
// - Install the scheduler hook and check it is installed successfully.
// - Enable scheduler callback, setup test threads, test the scheduler hook 
// before thread resume and after resume is as expected.
// - Perform some context switching and check rescheduler count is as expected.
// - Disable scheduler callback, perform some context switching and check 
// thread rescheduler count is as expected.
// - Re-enable scheduler callback, perform some context switching and check 
// thread rescheduler count is as expected.
// - Remove schedule hook and check it is successfully removed. perform some 
// context switching and check thread rescheduler count is as expected.
// - Check exception during context switching, user mode interrupt, WFAR, 
// exec call are as expected.
// - Uninstall scheduler hook and check it is as expected.
// Platforms/Drives/Compatibility:
// Hardware (Automatic).
// Assumptions/Requirement/Pre-requisites:
// Failures and causes:
// Base Port information:
// 
//

#include "context.h"
#define __KERNEL_MODE__
#include "nk_priv.h"
#include <nk_plat.h>
#undef __KERNEL_MODE__

RTest test(_L("T_SCHEDHOOK"));

RSchedhookTest ldd;

RThread Thread1;
TRequestStatus Request1;



RThread Thread2;
TRequestStatus Request2;

TInt Thread2Main(TAny*)
	{
	for(;;)
		{
		Request2 = KRequestPending;
		TRequestStatus* request = &Request1;
		Thread1.RequestComplete(request,KErrNone);
		User::WaitForRequest(Request2);
		}
	}



RThread		ThreadException;
TArmRegSet	ThreadExceptionData;
RThread		ThreadWFAR;
TArmRegSet	ThreadWFARData;
RThread		ThreadUserInt;
TArmRegSet	ThreadUserIntData;
RThread		ThreadExecCall;
TArmRegSet	ThreadExecCallData;



TInt GetThreadContext(RThread aThread,TArmRegSet& aContext)
	{
	TPtr8 context((TUint8*)&aContext,sizeof(TArmRegSet),sizeof(TArmRegSet));
	return ldd.GetThreadContext(aThread.Id(),context);
	}



void DumpContext(TArmRegSet& aContext,TInt aType)
	{
	test.Printf(_L("  Context type %d\n"),aType);
	test.Printf(_L("  r0 =%08x r1 =%08x r2 =%08x r3 =%08x\n"),aContext.iR0,aContext.iR1,aContext.iR2,aContext.iR3);
	test.Printf(_L("  r4 =%08x r5 =%08x r6 =%08x r7 =%08x\n"),aContext.iR4,aContext.iR5,aContext.iR6,aContext.iR7);
	test.Printf(_L("  r8 =%08x r9 =%08x r10=%08x r11=%08x\n"),aContext.iR8,aContext.iR9,aContext.iR10,aContext.iR11);
	test.Printf(_L("  r12=%08x r13=%08x r14=%08x r15=%08x\n"),aContext.iR12,aContext.iR13,aContext.iR14,aContext.iR15);
	test.Printf(_L("  cpsr=%08x"),aContext.iFlags);
	}



void TestContext()
	{
	TInt r;
	TArmRegSet context;

	test.Start(_L("Insert scheduler hook"));
	r = ldd.InsertHooks();
	test(r==KErrNone);

	test.Next(_L("Enable scheduler callback"));
	r = ldd.EnableCallback();
	test(r==KErrNone);

	test.Next(_L("Test exception context"));
	r = ThreadException.Create(KNullDesC,ThreadContextHwExc,KDefaultStackSize,&User::Allocator(),&ThreadExceptionData);
	test(r==KErrNone);
	ThreadException.SetPriority(EPriorityMore);
	r = ldd.SetTestThread(ThreadException.Id());   // So ldd handles the exception for this thread
	test(r==KErrNone);
	ThreadException.Resume();
	User::After(250000);   // Let thread run
	r = GetThreadContext(ThreadException,context);
	DumpContext(context,r);
	test(r==NThread::EContextException);
	test(CheckContextHwExc(&context,&ThreadExceptionData));

	test.Next(_L("Test user mode interupt context"));
	r = ThreadUserInt.Create(KNullDesC,ThreadContextUserInt,KDefaultStackSize,&User::Allocator(),&ThreadUserIntData);
	test(r==KErrNone);
	ThreadUserInt.SetPriority(EPriorityLess);
	ThreadUserInt.Resume();
	User::After(250000);   // Let thread run
	r = GetThreadContext(ThreadUserInt,context);
	DumpContext(context,r);
	test(r==NThread::EContextUserInterrupt);
	test(CheckContextUserInt(&context,&ThreadUserIntData));

	test.Next(_L("Test WFAR context"));
	r = ThreadWFAR.Create(KNullDesC,ThreadContextWFAR,KDefaultStackSize,&User::Allocator(),&ThreadWFARData);
	test(r==KErrNone);
	ThreadWFAR.SetPriority(EPriorityMore);
	ThreadWFAR.Resume();
	User::After(250000);   // Let thread run
	r = GetThreadContext(ThreadWFAR,context);
	DumpContext(context,r);
	test(r==NThread::EContextWFAR);
	test(CheckContextWFAR(&context,&ThreadWFARData));

	test.Next(_L("Test exec call context"));
	r = ThreadExecCall.Create(KNullDesC,ThreadContextExecCall,KDefaultStackSize,&User::Allocator(),&ThreadExecCallData);
	test(r==KErrNone);
	ThreadExecCall.SetPriority(EPriorityMore);
	ThreadExecCall.Resume();
	User::After(250000);   // Let thread run
	r = GetThreadContext(ThreadExecCall,context);
	DumpContext(context,r);
	test(r==NThread::EContextExec);
	test(CheckContextExecCall(&context,&ThreadExecCallData));

	test.End();
	}


TInt DoContextSwitches(TInt aCount)
	{
	TInt r = ldd.SetTestThread(Thread2.Id());   // Zero test count
	test(r==KErrNone);
	while(aCount)
		{
		Request1 = KRequestPending;
		TRequestStatus* request = &Request2;
		Thread2.RequestComplete(request,KErrNone);
		User::WaitForRequest(Request1);
		--aCount;
		}
	r = ldd.GetTestCount();
	test(r>=0);
	return r;
	}



GLDEF_C TInt E32Main()
    {
	test.Title();
	TInt r;
	
	test.Start(_L("Loading LDD"));
	r = User::LoadLogicalDevice(_L("D_SCHEDHOOK"));
	test(r==KErrNone || r==KErrAlreadyExists);

	test.Next(_L("Open channel to LDD"));
	r = ldd.Open();
	test(r==KErrNone);

	test.Next(_L("Installing scheduler hooks"));
	r = ldd.Install();
	if (r==KErrNotSupported)
		{
		test.Next(_L("Scheduler hooks not supported on this platform, skipping test"));
		ldd.Close();
		test.End();
		return KErrNone;
		}
	test(r==KErrNone);

	test.Next(_L("Enable scheduler callback"));
	r = ldd.EnableCallback();
	test(r==KErrNone);

	test.Next(_L("Setting up test thread"));
	r= Thread1.Open(RThread().Id());
	test(r==KErrNone);
	r = Thread2.Create(KNullDesC,Thread2Main,KDefaultStackSize,&User::Allocator(),NULL);
	test(r==KErrNone);
	r = ldd.SetTestThread(Thread2.Id());
	test(r==KErrNone);

	test.Next(_L("Test scheduler hook (wait)"));
	User::After(1000000); // 1 second
	TInt count = ldd.GetTestCount();
	test.Printf(_L("count=%d\n"),count);
	test(count==0);

	test.Next(_L("Test scheduler hook (resume)"));
	Request1 = KRequestPending;
	Thread2.Resume();
	User::WaitForRequest(Request1);
	count = ldd.GetTestCount();
	test.Printf(_L("count=%d\n"),count);
	test(count>0);

	test.Next(_L("Test scheduler hook (context switching)"));
	count = DoContextSwitches(1000);
	test.Printf(_L("count=%d\n"),count);
	test(count>=1000);

	test.Next(_L("Disable scheduler callback"));
	r = ldd.DisableCallback();
	test(r==KErrNone);
	count = DoContextSwitches(1000);
	test.Printf(_L("count=%d\n"),count);
	test(count==0);

	test.Next(_L("Re-enable scheduler callback"));
	r = ldd.EnableCallback();
	test(r==KErrNone);
	count = DoContextSwitches(1000);
	test.Printf(_L("count=%d\n"),count);
	test(count>=1000);

	test.Next(_L("Removing scheduler hook"));
	r = ldd.RemoveHooks();
	test(r==KErrNone);
	count=DoContextSwitches(1000);
	r = ldd.GetTestCount();
	test.Printf(_L("count=%d\n"),r);
	test(r==0);

	test.Next(_L("Test thread context"));
	TestContext();

	test.Next(_L("Uninstalling scheduler hook"));
	r = ldd.Uninstall();
	test(r==KErrNone);

	test.Next(_L("Closing ldd"));
	ldd.Close();
	
	test.End();

	return(0);
    }