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

// f32test\loader\handshare.cpp

// 

//

#include <e32svr.h>

#include <e32test.h>

#include "handshare.h"

#ifdef __VC32__

#pragma warning(disable:4706)

#endif

GLDEF_D RTest test(_L("HANDSHARE_SVR"));

#define PANIC()		FHSvrPanic(__LINE__)

#define FHS_ASSERT(c)	((void)((c)||(PANIC(),0)))

const TTimeIntervalMicroSeconds32 KHalfSecond(500000);

const TInt KHeapSize=0x2000;

void FHSvrPanic(TInt aLine)

	{

	User::Panic(_L("FHServer"),aLine);

	}

LOCAL_D TInt gTestDrive;

/******************************************************************************

 * Class Definitions

 ******************************************************************************/

class CFHServer : public CServer2

	{

public:

	static CFHServer* NewL();

	void ConstructL();

	virtual ~CFHServer();

	virtual CSession2* NewSessionL(const TVersion& aVersion, const RMessage2& aMessage) const;

	virtual TInt RunError(TInt aError);

private:

	CFHServer();

	};

class CFHSession : public CSession2

	{

public:

	virtual ~CFHSession();

	virtual void CreateL();

	virtual void ServiceL(const RMessage2& aMessage);

public:

	void GetFileHandle(const RMessage2& aMsg);

	void GetFileHandle2(const RMessage2& aMsg);

	void PassFileHandle(const RMessage2& aMsg);

	void PassFileHandleProcess(const RMessage2& aMsg);

	void PassInvalidFileHandle(const RMessage2& aMsg);

	};

// a second server so we can test passing file handles from 

// client to server to server2

class CFHServer2 : public CServer2

	{

public:

	static CFHServer2* NewL();

	void ConstructL();

	virtual ~CFHServer2();

	virtual CSession2* NewSessionL(const TVersion& aVersion, const RMessage2& aMessage) const;

	virtual TInt RunError(TInt aError);

private:

	CFHServer2();

public:

	};

class CFHSession2 : public CSession2

	{

public:

	virtual ~CFHSession2();

	virtual void CreateL();

	virtual void ServiceL(const RMessage2& aMessage);

public:

	void PassFileHandle(const RMessage2& aMsg);

	void GetFileHandle(const RMessage2& aMsg);

	};

/******************************************************************************

 * Class CFHSession/CFHServer

 ******************************************************************************/

void ExceptionHandler(TExcType)

	{

	User::Leave(KErrGeneral);

	}

CFHSession::~CFHSession()

	{

	}

void CFHSession::CreateL()

	{

	}

void CFHSession::ServiceL(const RMessage2& aMessage)

	{

	__UHEAP_MARK;

	TInt mid=aMessage.Function();

	switch(mid)

		{

		case RFileHandleSharer::EMsgGetFileHandle:

			GetFileHandle(aMessage);

			break;

		case RFileHandleSharer::EMsgGetFileHandle2:

			GetFileHandle2(aMessage);

			break;

		case RFileHandleSharer::EMsgPassFileHandle:

			PassFileHandle(aMessage);

			break;

		case RFileHandleSharer::EMsgPassFileHandleProcess:

			PassFileHandleProcess(aMessage);

			break;

		case RFileHandleSharer::EMsgPassInvalidFileHandle:

			PassInvalidFileHandle(aMessage);

			break;

		case RFileHandleSharer::EMsgExit:

			{

			// stop server2

			RFileHandleSharer2 handsvr2;

			TInt r=handsvr2.Connect();

			test(r==KErrNone);

			r = handsvr2.Exit();

			test(r==KErrNone || r == KErrServerTerminated);

			handsvr2.Close();

			aMessage.Complete(KErrNone);	

			CActiveScheduler::Stop();

			}

			break;

		case RFileHandleSharer::EMsgSync:

			aMessage.Complete(KErrNone);	

			break;

		case RFileHandleSharer::EMsgDrive:

			gTestDrive=aMessage.Int0();

			aMessage.Complete(KErrNone);	

			break;

		default:

			break;

		}

	__UHEAP_MARKEND;

	}

//

// Returns session and relevant file handle to Client in read mode

// This is to allow the client to test the deprcated function RFile::Adopt()

void CFHSession::GetFileHandle(const RMessage2& aMsg)

	{

	test.Printf(_L("Get file handle"));

	// get the requested file mode

	TInt fileMode = aMsg.Int1();

	RFs fs;

	TInt r=fs.Connect();

	r=fs.CreatePrivatePath(gTestDrive);

	test(r==KErrNone);

	r=fs.SetSessionToPrivate(gTestDrive);

	test(r==KErrNone);

	r=fs.ShareProtected();

	test(r==KErrNone);

	RFile file1;

	r=file1.Create(fs,KSvrFileName,EFileWrite);

	test(r==KErrNone || r==KErrAlreadyExists);

	if (r==KErrAlreadyExists)

		{

		r=file1.Open(fs,KSvrFileName, EFileWrite);

		test(r==KErrNone);

		}

	r=file1.Write(KTestData1());

	test(r==KErrNone);

	file1.Close();

	r=file1.Open(fs,KSvrFileName, fileMode);

	test(r==KErrNone);

	TInt fssh=file1.SubSessionHandle();

	r=aMsg.Write(0, TPckgC<TInt>(fssh));

	test(r==KErrNone);

	aMsg.Complete(fs);

	fs.Close();

	}

//

//	Returns a file handle from server2

//

void CFHSession::GetFileHandle2(const RMessage2& aMsg)

	{

	test.Next(_L("RFile::AdoptFromServer()"));

	// pass the request on to FHServer2 - this will create a file

	// which we can then adopt before returning it to the client

	// get the requested file mode

	TInt fileMode = aMsg.Int1();

	RFileHandleSharer2 handsvr2;

	TInt r = handsvr2.Connect();

	test(r==KErrNone);

	TInt ssh;

	TInt fsh = handsvr2.GetFileHandle(ssh, TFileMode(fileMode));

	test(fsh >= 0);

	// adopt the file from FHServer2

	RFile file;

	r=file.AdoptFromServer(fsh, ssh);

	test(r==KErrNone);

	test.Next(_L("RFile::TransferToClient()"));

	// transfer the file to the client

	r = file.TransferToClient(aMsg, 0);

	test(r==KErrNone);

	// test we can still use the file

	TInt pos = 0;

	r = file.Seek(ESeekStart, pos);

	test(r==KErrNone);

	TBuf8<100> rbuf;

	r=file.Read(0,rbuf);

	test(r==KErrNone);

	r=rbuf.CompareF(KTestData1());

	test(r==KErrNone);

	handsvr2.Close();

	file.Close();

	RDebug::Print(_L("completed"));	

	}

void CFHSession::PassFileHandle(const RMessage2& aMsg)

//

// Adopts file from test program and tests what it can and can't do

// Uses new AdoptFromClient() API

//	

	{

	test.Next(_L("RFile::AdoptFromClient()"));

	// connect to FHServer2

	RFileHandleSharer2 handsvr2;

	TInt r = handsvr2.Connect();

	test(r==KErrNone);

	RFile file;

	// Message slot 0 is a RFs handle

	// Message slot 1 is a RFile Subsession handle (RFile::SubSessionHandle())

	r = file.AdoptFromClient(aMsg, 0, 1);

	test(r==KErrNone);

	TBuf8<100> rbuf;

	r=file.Read(0,rbuf);

	test(r==KErrNone);

	r=rbuf.CompareF(KTestData());

	test(r==KErrNone);

	r=file.Write(KTestData1());

	test(r==KErrAccessDenied);

	r=file.ChangeMode(EFileWrite);

	test(r==KErrArgument);

	r=file.Rename(_L("\\newname.txt"));

	test(r==KErrPermissionDenied || r==KErrAccessDenied);

//	should try a delete

	// pass the file handle to FHServer2

	test.Next(_L("RFile::TransferToServer()"));

	TIpcArgs ipcArgs;

	file.TransferToServer(ipcArgs, 0, 1);

	r = handsvr2.PassFileHandle(ipcArgs);

	test(r==KErrNone);

	TInt pos = 0;

	r = file.Seek(ESeekStart, pos);

	test(r==KErrNone);

	r=file.Read(0,rbuf);

	test(r==KErrNone);

	r=rbuf.CompareF(KTestData());

	test(r==KErrNone);

	file.Close();

	handsvr2.Close();

	aMsg.Complete(KErrNone);

	}

//

// Adopts file from test program and tests what it can and can't do

// Uses new AdoptFromCreator() API

//	

void CFHSession::PassFileHandleProcess(const RMessage2& aMsg)

	{

	test.Next(_L("RFile::AdoptFromCreator()"));

	RFile file;

	TInt r = file.AdoptFromCreator(1, 2);

	test(r == KErrNone);

	TBuf8<100> rbuf;

	r=file.Read(0,rbuf);

	test(r==KErrNone);

	r=rbuf.CompareF(KTestData());

	test(r==KErrNone);

	test.Next(_L("RFile::Rename()"));

	// define a filename in our private path

	RFs fs;

	r=fs.Connect();

	test(r==KErrNone);

	TFileName sessionp;

	fs.SessionPath(sessionp);

	r = fs.MkDirAll(sessionp);

	test(r==KErrNone || r==KErrAlreadyExists);

	r=fs.ShareProtected();

	test(r==KErrNone);

	r=fs.CreatePrivatePath(gTestDrive);

	test(r==KErrNone);

	r=fs.SetSessionToPrivate(gTestDrive);

	test(r==KErrNone);

	TPath newPath;

	fs.PrivatePath(newPath);

	TFileName newFileName;

	newFileName = newPath;

	newFileName.Append(_L("newname.txt"));

	// delete the file before we try to rename anything to it

	r = fs.Delete(newFileName);

	test(r == KErrNone || r == KErrNotFound);

	r=file.Rename(newFileName);

	test(r==KErrNone);

	file.Close();

	// Next verify that we can delete the file (which should now 

	// have been moved to our private directory)

	test.Next(_L("RFs::Delete()"));

	r = fs.Delete(newFileName);

	test(r == KErrNone);

	fs.Close();

	aMsg.Complete(KErrNone);

	}

void CFHSession::PassInvalidFileHandle(const RMessage2& aMsg)

//

// Attempts to adopt an invalid file handle from test program 

// and tests that KErrBadHandle is returned by AdoptFromClient()

//	

	{

	test.Next(_L("PassInvalidFileHandle - RFile::AdoptFromClient()"));

	RFile file;

	// Message slot 0 is a RFs handle

	// Message slot 1 is a RFile Subsession handle (RFile::SubSessionHandle())

	TInt r = file.AdoptFromClient(aMsg, 0, 1);

	test(r==KErrBadHandle);

	aMsg.Complete(r);

	}

CFHServer* CFHServer::NewL()

	{

	CFHServer* server = new (ELeave) CFHServer;

	CleanupStack::PushL(server);

	server->ConstructL();

	CleanupStack::Pop(server);

	return server;

	}

void CFHServer::ConstructL()

	{

	}

CFHServer::CFHServer()

	: CServer2(0,ESharableSessions)

	{

	}

CFHServer::~CFHServer()

	{

	}

CSession2* CFHServer::NewSessionL(const TVersion& aVersion, const RMessage2&) const

//

//	 Create New Session

//

	{

	(void)aVersion;

	CFHSession* s = new (ELeave) CFHSession;

	return s;

	}

_LIT(KErr,"FHSERVER_ERR");

CFHServer2* CFHServer2::NewL()

	{

	CFHServer2* server = new (ELeave) CFHServer2;

	CleanupStack::PushL(server);

	server->ConstructL();

	CleanupStack::Pop(server);

	return server;

	}

void CFHServer2::ConstructL()

	{

	}

TInt CFHServer::RunError(TInt aError)

	{

	User::Panic(KErr,aError);

	return 0;

	}

// File handle server #2

CFHServer2::CFHServer2()

	: CServer2(0,ESharableSessions)

	{

	}

CFHServer2::~CFHServer2()

	{

	}

CSession2* CFHServer2::NewSessionL(const TVersion& aVersion, const RMessage2&) const

//

//	 Create New Session

//

	{

	(void)aVersion;

	CFHSession2* s = new (ELeave) CFHSession2;

	return s;

	}

_LIT(KErr2,"FHSERVER2_ERR");

TInt CFHServer2::RunError(TInt aError)

	{

	User::Panic(KErr2,aError);

	return 0;

	}

CFHSession2::~CFHSession2()

	{

	}

void CFHSession2::CreateL()

	{

	}

void CFHSession2::ServiceL(const RMessage2& aMessage)

	{

	__UHEAP_MARK;

	TInt mid=aMessage.Function();

	switch(mid)

		{

		case RFileHandleSharer::EMsgPassFileHandle:

			PassFileHandle(aMessage);

			break;

		case RFileHandleSharer::EMsgGetFileHandle:

			GetFileHandle(aMessage);

			break;

		case RFileHandleSharer::EMsgExit:

			aMessage.Complete(KErrNone);	

			CActiveScheduler::Stop();

			break;

		default:

			break;

		}

	__UHEAP_MARKEND;

	}

//

// Adopts file from server 1

//	

void CFHSession2::PassFileHandle(const RMessage2& aMsg)

	{

	RFile file;

	// Message slot 0 is a RFs handle

	// Message slot 1 is a RFile Subsession handle (RFile::SubSessionHandle())

	TInt r = file.AdoptFromClient(aMsg, 0, 1);

	if (r != KErrNone)

		{

		aMsg.Complete(r);

		return;

		}

	TBuf8<100> rbuf;

	if (r == KErrNone)

		r=file.Read(0,rbuf);

	if (r == KErrNone)

		r = rbuf.CompareF(KTestData());

	if (r == KErrNone)

		{

		r = file.Write(KTestData1());

		if (r == KErrAccessDenied)

			r = KErrNone;

		}

	if (r == KErrNone)

		{

		r = file.ChangeMode(EFileWrite);

		if (r == KErrArgument)

			r = KErrNone;

		}

	if (r == KErrNone)

		{

		r = file.Rename(_L("\\newname.txt"));

		if (r == KErrPermissionDenied || r == KErrAccessDenied)

			r = KErrNone;

		}

	file.Close();

	aMsg.Complete(r);

	}

void CFHSession2::GetFileHandle(const RMessage2& aMsg)

//

//	Returns a file handle in write mode

//

	{

	RFs fs;

	TInt r = fs.Connect();

	if (r == KErrNone)

		r = fs.CreatePrivatePath(gTestDrive);

	if (r == KErrNone)

		r = fs.SetSessionToPrivate(gTestDrive);

	if (r == KErrNone)

		r = fs.ShareProtected();

	// make sure file exists & has valid data in it

	RFile file1;

	if (r == KErrNone)

		r = file1.Replace(fs,KSvrFileName,EFileWrite);

	if (r == KErrNone)

		r = file1.Write(KTestData1());

	file1.Close();

	// re-open the file with the mode the client has requested & pass it to the client

	TFileMode fileMode = TFileMode(aMsg.Int1());

	if (r == KErrNone)

		r = file1.Open(fs,KSvrFileName, fileMode);

	if (r == KErrNone)

		r = file1.TransferToClient(aMsg, 0);

	file1.Close();

	fs.Close();

	if (r != KErrNone)

		aMsg.Complete(r);

	}

LOCAL_C TInt FHServer2(TAny * /*anArg*/)

	{	

	RTest test(_L("FHServer2"));

	// Remember the number of open handles. Just for a sanity check ....

	TInt start_thc, start_phc;

	RThread().HandleCount(start_phc, start_thc);

	CTrapCleanup* cleanup;

	cleanup=CTrapCleanup::New();

	CActiveScheduler* sched=new CActiveScheduler;

	FHS_ASSERT(sched);

	CActiveScheduler::Install(sched);

	CFHServer2* svr2 = NULL;

	TRAP_IGNORE(svr2 = CFHServer2::NewL());

	FHS_ASSERT(svr2);

	FHS_ASSERT(svr2->Start(_L("FHServer2"))==KErrNone);

	CActiveScheduler::Start();

	delete svr2;

	delete sched;

	delete cleanup;

	// Sanity check for open handles

	TInt end_thc, end_phc;

	RThread().HandleCount(end_phc, end_thc);

	test(start_thc == end_thc);

	test(start_phc == end_phc);

	// and also for pending requests ...

	test(RThread().RequestCount() == 0);

	return KErrNone;

	}

GLDEF_C TInt E32Main()

//

// Test Server for file handle sharing

//

	{

	test.Title();

	test.Start(_L("Starting FHServer..."));

	// Remember the number of open handles. Just for a sanity check ....

	TInt start_thc, start_phc;

	RThread().HandleCount(start_phc, start_thc);

	CTrapCleanup* cleanup=CTrapCleanup::New();

	FHS_ASSERT(cleanup);

	CActiveScheduler* sched=new CActiveScheduler;

	FHS_ASSERT(sched);

	CActiveScheduler::Install(sched);

	// start server1

	CFHServer* svr = NULL;

	TRAP_IGNORE(svr = CFHServer::NewL());

	FHS_ASSERT(svr);

	FHS_ASSERT(svr->Start(_L("FHServer"))==KErrNone);

	test.Title();

	test.Start(_L("Starting tests..."));

	// start server2 in a seperate thread

	RThread server2Thread;

	TInt r = server2Thread.Create(_L("FHServer2"), FHServer2, KDefaultStackSize, KHeapSize, KHeapSize, NULL);	

	test(r==KErrNone);	

	TRequestStatus statq;

	server2Thread.Logon(statq);

	server2Thread.Resume();

	CActiveScheduler::Start();

	// wait for server2's thread to end gracefully

	User::WaitForRequest(statq);

	server2Thread.Close();

	RFs cleanupfs;

	r = cleanupfs.Connect();

	test(r==KErrNone);

	r=cleanupfs.SetSessionToPrivate(gTestDrive);

	test(r==KErrNone);

	r=cleanupfs.Delete(KSvrFileName);

	test(r==KErrNone || r==KErrNotFound);

	cleanupfs.Close();

	test.End();

	delete svr;

	delete sched;

	delete cleanup;

	// Sanity check for open handles and pending requests

	TInt end_thc, end_phc;

	RThread().HandleCount(end_phc, end_thc);

	test(start_thc == end_thc);

	test(start_phc == end_phc);

	test(RThread().RequestCount() == 0);

	return 0;

	}