// Copyright (c) 2006-2009 Nokia Corporation and/or its subsidiary(-ies).
// All rights reserved.
// This component and the accompanying materials are made available
// under the terms of "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:
// DEF075471 buffer security test
// 
//

/**
 @file
 @test
 @internalComponent - Internal Symbian test code
*/

#include "TBUFFERSECURITY.H"
#include "graphics/windowserverconstants.h"

//Set this #define in buffersecurity.h to add extra logging to this test case (useful when debugging a test fail)
//#define _TBUFS_LOGGING

//Set this #define in buffersecurity.h to initiate a long running soak test, this should be done periodically
//#define _TBUFS_TEST_SOAK_TEST



LOCAL_C TInt TestWsThreadFunc(TAny* aPtr)
	{
	TTestThreadData* data = (TTestThreadData*)aPtr;
	RWsSession ws;

	TInt err=ws.Connect();
	if (err!=KErrNone)
		return(err);
	ws.TestWrite(ws.WsHandle(), data->iOpCode, data->iData, data->iDataLength);
	ws.Flush();
	ws.Close();

	return(KErrNone);
	}

LOCAL_C TInt TestAnimDllThreadFunc(TInt aInt, TAny* /*aPtr*/)
	{
	TBool useTestWrite = (TBool)aInt;

	RWsSession ws;
	TInt err=ws.Connect();
	if (err != KErrNone)
		return (err);
	if (useTestWrite)
		{
		TUint32 data[2];
		data[0] = 200;
		data[1] = 4;
		ws.TestWrite(ws.WsHandle(), EWsClOpCreateAnimDll, REINTERPRET_CAST(TUint8*,data), 8);
		ws.Flush();
		}
	else
		{
		RAnimDll animDll(ws);
		const TPtrC	filename(REINTERPRET_CAST(TUint16*,4),200);
		animDll.Load(filename);
		animDll.Close();
		}
	ws.Close();

	return(KErrNone);
	}

TInt RTestIpcSession::Connect()
	{
	TVersion v(KWservMajorVersionNumber,KWservMinorVersionNumber,KWservBuildVersionNumber);
	TInt err = CreateSession(KWSERVServerName,v);
	if (err == KErrNone)
		{
		err=iWsHandle=SendReceive(EWservMessInit,TIpcArgs());
		}
	return err;
	}

TInt RTestIpcSession::SendBadBuffer()
	{
	TIpcArgs ipcArgs;
	TUint32 iData[2];
	iData[0] = 200;
	iData[1] = 4;
	ipcArgs.Set(KBufferMessageSlot,&iData);
	return SendReceive(EWservMessCommandBuffer,ipcArgs);
	}

LOCAL_C TInt TestIpcThreadFunc(TInt /*aInt*/, TAny* /*aPtr*/)
	{
	RTestIpcSession server;

	TInt handle=server.Connect();
	if (handle >= KErrNone)
		server.SendBadBuffer();

	return(KErrNone);
	}


CTBufferSecurity::CTBufferSecurity(CTestStep* aStep):
	CTWsGraphicsBase(aStep)
	{
	}

CTBufferSecurity::~CTBufferSecurity()
	{
	}

void CTBufferSecurity::ConstructL()
	{
	RProperty securityTesting;
	TInt err=securityTesting.Attach(KUidWServSecurityTesting,EWServSecTestBufferSecurity);
	User::LeaveIfError(err);
	err=securityTesting.Define(KUidWServSecurityTesting,EWServSecTestBufferSecurity,RProperty::EInt,KAllowAllPolicy,KWriteDeviceDataMgmtPolicy);
	if (err!=KErrAlreadyExists)
    	User::LeaveIfError(err);
	TInt value = ETrue;
	err = securityTesting.Set(value);
	User::LeaveIfError(err);
	}

void CTBufferSecurity::TestWsBufferL(TInt aOpCode, TUint aDataFill, TBool aEightBit)
	{
	// create a new thread in which to test each of the TWsClientOpCode values
	// so if the thread is panicked the test will not fail.
	RThread testThread;
	TTestThreadData data;
	TRequestStatus status;
	TUint8 storeData[KTestDataMax];
	TTestDataStore store;
	store.any=storeData;

	//collect initial values
	for (TInt count = 0; count < KTestDataMax; count++)
		storeData[count] = 0;
	switch (aOpCode)
		{
		case EWsClOpHeapSetFail:
			store.heapSetFail->type = RAllocator::ENone;
			break;
		case EWsClOpSetPointerCursorArea:
			store.cursorArea->area = TheClient->iWs.PointerCursorArea();
			break;
		case EWsClOpRawEvent:
			//skip switch off (would need a timer to turn it back on again)
			if (aDataFill == TRawEvent::ESwitchOff)
				aDataFill = TRawEvent::ENone;
			break;
#if defined(__WINS__)
		case EWsClOpSimulateXyInput:
			*store.xyInputType = EXYInputMouse;
			break;
#endif
		}

	data.iOpCode = aOpCode;
	if (aEightBit)
		{
		for (TInt count = 0; count < KTestDataMax; count++)
			data.iData[count] = (TUint8)aDataFill;
		}
	else
		{
		for (TInt count = 0; count < KTestDataMax32; count++)
			data.iData32[count] = aDataFill;
		}
	data.iDataLength = KTestDataMax;
	_LIT(KThreadNameFormat,"BufSecTestWsThread-%d-%d");
	HBufC* threadName = HBufC::NewLC(KThreadNameFormat().Size() + 32);
	TPtr threadNamePtr(threadName->Des());
	threadNamePtr.Format(KThreadNameFormat(), aOpCode, aDataFill);
	TInt err = testThread.Create(threadNamePtr, TestWsThreadFunc,KDefaultStackSize,KPanicThreadHeapSize,KPanicThreadHeapSize,(TAny*)&data,EOwnerThread);

	if (err != KErrNone)
		{
		User::After(100000);
		err = testThread.Create(threadNamePtr,TestWsThreadFunc,KDefaultStackSize,KPanicThreadHeapSize,KPanicThreadHeapSize,(TAny*)&data,EOwnerThread);
		}
	CleanupStack::PopAndDestroy(threadName);
	testThread.Logon(status);
	User::SetJustInTime(EFalse);
	testThread.Resume();
	User::WaitForRequest(status);
	User::SetJustInTime(ETrue);
#ifdef _TBUFS_LOGGING
	TLogMessageText logMessageText;
	TBufSStartLogText("TestWsBufferL");
	logMessageText.Format(_L(" OpCode(%d), ExitReason: %d"),aOpCode,testThread.ExitReason());
	TBufSLogFormat(logMessageText);
#endif

	testThread.Close();
	//reset some values to sensible ones
	switch (aOpCode)
		{
		case EWsClOpHeapSetFail:
		case EWsClOpSetPointerCursorArea:
#if defined(__WINS__)
		case EWsClOpSimulateXyInput:
#endif
			RThread resetThread;
			TheClient->iWs.TestWrite(TheClient->iWs.WsHandle(), data.iOpCode, storeData, data.iDataLength);
			TheClient->iWs.Flush();
			break;
		}
	}

void CTBufferSecurity::TestBadStringAnimDllL()
	{
	TEST(iTest->TestPanicL(&TestAnimDllThreadFunc,3,EFalse,NULL,KLitKernExec));
	}

void CTBufferSecurity::TestBadStringL()
	{
	TEST(iTest->TestWsPanicL(&TestAnimDllThreadFunc,EWservPanicBufferPtr,ETrue,NULL));
	}

void CTBufferSecurity::TestBadIpcL()
	{
	TEST(iTest->TestWsPanicL(&TestIpcThreadFunc,EWservPanicDescriptor,1));
	}

void CTBufferSecurity::RunTestCaseL(TInt /*aCurTestCase*/)
	{
	TInt ii;
	((CTBufferSecurityStep*)iStep)->SetTestStepID(KUnknownSYMTestCaseIDName);
	switch(++iTest->iState)
		{
		case 1:
			((CTBufferSecurityStep*)iStep)->SetTestStepID(_L("GRAPHICS-WSERV-0022"));
			_LIT(KBuffSecTestBadIpcL,"TestBadIpcL");
			iTest->LogSubTest(KBuffSecTestBadIpcL);
			TestBadIpcL();
			break;
		case 2:
/**
@SYMTestCaseID		GRAPHICS-WSERV-0549
*/
			((CTBufferSecurityStep*)iStep)->SetTestStepID(_L("GRAPHICS-WSERV-0549"));
			_LIT(KBuffSecTestBadStringLRAnimDll,"TestBadStringL use RAnimDll");
			iTest->LogSubTest(KBuffSecTestBadStringLRAnimDll);
			TestBadStringAnimDllL();
			break;
		case 3:
/**
@SYMTestCaseID		GRAPHICS-WSERV-0550
*/
			((CTBufferSecurityStep*)iStep)->SetTestStepID(_L("GRAPHICS-WSERV-0550"));
			_LIT(KBuffSecTestBadStringLTestWrite,"TestBadStringL use TestWrite");
			iTest->LogSubTest(KBuffSecTestBadStringLTestWrite);
			TestBadStringL();
			break;
#ifdef _TBUFS_TEST_SOAK_TEST
		case 4:
/**
@SYMTestCaseID		GRAPHICS-WSERV-0551
*/
			((CTBufferSecurityStep*)iStep)->SetTestStepID(_L("GRAPHICS-WSERV-0551"));
			_LIT(KBuffSecTestWsBufferLSoak8bit,"TestWsBufferL Soak 8bit");
			iTest->LogSubTest(KBuffSecTestWsBufferLSoak8bit);
			for(ii = EWsClOpDisconnect; ii < EWsClOpLastEnumValue; ii++)
				{
				TUint8 soak8=0;//gets set as KMaxTUint8 on first iteration
				do
					{
					--soak8;
					TestWsBufferL(ii,soak8);
					}
				while (soak8>0);
				}
			break;
		case 5:
/**
@SYMTestCaseID		GRAPHICS-WSERV-0552
*/
			((CTBufferSecurityStep*)iStep)->SetTestStepID(_L("GRAPHICS-WSERV-0552"));
			_LIT(KBuffSecTestWsBufferLSoak32bit,"TestWsBufferL Soak 32bit");
			iTest->LogSubTest(KBuffSecTestWsBufferLSoak32bit);
			for(ii = EWsClOpDisconnect; ii < EWsClOpLastEnumValue; ii++)
				{
				TUint32 soak32=0;//gets set as KMaxTUint8 on first iteration
				do
					{
					--soak32;
					TestWsBufferL(ii,soak32,EFalse);
					}
				while (soak32>0);
				}
			break;
#else
		case 4:
/**
@SYMTestCaseID		GRAPHICS-WSERV-0553
*/
			((CTBufferSecurityStep*)iStep)->SetTestStepID(_L("GRAPHICS-WSERV-0553"));
			_LIT(KBuffSecTestWsBufferL0,"TestWsBufferL 0");
			iTest->LogSubTest(KBuffSecTestWsBufferL0);
			for(ii = EWsClOpDisconnect; ii < EWsClOpLastEnumValue; ii++)
				{
				TestWsBufferL(ii,0);
				}
			break;
		case 5:
/**
@SYMTestCaseID		GRAPHICS-WSERV-0554
*/
			((CTBufferSecurityStep*)iStep)->SetTestStepID(_L("GRAPHICS-WSERV-0554"));
			_LIT(KBuffSecTestWsBufferL5,"TestWsBufferL 5");
			iTest->LogSubTest(KBuffSecTestWsBufferL5);
			for(ii = EWsClOpDisconnect; ii < EWsClOpLastEnumValue; ii++)
				{
				TestWsBufferL(ii,5);
				}
			break;
		case 6:
/**
@SYMTestCaseID		GRAPHICS-WSERV-0555
*/
			((CTBufferSecurityStep*)iStep)->SetTestStepID(_L("GRAPHICS-WSERV-0555"));
			_LIT(KBuffSecTestWsBufferL32,"TestWsBufferL 32");
			iTest->LogSubTest(KBuffSecTestWsBufferL32);
			for(ii = EWsClOpDisconnect; ii < EWsClOpLastEnumValue; ii++)
				{
				TestWsBufferL(ii,32);
				}
			break;
		case 7:
/**
@SYMTestCaseID		GRAPHICS-WSERV-0556
*/
			((CTBufferSecurityStep*)iStep)->SetTestStepID(_L("GRAPHICS-WSERV-0556"));
			_LIT(KBuffSecTestWsBufferL64,"TestWsBufferL 64");
			iTest->LogSubTest(KBuffSecTestWsBufferL64);
			for(ii = EWsClOpDisconnect; ii < EWsClOpLastEnumValue; ii++)
				{
				TestWsBufferL(ii,64);
				}
			break;
		case 8:
/**
@SYMTestCaseID		GRAPHICS-WSERV-0557
*/
			((CTBufferSecurityStep*)iStep)->SetTestStepID(_L("GRAPHICS-WSERV-0557"));
			_LIT(KBuffSecTestWsBufferL128,"TestWsBufferL 128");
			iTest->LogSubTest(KBuffSecTestWsBufferL128);
			for(ii = EWsClOpDisconnect; ii < EWsClOpLastEnumValue; ii++)
				{
				TestWsBufferL(ii,128);
				}
			break;
		case 9:
/**
@SYMTestCaseID		GRAPHICS-WSERV-0558
*/
			((CTBufferSecurityStep*)iStep)->SetTestStepID(_L("GRAPHICS-WSERV-0558"));
			_LIT(KBuffSecTestWsBufferL255,"TestWsBufferL 255");
			iTest->LogSubTest(KBuffSecTestWsBufferL255);
			for(ii = EWsClOpDisconnect; ii < EWsClOpLastEnumValue; ii++)
				{
				TestWsBufferL(ii,255);
				}
			break;
		case 10:
/**
@SYMTestCaseID		GRAPHICS-WSERV-0559
*/
			((CTBufferSecurityStep*)iStep)->SetTestStepID(_L("GRAPHICS-WSERV-0559"));
			_LIT(KBuffSecTestWsBufferL325,"TestWsBufferL 32-5");
			iTest->LogSubTest(KBuffSecTestWsBufferL325);
			for(ii = EWsClOpDisconnect; ii < EWsClOpLastEnumValue; ii++)
				{
				TestWsBufferL(ii,5,EFalse);
				}
			break;
		case 11:
/**
@SYMTestCaseID		GRAPHICS-WSERV-0560
*/
			((CTBufferSecurityStep*)iStep)->SetTestStepID(_L("GRAPHICS-WSERV-0560"));
			_LIT(KBuffSecTestWsBufferL32128,"TestWsBufferL 32-128");
			iTest->LogSubTest(KBuffSecTestWsBufferL32128);
			for(ii = EWsClOpDisconnect; ii < EWsClOpLastEnumValue; ii++)
				{
				TestWsBufferL(ii,128,EFalse);
				}
			break;
		case 12:
/**
@SYMTestCaseID		GRAPHICS-WSERV-0561
*/
			((CTBufferSecurityStep*)iStep)->SetTestStepID(_L("GRAPHICS-WSERV-0561"));
			_LIT(KBuffSecTestWsBufferL32512,"TestWsBufferL 32-512");
			iTest->LogSubTest(KBuffSecTestWsBufferL32512);
			for(ii = EWsClOpDisconnect; ii < EWsClOpLastEnumValue; ii++)
				{
				TestWsBufferL(ii,512,EFalse);
				}
			break;
		case 13:
/**
@SYMTestCaseID		GRAPHICS-WSERV-0562
*/
			((CTBufferSecurityStep*)iStep)->SetTestStepID(_L("GRAPHICS-WSERV-0562"));
			_LIT(KBuffSecTestWsBufferL320x01234567,"TestWsBufferL 32-0x01234567");
			iTest->LogSubTest(KBuffSecTestWsBufferL320x01234567);
			for(ii = EWsClOpDisconnect; ii < EWsClOpLastEnumValue; ii++)
				{
				TestWsBufferL(ii,0x01234567,EFalse);
				}
			break;
		case 14:
/**
@SYMTestCaseID		GRAPHICS-WSERV-0563
*/
			((CTBufferSecurityStep*)iStep)->SetTestStepID(_L("GRAPHICS-WSERV-0563"));
			_LIT(KBuffSecTestWsBufferL320x89abcdef,"TestWsBufferL 32-0x89abcdef");
			iTest->LogSubTest(KBuffSecTestWsBufferL320x89abcdef);
			for(ii = EWsClOpDisconnect; ii < EWsClOpLastEnumValue; ii++)
				{
				TestWsBufferL(ii,0x89abcdef,EFalse);
				}
			break;
		case 15:
/**
@SYMTestCaseID		GRAPHICS-WSERV-0564
*/
			((CTBufferSecurityStep*)iStep)->SetTestStepID(_L("GRAPHICS-WSERV-0564"));
			_LIT(KBuffSecTestWsBufferL320xCCCCCCCC,"TestWsBufferL 32-0xCCCCCCCC");
			iTest->LogSubTest(KBuffSecTestWsBufferL320xCCCCCCCC);
			for(ii = EWsClOpDisconnect; ii < EWsClOpLastEnumValue; ii++)
				{
				TestWsBufferL(ii,0xCCCCCCCC,EFalse);
				}
			break;
#endif
		default:
			((CTBufferSecurityStep*)iStep)->SetTestStepID(KNotATestSYMTestCaseIDName);
			((CTBufferSecurityStep*)iStep)->CloseTMSGraphicsStep();
			TInt err = RProperty::Set(KUidWServSecurityTesting,EWServSecTestBufferSecurity,EFalse);
			User::LeaveIfError(err);
			TestComplete();
		}
	((CTBufferSecurityStep*)iStep)->RecordTestResultL();
	}

__WS_CONSTRUCT_STEP__(BufferSecurity)