sl@0: // Copyright (c) 2008-2009 Nokia Corporation and/or its subsidiary(-ies).
sl@0: // All rights reserved.
sl@0: // This component and the accompanying materials are made available
sl@0: // under the terms of "Eclipse Public License v1.0"
sl@0: // which accompanies this distribution, and is available
sl@0: // at the URL "http://www.eclipse.org/legal/epl-v10.html".
sl@0: //
sl@0: // Initial Contributors:
sl@0: // Nokia Corporation - initial contribution.
sl@0: //
sl@0: // Contributors:
sl@0: //
sl@0: // Description:
sl@0: //
sl@0: 
sl@0: 
sl@0: #include <e32test.h>
sl@0: #include <e32math.h>
sl@0: #include <e32debug.h>
sl@0: 
sl@0: #include "backup_std.h"
sl@0: 
sl@0: LOCAL_D RTest				TheTest (_L("T_BackupServerIPC"));
sl@0: LOCAL_D CActiveScheduler*	TheScheduler;
sl@0: _LIT(KServerLauncherProcess, "T_BackupServerLauncher");
sl@0: _LIT(KServerStartProcess, "baksrvs");
sl@0: 
sl@0: _LIT(KServerName, "!BackupServer");
sl@0: 
sl@0: struct TTestInfo
sl@0: 	{
sl@0: 	TInt iFunction;
sl@0: 	TInt iType;
sl@0: 	TInt iArgCount;
sl@0: 	};
sl@0: 
sl@0: struct TExitDetails
sl@0: 	{
sl@0: 	TExitCategoryName iCategory;
sl@0: 	TExitType iExitType;
sl@0: 	TInt iReason;
sl@0: 	};
sl@0: 
sl@0: const TInt KAsynchDelay = 500000;
sl@0: 
sl@0: //===============================================================================
sl@0: 
sl@0: TBool IsFunctionAsynchronous(TInt aFunc)
sl@0: 	{
sl@0: 	TBool asynch = EFalse;
sl@0: 	switch(aFunc)
sl@0: 		{
sl@0: 		case EBakOpCodeEventReady:
sl@0: 		case EBakOpCodeCloseAllFiles:
sl@0: 		case EBakOpCodeCloseServer:
sl@0: 		case EBakOpCodeBackupOperationEventReady:
sl@0: 			asynch = ETrue;
sl@0: 			break;
sl@0: 
sl@0: 		default:
sl@0: 			break;
sl@0: 		}
sl@0: 	return asynch;
sl@0: 	}
sl@0: 
sl@0: class RIpcFuzzTest : public RSessionBase
sl@0: {
sl@0: public: // Constructors and destructor
sl@0: 
sl@0: 	/**
sl@0: 		* Constructor for performing 1st stage construction
sl@0: 		*/
sl@0: 	RIpcFuzzTest();
sl@0: 
sl@0: 	/**
sl@0: 		* Destructor.
sl@0: 		*/
sl@0: 	~RIpcFuzzTest();
sl@0: 
sl@0: 	/**
sl@0: 		* Performs test steps
sl@0: 		*/
sl@0: 
sl@0: 	void RunTestL(const TDesC& aTargetSrvName, TInt aFunc,
sl@0: 				TInt aTestType, TInt aArgCount);
sl@0: 
sl@0: private:
sl@0: 	TInt Fuzz(TInt aMsg, TInt aArgCount);
sl@0: 	TInt FuzzL(TInt aMsg, TInt aArgCount);
sl@0: 	TInt Fuzz8L(TInt aMsg, TInt aArgCount);
sl@0: };
sl@0: 
sl@0: RIpcFuzzTest::RIpcFuzzTest()
sl@0: 	{
sl@0: 	// No implementation required
sl@0: 	}
sl@0: 
sl@0: 
sl@0: RIpcFuzzTest::~RIpcFuzzTest()
sl@0: 	{
sl@0: 	Close();
sl@0: 	}
sl@0: 
sl@0: TInt RIpcFuzzTest::Fuzz(TInt aMsg, TInt aArgCount)
sl@0: 	{
sl@0: 	TIpcArgs args;
sl@0: 
sl@0: 	for(TInt i = 0; i < aArgCount;i++)
sl@0: 		{
sl@0: 		args.Set(i,Math::Random());
sl@0: 		}
sl@0: 
sl@0: 	TInt ret;
sl@0: 
sl@0: 	if(IsFunctionAsynchronous(aMsg))
sl@0: 		{
sl@0: 		ret = Send(aMsg, args);
sl@0: 		User::After(KAsynchDelay);
sl@0: 		}
sl@0: 	else
sl@0: 		{
sl@0: 		ret = SendReceive(aMsg, args);
sl@0: 		}
sl@0: 	return ret;
sl@0: 	}
sl@0: 
sl@0: TInt RIpcFuzzTest::Fuzz8L(TInt aMsg, TInt aArgCount)
sl@0: 	{
sl@0: 	HBufC8* buf = HBufC8::NewLC(255);
sl@0: 	TPtr8 ptr = buf->Des();
sl@0: 	ptr.Fill(Math::Random(),255);
sl@0: 
sl@0: 	TIpcArgs args;
sl@0: 
sl@0: 	for(TInt i = 0; i < aArgCount;i++)
sl@0: 		{
sl@0: 		args.Set(i,&ptr);
sl@0: 		}
sl@0: 
sl@0: 	TInt ret;
sl@0: 
sl@0: 	if(IsFunctionAsynchronous(aMsg))
sl@0: 		{
sl@0: 		ret = Send(aMsg, args);
sl@0: 		User::After(KAsynchDelay);
sl@0: 		}
sl@0: 	else
sl@0: 		{
sl@0: 		ret = SendReceive(aMsg, args);
sl@0: 		}
sl@0: 
sl@0: 	CleanupStack::PopAndDestroy(buf);
sl@0: 	return ret;
sl@0: 	}
sl@0: 
sl@0: TInt RIpcFuzzTest::FuzzL(TInt aMsg, TInt aArgCount)
sl@0: 	{
sl@0: 	HBufC* buf = HBufC::NewLC(255);
sl@0: 	TPtr ptr = buf->Des();
sl@0: 	ptr.Fill(Math::Random(),255);
sl@0: 
sl@0: 	TIpcArgs args;
sl@0: 
sl@0: 	for(TInt i = 0; i < aArgCount;i++)
sl@0: 		{
sl@0: 		args.Set(i,&ptr);
sl@0: 		}
sl@0: 
sl@0: 	TInt ret;
sl@0: 
sl@0: 	if(IsFunctionAsynchronous(aMsg))
sl@0: 		{
sl@0: 		ret = Send(aMsg, args);
sl@0: 		User::After(KAsynchDelay);
sl@0: 		}
sl@0: 	else
sl@0: 		{
sl@0: 		ret = SendReceive(aMsg, args);
sl@0: 		}
sl@0: 
sl@0: 	CleanupStack::PopAndDestroy(buf);
sl@0: 	return ret;
sl@0: 	}
sl@0: 
sl@0: void RIpcFuzzTest::RunTestL(const TDesC& aTargetSrvName,
sl@0: 							TInt aFunc, TInt aTestType, TInt aArgCount)
sl@0: 	{
sl@0: 		TVersion version(0,0,0);
sl@0: 
sl@0: 		User::LeaveIfError(CreateSession(aTargetSrvName, version, 200));
sl@0: 
sl@0: 		switch(aTestType)
sl@0: 			{
sl@0: 			case 0:
sl@0: 				Fuzz(aFunc,aArgCount);
sl@0: 				break;
sl@0: 
sl@0: 			case 1:
sl@0: 				Fuzz8L(aFunc,aArgCount);
sl@0: 				break;
sl@0: 
sl@0: 			case 2:
sl@0: 				FuzzL(aFunc,aArgCount);
sl@0: 				break;
sl@0: 			}
sl@0: 	}
sl@0: 
sl@0: 
sl@0: TInt KillProcess(const TDesC& aProcessName)
sl@0: 	{
sl@0: 	TFullName name;
sl@0: 
sl@0: 	RDebug::Print(_L("Find and kill \"%S\" process.\n"), &aProcessName);
sl@0: 
sl@0: 	TBuf<64> pattern(aProcessName);
sl@0: 	TInt length = pattern.Length();
sl@0: 	pattern += _L("*");
sl@0: 	TFindProcess procFinder(pattern);
sl@0: 
sl@0: 	while (procFinder.Next(name) == KErrNone)
sl@0: 		{
sl@0: 		if (name.Length() > length)
sl@0: 			{//If found name is a string containing aProcessName string.
sl@0: 			TChar c(name[length]);
sl@0: 			if (c.IsAlphaDigit() ||
sl@0: 				c == TChar('_') ||
sl@0: 				c == TChar('-'))
sl@0: 				{
sl@0: 				// If the found name is other valid application name
sl@0: 				// starting with aProcessName string.
sl@0: 				RDebug::Print(_L(":: Process name: \"%S\".\n"), &name);
sl@0: 				continue;
sl@0: 				}
sl@0: 			}
sl@0: 		RProcess proc;
sl@0: 		if (proc.Open(name) == KErrNone)
sl@0: 			{
sl@0: 			proc.Kill(0);
sl@0: 			RDebug::Print(_L("\"%S\" process killed.\n"), &name);
sl@0: 			}
sl@0: 		proc.Close();
sl@0: 		}
sl@0: 	return KErrNone;
sl@0: 	}
sl@0: 
sl@0: 
sl@0: TInt FuzzServerL(TAny* aTestInfo)
sl@0: 	{
sl@0:    	CTrapCleanup* cleanup=CTrapCleanup::New();
sl@0:    	TInt err=KErrNoMemory;
sl@0:    	if (cleanup)
sl@0:    		{
sl@0: 
sl@0: 		TTestInfo* info = (TTestInfo*)aTestInfo;
sl@0: 		RIpcFuzzTest fuzzer;
sl@0: 
sl@0: 		TRAP(err,fuzzer.RunTestL(KServerName,info->iFunction
sl@0: 				,info->iType, info->iArgCount));
sl@0: 
sl@0: 		fuzzer.Close();
sl@0: 
sl@0:    		delete cleanup;
sl@0:    		}
sl@0:    	return err;
sl@0: 	}
sl@0: 
sl@0: void TestServerApi(TInt aFunctionNumber,
sl@0: 			TInt aTestType,TInt aArgCount, TExitDetails& aExitDetails)
sl@0: 	{
sl@0: 
sl@0:     TTestInfo testInfo;
sl@0: 	testInfo.iFunction = aFunctionNumber;
sl@0: 	testInfo.iType = aTestType;
sl@0: 	testInfo.iArgCount = aArgCount;
sl@0: 
sl@0:     RThread thread;
sl@0:     _LIT(KThreadName,"FuzzerThread" );
sl@0: 	thread.Create(KThreadName,&FuzzServerL, KDefaultStackSize, NULL,&testInfo);
sl@0: 
sl@0: 	TRequestStatus threadStat;
sl@0: 	thread.Logon(threadStat);
sl@0: 
sl@0: 	TBool jit = User::JustInTime();
sl@0: 	User::SetJustInTime(EFalse);
sl@0: 
sl@0: 	thread.Resume();
sl@0: 
sl@0: 	User::WaitForRequest(threadStat);
sl@0: 
sl@0: 	User::SetJustInTime(jit);
sl@0: 
sl@0: 	aExitDetails.iCategory = thread.ExitCategory();
sl@0: 	aExitDetails.iReason = thread.ExitReason();
sl@0: 	aExitDetails.iExitType = thread.ExitType();
sl@0: 
sl@0: 	thread.Close();
sl@0: 
sl@0: 	}
sl@0: 
sl@0: 
sl@0: TInt LaunchServer(RProcess& aServer)
sl@0: 	{
sl@0: 
sl@0: 	TheTest.Printf(_L("Launching BackupServer...\n"));
sl@0: 
sl@0: 	TInt err = aServer.Create(KServerLauncherProcess, _L(""));
sl@0: 
sl@0: 	 if(err == KErrNone)
sl@0: 		 {
sl@0: 		 //Start server and wait until it is running
sl@0: 		 TRequestStatus serverStat;
sl@0: 		 aServer.SetJustInTime(false);
sl@0: 		 aServer.Resume();
sl@0: 
sl@0: 		 aServer.Rendezvous(serverStat);
sl@0: 		 User::WaitForRequest(serverStat);
sl@0: 		 }
sl@0: 
sl@0: 	 return err;
sl@0: 
sl@0: 	}
sl@0: 
sl@0: void PrintTestMessage(TInt iFunc, TInt iType, TInt iArgCount)
sl@0: 	{
sl@0: 	switch(iType)
sl@0: 		{
sl@0: 		case 0:
sl@0: 			TheTest.Printf(_L("\nFuzz Test on function number %d using random Int data. Number of Args = %d\n"), iFunc, iArgCount);
sl@0: 			break;
sl@0: 
sl@0: 		case 1:
sl@0: 			TheTest.Printf(_L("\nFuzz Test on function number %d using random Des8 data. Number of Args = %d\n"), iFunc, iArgCount);
sl@0: 			break;
sl@0: 
sl@0: 		case 2:
sl@0: 			TheTest.Printf(_L("\nFuzz Test on function number %d using random Des data. Number of Args = %d\n"), iFunc, iArgCount);
sl@0: 			break;
sl@0: 
sl@0: 		}
sl@0: 
sl@0: 	}
sl@0: 
sl@0: /**
sl@0: Invoke the tests
sl@0: */
sl@0: /**
sl@0: @SYMTestCaseID          SYSLIB-BAFL-CT-4039
sl@0: @SYMTestCaseDesc	    Tests Bafl Backup Server APIs for IPC Robustness
sl@0: @SYMTestPriority 	    High
sl@0: @SYMTestActions  	    The function calls each of the Bafl Backup Server APIs through a
sl@0: 						custom session object passing random TInt, Des8 and Des16 data .
sl@0: @SYMTestExpectedResults The server should be robust to all malformed messages and should not
sl@0: 						hang or panic.
sl@0: @SYMDEF                	INC113760
sl@0: */
sl@0: LOCAL_C void DoTestsL ()
sl@0:     {
sl@0: 
sl@0:     RProcess server;
sl@0:     const TInt KMinFuncNumber = 20;
sl@0:     const TInt KMaxFuncNumber = 35;
sl@0: 
sl@0:     TInt err = LaunchServer(server);
sl@0:     TheTest(err == 0);
sl@0: 
sl@0:     TExitDetails exitDetails;
sl@0: 
sl@0:     for(TInt i = KMinFuncNumber;i<= KMaxFuncNumber;i++)
sl@0:     	{
sl@0: 
sl@0:     	//Carry out each type of test
sl@0:     	for(TInt testType = 0; testType < 3;testType++)
sl@0:     		{
sl@0:        		//Carry out each test with number of arguments 1 - 4
sl@0:     		for(TInt argCount = 1;argCount <= 4;argCount++)
sl@0:     			{
sl@0:     			PrintTestMessage(i, testType, argCount);
sl@0: 
sl@0:     			TestServerApi(i, testType, argCount, exitDetails);
sl@0: 		    	//Kill the server process and verify that it was still running
sl@0: 		    	//If the server was already dead it would return the reason it exited
sl@0: 	    		if(server.ExitType() != EExitPending)
sl@0: 	    			{
sl@0: 	    			server.Kill(0);
sl@0: 	    			TInt exitReason = server.ExitReason();
sl@0: 	    			server.Close();
sl@0: 	    			TheTest(exitReason == 0);
sl@0: 	    			User::LeaveIfError(LaunchServer(server));
sl@0: 	    			}
sl@0:     			}
sl@0: 
sl@0: 	    	TheTest.Printf(_L("\nFuzz Test Successful\n"));
sl@0:     		}
sl@0:     	}
sl@0:     }
sl@0: 
sl@0: void SetupTestL()
sl@0: 	{
sl@0: 	TheScheduler = new (ELeave) CActiveScheduler;
sl@0:    	CActiveScheduler::Install (TheScheduler);
sl@0: 
sl@0: 	}
sl@0: 
sl@0: 
sl@0: GLDEF_C TInt E32Main ()
sl@0: 	{
sl@0: 
sl@0: 	TheTest.Printf (_L ("\n"));
sl@0: 	TheTest.Title ();
sl@0: 	TheTest.Start (_L(" @SYMTestCaseID:SYSLIB-BAFL-CT-4039 IPC Fuzz Tests "));
sl@0: 
sl@0:    	CTrapCleanup* cleanup=CTrapCleanup::New();
sl@0: 
sl@0:    	TInt err=KErrNoMemory;
sl@0:    	if (cleanup)
sl@0:    		{
sl@0:    		// Construct and install the active scheduler
sl@0:    		TRAP(err, SetupTestL());
sl@0: 
sl@0:    		//Kill baksrvs which may have already been launched by another test
sl@0:    		KillProcess(KServerStartProcess);
sl@0:    		KillProcess(KServerLauncherProcess);
sl@0: 
sl@0:    		TRAP (err, DoTestsL ());
sl@0:    		TheTest (err == KErrNone);
sl@0: 
sl@0:    		TheTest.End ();
sl@0:    		TheTest.Close ();
sl@0: 
sl@0:    		delete TheScheduler;
sl@0:    		delete cleanup;
sl@0: 
sl@0:    		}
sl@0: 	return err;
sl@0: 	}