--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/kerneltest/e32test/debug/t_schedhook.cpp	Mon Oct 19 15:55:17 2009 +0100
@@ -0,0 +1,279 @@
+// 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);
+    }
+