/*
* Copyright (c) 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:
*
*/


/**
Overview:
	Test DLL Writeable Static Data support

API Information:


Details:
	- Each process has independent DLL WSD
	- Whether DLL linked directly or indirectly
	- Whether DLL loaded dynamically or statically
	- DLL WSD is consistent under heavy usage by multiple processes
	- IPC works to/from DLL WSD descriptors & TRequestStatus
	This source file builds in 4 configurations, with each of
	direct and indirect linking either used or not used.
	These configurations are set by 4 MM files, t_dllwsd[d][i].mmp
	Any of the exe created from the MMP files can be started 
	to run the tests, it does not matter which is used. 
	All exe configurations will be used during the tests.

Platforms/Drives/Compatibility:
	All.

Assumptions/Requirement/Pre-requisites:
	

Failures and causes:
	

Base Port information:

*/

#define __E32TEST_EXTENSION__

#include <e32test.h>
#include <e32svr.h>
#include <f32dbg.h>
#include <u32std.h>
#include "t_dllwsd_dll.h"
#include "t_dllwsd_dlli.h"


LOCAL_D RTest test(_L("T_DLLWSD"));

enum TTestFunc
	{
	ETestFuncTestCons=1,
	ETestFuncThrash1,
	ETestFuncIpcTest,
	ETestFuncIpcGet,
	ETestFuncIpcReverse,
	ETestFuncIpcSet,
	ETestFuncPanic,
	};

// Test session for IPC use of WSD, talks to the same server as RDllWsd
class RIpcTestSession : public RSessionBase
	{
	public:
	TInt Connect()
		{
		return CreateSession(_L("IpcTestServer"), TVersion());
		}
	void Get(TBuf<60000>& buf, TRequestStatus& req)
		{
		SendReceive(ETestFuncIpcGet, TIpcArgs(&buf), req);
		}
	void Reverse(TRequestStatus& req)
		{
		SendReceive(ETestFuncIpcReverse, req);
		}
	void Set(const TBuf<60000>& buf, TRequestStatus& req)
		{
		SendReceive(ETestFuncIpcSet, TIpcArgs(&buf), req);
		}
	};

#ifdef T_DLLWSD_DIRECT
void FillBuf(TInt start, TInt inc)
	{
	for (int ii=0; ii<WsdBuf().Length(); ii++)
		{
		WsdBuf()[ii] = (unsigned short)start;
		start += inc;
		}
	}

TInt CheckBuf(TInt start, TInt inc)
	{
	for (int ii=0; ii<WsdBuf().Length(); ii++)
		{
		if (WsdBuf()[ii] != start)
			return KErrGeneral;
		start += inc;
		}
	return KErrNone;
	}
#endif

class CDllWsdServer : public CServer2
	{
public:
	CDllWsdServer() : CServer2(EPriorityStandard)
		{
		}
	CSession2* NewSessionL(const TVersion& /*aVersion*/,const RMessage2& /*aMessage*/) const;
	mutable TInt iCount;
	};

class CDllWsdSession : public CSession2
	{
public:
	CDllWsdSession()
		{
		ResetConsistencyCheck();
		}
		
	~CDllWsdSession()
		{
		// transient server immediate shutdown when last client disconnects
		if (--((CDllWsdServer*)Server())->iCount == 0)
			CActiveScheduler::Stop();
		}
		
	void ResetConsistencyCheck()
		{
		iX=42;
		iY=0;
		}

	void OptResetConsistencyCheck()
		{
#if !defined(T_DLLWSD_DIRECT) && !defined(T_DLLWSD_INDIRECT)
		// if DLL has been unloaded (dynamic library closed, no static link)
		// WSD will be reset
		ResetConsistencyCheck();
#endif
		}
		
	TInt TestConsistency()
		{
#ifdef T_DLLWSD_DIRECT
		// static direct
		if (WsdFuncX() != iX++)
			return KErrGeneral;
		if (WsdFuncY() != iY++)
			return KErrGeneral;
#endif

#ifdef T_DLLWSD_INDIRECT
		// static indirect
		if (IndWsdFuncX() != iX++)
			return KErrGeneral;
		if (IndWsdFuncY() != iY++)
			return KErrGeneral;
#endif

		// dynamic direct
		OptResetConsistencyCheck();
		RLibrary lib;
		TInt err = lib.Load(_L("t_dllwsd_dll"));
		if (err) return err;
		if ((*lib.Lookup(1))/*WsdFuncX*/() != iX++)
			return KErrGeneral;
		if ((*lib.Lookup(2))/*WsdFuncX*/() != iY++)
			return KErrGeneral;
		lib.Close();

		// dynamic indirect
		OptResetConsistencyCheck();
		err = lib.Load(_L("t_dllwsd_dlli"));
		if (err) return err;
		if ((*lib.Lookup(1))/*IndWsdFuncX*/() != iX++)
			return KErrGeneral;
		if ((*lib.Lookup(2))/*IndWsdFuncX*/() != iY++)
			return KErrGeneral;
		lib.Close();
		
		return KErrNone;
		}
		
	TInt Thrash1()
		{
		TTime start;
		start.HomeTime();
		TInt count = 0;
		const TTimeIntervalMicroSeconds limit(10000000); // 10 seconds
		for (;; count++)
			{
			TInt err = TestConsistency();
			if (err) return err;
			TTime now;
			now.HomeTime();
			if (now.MicroSecondsFrom(start) > limit)
				break;
			}
		return count > 0 ? count : -count;
		}
		
	TInt IpcTest()
		{
#ifdef T_DLLWSD_DIRECT
    	RIpcTestSession s;
    	TInt err = s.Connect();
    	if (!err) return err;
        WsdBuf().SetLength(WsdBuf().MaxLength());
    	for (int i=0; i<10; i++)
    		{
    		// 0..n -> buf
	        FillBuf(0,1);
	        err = CheckBuf(0,1);
	    	if (!err) return err;
	        
	        // buf -> server
	        s.Set(WsdBuf(), WsdReq());
	        err = CheckBuf(0,1);
	    	if (!err) return err;
	    	
	    	// use TReqestStatus in WSD
	        User::WaitForRequest(WsdReq());
	        if (!WsdReq().Int()) return WsdReq().Int();
	        
	        // 0..0 -> buf
	        FillBuf(0,0);
	        err = CheckBuf(0,0);
	    	if (!err) return err;
	        WsdReq() = KRequestPending;

			// reverse buf on server
	        s.Reverse(WsdReq());
	        
			// local buf is still 0..0
	        err = CheckBuf(0,0);
	    	if (!err) return err;

	    	// use TReqestStatus in WSD
	        User::WaitForRequest(WsdReq());
	        if (!WsdReq().Int()) return WsdReq().Int();

			// local buf is still 0..0
	        err = CheckBuf(0,0);
	    	if (!err) return err;

	        // get buf from server
	        s.Get(WsdBuf(), WsdReq());
	        User::WaitForRequest(WsdReq());
	        
	        // buf is n..0
	        err = CheckBuf(59999,-1);
	    	if (!err) return err;
    		}
    	s.Close();
		return KErrNone;
#else
		return KErrNotSupported;
#endif
		}
	
	void ServiceL(const RMessage2& aMessage)
		{
#ifdef T_DLLWSD_DIRECT
		TInt ii=0;
#endif
		switch (aMessage.Function())
			{
			case ETestFuncTestCons:
				aMessage.Complete(TestConsistency());
				break;
			case ETestFuncThrash1:
				aMessage.Complete(Thrash1());
				break;
			case ETestFuncIpcTest:
				aMessage.Complete(IpcTest());
				break;
			case ETestFuncIpcGet:
#ifdef T_DLLWSD_DIRECT
				aMessage.WriteL(0, WsdBuf());
				aMessage.Complete(KErrNone);
#else
				aMessage.Complete(KErrNotSupported);
#endif
				break;
			case ETestFuncIpcReverse:
#ifdef T_DLLWSD_DIRECT
				for (ii=0; ii<WsdBuf().Length()/2; ii++)
					{
					TInt o = WsdBuf().Length() - 1 - ii;
					TInt t = WsdBuf()[ii];
					WsdBuf()[ii] = WsdBuf()[o];
					WsdBuf()[o] = (unsigned short)t;
					}
				aMessage.Complete(KErrNone);
#else
				aMessage.Complete(KErrNotSupported);
#endif
				break;
			case ETestFuncIpcSet:
#ifdef T_DLLWSD_DIRECT
				aMessage.ReadL(0, WsdBuf());
				aMessage.Complete(KErrNone);
#else
				aMessage.Complete(KErrNotSupported);
#endif
				break;
			case ETestFuncPanic:
				User::Panic(_L("As requested..."), 0);
				break;
			default:
				aMessage.Panic(_L("Unrecognised"), aMessage.Function());
				break;
			}
		}
	
	int iX;
	int iY;
	};
	
CSession2* CDllWsdServer::NewSessionL(const TVersion& /*aVersion*/,const RMessage2& /*aMessage*/) const
	{
	iCount++;
	return new(ELeave) CDllWsdSession;
	}
	
TInt SlaveMain()
	{
	TName name;
	User::CommandLine(name);
	
	CTrapCleanup* cleanup = CTrapCleanup::New();
	if (!cleanup)
		return KErrNoMemory;
	
	TRAPD(err, 
		{
		CActiveScheduler* sched=new(ELeave) CActiveScheduler;
		CActiveScheduler::Install(sched);

		CDllWsdServer* server = new(ELeave) CDllWsdServer;
		server->StartL(name);

		RProcess::Rendezvous(KErrNone);
		CActiveScheduler::Start();
		
		delete server;
		delete sched;
		});
	delete cleanup;

	return err;
	}

//
// Master test controller
//

class RDllWsd : public RSessionBase
	{
public:
	RDllWsd(const TDesC& aServerName, const TDesC& aExeName = _L("t_dllwsddi"))
		{
		test.Start(_L("RDllWsd create"));
		RProcess proc;
		test_KErrNone(proc.Create(aExeName, aServerName));
		TRequestStatus req;
		proc.Rendezvous(req);
		proc.Resume();
		User::WaitForRequest(req);
		test_KErrNone(req.Int());
		test_KErrNone(CreateSession(aServerName, TVersion()));
		proc.Close();
		test.End();
		}
	TInt ConsistencyTest()
		{
		return SendReceive(ETestFuncTestCons);
		}
	void ThrashTest1(TRequestStatus& aStatus)
		{
		SendReceive(ETestFuncThrash1, aStatus);
		}
	TInt IpcTest()
		{
		return SendReceive(ETestFuncIpcTest);
		}
	TInt Panic()
		{
		return SendReceive(ETestFuncPanic);
		}
	};

void BasicTest()
	{
	test.Start(_L("BasicConsistency"));

	// create a test server/process for each link variant
	RDllWsd slaves[] =
		{
		RDllWsd(_L("slave1"), _L("t_dllwsd")),
		RDllWsd(_L("slave2"), _L("t_dllwsdd")),
		RDllWsd(_L("slave3"), _L("t_dllwsdi")),
		RDllWsd(_L("slave4"), _L("t_dllwsddi")),
		RDllWsd(_L("slave5"), _L("t_dllwsd")),
		RDllWsd(_L("slave6"), _L("t_dllwsdd")),
		RDllWsd(_L("slave7"), _L("t_dllwsdi")),
		RDllWsd(_L("slave8"), _L("t_dllwsddi"))
		};
	TInt nSlaves = sizeof(slaves)/sizeof(slaves[0]);
	TInt ii;
	// do this a few times
	for (TInt jj=0; jj<10; jj++)
		{
		// all four test variants
		for (ii=0; ii<nSlaves; ii++)
			{
			// repeat the test different numbers of times, to ensure WSD values diverge
			for (TInt kk=0; kk<ii+2; kk++)
				{
				// change order in which processes run the tests
				int idx = (ii + jj) % nSlaves;
				test_KErrNone(slaves[idx].ConsistencyTest());
				}
			}
		// start and stop an extra process
		RDllWsd extra(_L("slave9"), _L("t_dllwsddi"));
		test_KErrNone(extra.ConsistencyTest());
		extra.Close();
		}

	for (ii=nSlaves-1; ii>=0; ii--)
		slaves[ii].Close();

	test.End();
	}

void ThrashTest1()
	{
	test.Start(_L("ThrashTest1"));

	// create a test server/process for each link variant
	RDllWsd slaves[4] =
		{
		RDllWsd(_L("slaveA"), _L("t_dllwsd")),
		RDllWsd(_L("slaveB"), _L("t_dllwsdd")),
		RDllWsd(_L("slaveC"), _L("t_dllwsdi")),
		RDllWsd(_L("slaveD"), _L("t_dllwsddi"))
		};
	
	TRequestStatus req[4];
	TInt ii;
	// start the thrash tests
	for (ii=0; ii<4; ii++)
		{
		slaves[ii].ThrashTest1(req[ii]);
		test.Printf(_L("slave %d thrash started\n"), ii);
		}

	// show some progress to indicate that things are running		
	for (ii=0; ii<8; ii++)
		{
		test.Printf(_L("Waiting %d\n"), ii);
		User::After(1000000);
		}
	// demonstrate that test processes are still doing their stuff
	test.Printf(_L("Still a couple of seconds to wait...\n"));

	// wait till the test process are done
	for (ii=0; ii<4; ii++)
		{
		User::WaitForRequest(req[ii]);
		// show how much each process did
		test.Printf(_L("Slave %d count = %d\n"), ii, req[ii].Int());
		test_NotNegative(req[ii].Int());
		}
		
	for (ii=3; ii>=0; ii--)
		slaves[ii].Close();

	test.End();
	}

void PanicTest()
	{
	test.Start(_L("PanicTest1"));

	// create a test server/process for each link variant
	RDllWsd slaves[4] =
		{
		RDllWsd(_L("slaveP1"), _L("t_dllwsd")),
		RDllWsd(_L("slaveP2"), _L("t_dllwsdd")),
		RDllWsd(_L("slaveP3"), _L("t_dllwsdi")),
		RDllWsd(_L("slaveP4"), _L("t_dllwsddi"))
		};
	TInt ii;
	for (ii=0; ii<4; ii++)
		slaves[ii].Panic();
		
	for (ii=0; ii<4; ii++)
		slaves[ii].Close();
	}

void IpcTest()
	{
	test.Start(_L("IPC test"));
	// these two processes will use t_dllwsddi, static link variant
	RDllWsd server(_L("IpcTestServer"));
	RDllWsd client(_L("IpcTestClient"));
	// client will talk to IpcTestServer, ie the server
	test_KErrNone(client.IpcTest());
	client.Close();
	server.Close();
	test.End();
	}
	
TInt MasterMain()
	{
	test.Title();
	test.Start(_L("Test"));

	BasicTest();
	ThrashTest1();
	IpcTest();
//	PanicTest();

	test.End();
	return KErrNone;
	}

TInt E32Main()
	{
	if (User::CommandLineLength() > 0)	// command line contains server name
		return SlaveMain();
	else
		return MasterMain();
	}