kerneltest/e32test/emul/t_emul.cpp
changeset 0 a41df078684a
child 6 0173bcd7697c
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/kerneltest/e32test/emul/t_emul.cpp	Mon Oct 19 15:55:17 2009 +0100
@@ -0,0 +1,374 @@
+// 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\emul\t_emul.cpp
+// Overview:
+// Test the mechanism to escape threads from the emulator in order
+// to block on Windows objects
+// Test launching processes doesn't leak windows TLS indices
+// API Information:
+// Emulator
+// Details:
+// - Test the mechanism to escape threads from the emulator in order
+// to block on Windows objects:
+// - test escape and re-enter mechanism.
+// - block on Windows in EPOC thread and escaped EPOC thread.
+// Platforms/Drives/Compatibility:
+// All.
+// Assumptions/Requirement/Pre-requisites:
+// Failures and causes:
+// Base Port information:
+// 
+//
+
+#define __E32TEST_EXTENSION__
+#include <f32file.h>
+#include <e32atomics.h>
+#include "e32std.h"
+#include "e32std_private.h"
+#include "e32test.h"
+#include "emulator.h"
+#include "t_emul.h"
+
+#define WIN32_LEAN_AND_MEAN
+#pragma warning( disable : 4201 ) // nonstandard extension used : nameless struct/union
+#include <windows.h>
+#pragma warning( default : 4201 ) // nonstandard extension used : nameless struct/union
+
+LOCAL_D RTest test(_L("t_emul"));
+
+static TInt ELock;
+static HANDLE ESemaphore;
+static TInt EResult;
+
+TInt EscapeTimeoutThread(TAny*)
+	{
+	User::After(1000000);
+	if (__e32_atomic_add_ord32(&ELock, 1) == 0)
+		{
+		EResult=KErrTimedOut;
+		ReleaseSemaphore(ESemaphore,1,NULL);
+		}
+	return KErrNone;
+	}
+
+TInt EscapeSignalThread(TAny*)
+	{
+	if (__e32_atomic_add_ord32(&ELock, 1) == 0)
+		{
+		EResult=KErrNone;
+		ReleaseSemaphore(ESemaphore,1,NULL);
+		}
+	return KErrNone;
+	}
+
+TInt DoTestEscape(TBool aDoEscape)
+//
+// Test the mechanism to escape threads from the emulator in order to block on Windows objects
+//
+	{
+	ELock = 0;
+	EResult = KRequestPending;
+	ESemaphore = CreateSemaphoreA(NULL,0,1,NULL);
+	test (ESemaphore != NULL);
+	RThread t1,t2;
+	TRequestStatus s1,s2;
+	TInt r = t1.Create(KNullDesC,&EscapeSignalThread,0x1000,NULL,NULL);
+	test (r == KErrNone);
+	t1.SetPriority(EPriorityLess);
+	t1.Logon(s1);
+	t1.Resume();
+	r = t2.Create(KNullDesC,&EscapeTimeoutThread,0x1000,NULL,NULL);
+	test (r == KErrNone);
+	t2.SetPriority(EPriorityMore);
+	t2.Logon(s2);
+	t2.Resume();
+//
+	if (aDoEscape)
+		Emulator::Escape();
+	r = WaitForSingleObject(ESemaphore,INFINITE);
+	if (aDoEscape)
+		Emulator::Reenter();
+	test (r==WAIT_OBJECT_0);
+//
+	r = EResult;
+	t1.Kill(0);
+	t2.Kill(0);
+	t1.Close();
+	t2.Close();
+	User::WaitForRequest(s1);
+	User::WaitForRequest(s2);
+//
+	CloseHandle(ESemaphore);
+	return r;
+	}
+
+void TestEscape()
+//
+// Test the mechanism to escape threads from the emulator in order to block on Windows objects
+//
+	{
+	test.Start(_L("Test escape and reenter mechanism"));
+	for (TInt i = 0;i<10000;++i)
+		{
+		Emulator::Escape();
+		if (i%100 == 0)
+			Sleep(10);
+		Emulator::Reenter();
+		}
+	test.Next(_L("Block on Windows in EPOC thread"));
+	TInt r = DoTestEscape(EFalse);
+	test (r == KErrTimedOut);
+	test.Next(_L("Block on Windows in escaped EPOC thread"));
+	r = DoTestEscape(ETrue);
+	test (r == KErrNone);
+	test.End();
+	}
+
+TInt CountRemainingTlsIndicies()
+	{
+	const TInt KMax = 2000;
+	
+	TBool allocated[KMax];
+	memclr(allocated, sizeof(TBool) * KMax);
+	TInt i;
+	for (i = 0 ; i < KMax ; ++i)
+		{
+		TInt index = TlsAlloc();
+		if (index == TLS_OUT_OF_INDEXES)
+			break;
+		test(index >= 0 && index < KMax);
+		allocated[index] = ETrue;
+		}
+	for (TInt j = 0 ; j < KMax ; ++j)
+		{
+		if (allocated[j])
+			test(TlsFree(j));
+		}
+	return i;
+	}
+	   
+void RunSlave(TSlaveAction aAction)
+	{
+	RProcess p;
+	TBuf<8> arg;
+	arg.Format(_L("%d"), aAction);
+	test_KErrNone(p.Create(KTEmulSlaveName, arg));
+	p.Resume();
+	TRequestStatus status;
+	p.Logon(status);
+	User::WaitForRequest(status);
+	test_KErrNone(status.Int());
+	test_Equal(EExitKill, p.ExitType());
+	p.Close();
+	}
+
+void TestRuntimeCleanup()
+	{
+	test.Start(_L("Test Codewarrior runtime library is correctly cleaned up"));
+	TInt initIndicies = CountRemainingTlsIndicies();
+	
+	test.Next(_L("Test creating a process doesn't leak windows TLS indicies"));
+	RunSlave(ESlaveDoNothing);
+	test_Equal(initIndicies, CountRemainingTlsIndicies());
+
+	test.Next(_L("Test leaving in an exe doesn't leak windows TLS indicies"));
+	RunSlave(ESlaveTrapExceptionInExe);
+	test_Equal(initIndicies, CountRemainingTlsIndicies());
+	
+	test.Next(_L("Test leaving in a linked DLL doesn't leak windows TLS indicies"));
+	RunSlave(ESlaveTrapExceptionInLinkedDll);
+	test_Equal(initIndicies, CountRemainingTlsIndicies());
+	
+	test.Next(_L("Test leaving in a loaded DLL doesn't leak windows TLS indicies"));
+	RunSlave(ESlaveTrapExceptionInLoadedDll);
+	test_Equal(initIndicies, CountRemainingTlsIndicies());
+
+	test.Next(_L("Test cleanup doesn't happen while DLL still loaded"));
+	RLibrary l;
+	test_KErrNone(l.Load(KTEmulDll2Name));
+	RunSlave(ESlaveTrapExceptionInLoadedDll);
+	TInt midCount = CountRemainingTlsIndicies();
+	test(initIndicies > midCount);
+
+	test.Next(_L("Test previous detach doesn't cause runtime to be re-initalised"));
+	TTrapExceptionInDllFunc func =
+		(TTrapExceptionInDllFunc)l.Lookup(KTrapExceptionInDllOrdinal);
+	test_NotNull(func);
+	func();
+	test_Equal(midCount, CountRemainingTlsIndicies());	
+	l.Close();
+	test_Equal(initIndicies, CountRemainingTlsIndicies());
+	
+	test.End();
+	}
+
+
+void DoSomething2L()
+	{
+	test.Printf(_L("@\n"));
+	}
+
+
+void DoSomething1L()
+	{
+	TInt i = -1, j = 1;
+	for ( ; ; )
+		{
+		i++;
+		//DoSomething2L() must only be called once , else we know that trap mechanism didn't work 
+		test( i<2);
+		if (i == j)  
+			{
+			User::Leave(KErrNotFound);
+			}
+
+		TRAP_IGNORE(DoSomething2L());
+		TRAPD(errr, DoSomething2L());
+		TInt r;
+		TRAP(r, DoSomething2L());
+
+		}
+	}
+
+
+void LeaveIfArgIsTwo(TInt aCall);
+
+class CFred : public CBase
+{
+	public:
+static CFred* NewLC();
+	CFred();
+	~CFred();
+};
+
+void DoSomething3L()
+	{
+	// Push something on the cleanup stack so we can see whyen it gets cleaned up.
+	CFred *fred = CFred::NewLC();
+		
+	TInt beforeFunc=0;
+	TInt betweenFuncAndTRAPD=0;
+	TInt afterTRAPD=0;
+	for(TInt loop=1; loop<=2; ++loop)
+		{
+		++beforeFunc;
+		test.Printf(_L("Before LeaveIfArgIsTwo()\n"));
+
+		// The first time around the loop, this function call works.
+		// The second time, it leaves KErrGeneral
+		// Then when TRAP mechanism isn't working properly the emulator (correctly) 
+		// would delete fred, and (incorrectly) jump to the line
+		// just after the TRAPD call a few lines further down the file!
+		LeaveIfArgIsTwo(loop);
+
+		++betweenFuncAndTRAPD;
+		test.Printf(_L("Between LeaveIfArgIsTwo() and TRAPD\n"));
+
+		TRAPD(err, test.Printf(_L("Inside TRAPD\n") ) );
+
+
+		// It should only be possible to reach this section of code by executing all the lines
+		// between LeaveIfArgIsTwo and here.
+		++afterTRAPD;
+		
+		}
+
+	// Should NEVER get here because LeaveIfArgIsTwo did a leave the second time around the loop
+	test.Printf(_L("After loop (should NEVER get here) -\n\
+					beforeFunc %d\n\
+					betweenFuncAndTRAPD %d\n\
+					afterTRAPD %d\n"),
+					beforeFunc, betweenFuncAndTRAPD, afterTRAPD);
+	
+	test.Printf(_L("It should be impossible for afterTRAPD to be larger than betweenFuncAndTRAPD\n"));
+	test(afterTRAPD <= betweenFuncAndTRAPD);
+
+	// Cleanup our cleanup stack. 
+	CleanupStack::PopAndDestroy(&fred);
+	}
+
+void LeaveIfArgIsTwo(TInt aCall)
+	{
+	
+	test.Printf(_L("aCall    %d\n"), aCall);
+	if(aCall == 2)
+		{
+		User::Leave(KErrGeneral);
+		}
+	}
+
+CFred *CFred::NewLC()
+	{
+	CFred *self = new(ELeave) CFred;
+	CleanupStack::PushL(self);
+	return self;
+	}
+
+CFred::CFred()
+	{
+	test.Printf(_L("CFred %x\n"), this);
+	}
+
+CFred::~CFred()
+	{
+	test.Printf(_L("~CFred %x\n"), this);
+	}
+
+
+
+GLDEF_C TInt E32Main()
+//
+//
+//
+	{
+	test.Title();
+	test.Start(_L("Starting tests ..."));
+
+	// Turn off evil lazy dll unloading
+	RLoader l;
+	test(l.Connect()==KErrNone);
+	test(l.CancelLazyDllUnload()==KErrNone);
+	l.Close();
+	
+	TestEscape();
+#ifdef __CW32__
+	TestRuntimeCleanup();
+#endif
+
+	// The following tests were added to test that the TRAP mechanism works correctly in case of
+	// nested TRAP's. A compiler bug (winscw) caused the following tests to fail, since the wrong 
+	// trap handler would be invoked, when User::Leave() was called.
+	
+	test.Next(_L("Check User::Leave is handled by the correct TRAP handler when nested TRAPs are present - Simple scenario\n")); 
+		
+ 	TRAPD(err, DoSomething1L());
+	test(err == KErrNotFound);
+	
+	test.Next(_L("Check User::Leave is handled by the correct TRAP handler when nested TRAPs are present - Cleanup stack scenario\n")); 		
+	__UHEAP_MARK;
+	CTrapCleanup* tc = CTrapCleanup::New();
+	if (tc == 0) return KErrNoMemory;
+	
+	TRAPD(err2, DoSomething3L());
+	test(err2==KErrGeneral);	
+
+	delete tc;
+	__UHEAP_MARKEND;
+	
+	
+	test.End();
+	test.Close();
+	return KErrNone;
+	}
+