// Copyright (c) 2004-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:
// Contains declaration of CTestStepSocket class that is
// base class for all test step classes in this test.
// 
//

#if (!defined __TESTSTEPSOCKET_H__)
#define __TESTSTEPSOCKET_H__
#include <in_sock.h>
#include <test/testexecutestepbase.h>


class CTestStepSocket : public CTestStep
	{
	public:
		CTestStepSocket();
		~CTestStepSocket();
		
	struct TSocketThreadArg
		{
		CTestStepSocket* iHandle;
		RSemaphore* iSem;
		TUint iNumSockets;
		};
	
	protected:
			
		const TPtrC& EpocErrorToText(const TInt aErrCode);
		enum TVerdict doTestStepL();
		virtual enum TVerdict InternalDoTestStepL() =0;
		
		// test threads
		static TInt ConnectThread(TAny * anArg);
		static TInt SocketThread(TAny * anArg);
		static TInt BadHandleThread(TAny* anArg);
		static TInt ReadTwiceThread(TAny* anArg);
		static TInt BadDescriptorThread(TAny* anArg);
		
		static TInt OptimalConnect(RSocketServ &aSrv);

		// Helpers for test threads
		static void DoCreateSubsessionsL(CTestStepSocket *aTestStep, TInt aNumSubsess);
		
		static const TUint KNumTestSockets;
		static const TInt KNumExhaustiveSockets;
		static const TInt KNumStretchOpens;
		
	private:
		TPtrC iErrText;
		TBuf<16> iErrBuf;		// size must at least satisfy the number of digits used to represent a TInt
	};
	

class CSocketTestStep_OOMCapable : public CTestStepSocket
	{
public:
	~CSocketTestStep_OOMCapable()
		{
		iOOMSess.Close();
		}
		
	void SetHeapFailNext(TInt aFailureIndex)
		{
		if(iOOMSess.Handle() == KNullHandle)
			{
			iOOMSess.Connect();
			}
		if(iOOMSess.Handle() != KNullHandle)
			{
			iOOMSess.__DbgFailNext(aFailureIndex - 1);	// want to start with 0th
			}
		}
		
	void ResetHeapFailNext()
		{
		if(iOOMSess.Handle() != KNullHandle)
			{
			iOOMSess.__DbgFailNext(-1);
			}
		}
		
	TBool CheckForHeapFailNext();
	virtual enum TVerdict InternalDoTestStepL() =0;
private:
	RSocketServ iOOMSess;
	};



#endif /* __TESTSTEPSOCKET_H__ */