// Copyright (c) 2007-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:
// @file basetestcase.cpp
// @internalComponent
// 
//

#include "BaseTestCase.h"
#include <e32ver.h>
#include <d32usbdi.h>
#include "testdebug.h"
#include "testpolicy.h"

namespace NUnitTesting_USBDI
	{
	
	
const TUint8 KEpDirectionIn = 0x80;
const TUint8 KEpDirectionOut = 0x00;
const TUint8 KTransferTypeControl = 0x00;
const TUint8 KTransferTypeIsoch = 0x01;
const TUint8 KTransferTypeBulk = 0x02;
const TUint8 KTransferTypeInterrupt = 0x03;	

const TUint8 KChunkSize  		= 0x80 ; // 128 bytes
const TUint KTreeBufferSize 	= 32*1024 ; // 32k bytes

_LIT(KRefPath, "Z:\\scripts\\");
 _LIT(KGeneratedFilesPath,"C:\\");
_LIT(KExtensionFile,".txt"); 


CBaseTestCase::CBaseTestCase(const TDesC& aTestCaseId,TBool aHostFlag, TBool aHostOnly)
:	CActive(EPriorityStandard),
	iHost(aHostFlag),
	iHostOnly(aHostOnly)
	{
	iTestCaseId.Copy(aTestCaseId);
	CActiveScheduler::Add(this);
	}
	
void CBaseTestCase::BaseConstructL()
	{
	LOG_FUNC	
	RDebug::Printf("Creating test case timer");
	TInt err(iTimer.CreateLocal());
	if(err == KErrNone)
		{
		RDebug::Printf("Test case timer created");
		}
	else
		{
		RDebug::Printf("<Error %d> Test case timer could not be created",err);
		User::Leave(err);
		}	
	}

void CBaseTestCase::TimeoutIn(TInt aTimeoutPeriod)
	{
	LOG_FUNC
	
	CancelTimeout();
	iTimer.After(iStatus,aTimeoutPeriod*1000000);
	SetActive();
	}


void CBaseTestCase::CancelTimeout()
	{
	iTimer.Cancel();
	}

TInt CBaseTestCase::GenerateRefFile(const TDesC& aFileName)
	{
	
	LOG_FUNC
	TBuf<256> refTreeFullFileName(KGeneratedFilesPath);
	refTreeFullFileName.Append(aFileName);	
	refTreeFullFileName.Append(KExtensionFile);
	
 	RFile refFile; 
	TInt ret = KErrNone; 
	ret = iFs.Connect();
	if(ret!=KErrNone && ret!=KErrAlreadyExists)
	// if already connected, ignore
		{ 
		RDebug::Printf("iFs.Connect fails, ret = %d", ret);
		return ret;
		}
			
	ret = iFs.Delete(refTreeFullFileName);
	if(ret == KErrNone || ret == KErrNotFound)
		{
		ret = refFile.Create(iFs,refTreeFullFileName,EFileShareAny|EFileWrite);
		}		
	
	if(ret!=KErrNone) 
		{ 
		RDebug::Printf("refFile.Create fails, ret = %d", ret);
		return ret;
		}
		 
	refFile.Write(iTreeBuffer);
	refFile.Flush(); 
	refFile.Close();
			 
	return KErrNone;
	}	

TInt CBaseTestCase::CompareCurrentTreeToRef(const TDesC& aFileName, TBool& aIsIdentical)
	{
	
	LOG_FUNC								
	TBuf<256> refTreeFullFileName(KRefPath);
	refTreeFullFileName.Append(aFileName);
	refTreeFullFileName.Append(KExtensionFile);	 	

	TInt ret = KErrNone; 
	ret = iFs.Connect();
	if(ret!=KErrNone && ret!=KErrAlreadyExists)
	// if already connected, ignore
		{ 
		RDebug::Printf("iFs.Connect fails, ret = %d", ret);
		return ret;
		}

	RFile refFile;
	ret = refFile.Open(iFs,refTreeFullFileName,EFileShareAny|EFileRead);
		
	if(ret!=KErrNone)
		{
		RDebug::Printf("Reference File path: %S", &refTreeFullFileName);
		RDebug::Printf("refFile.Open fails ret = %d", ret);
		return ret;
		}
		
	TInt refFileSize;
	refFile.Size(refFileSize);   
	
	// check size is identical
	if(refFileSize != iTreeBuffer.Size())
		{		
		RDebug::Printf("sizes are NOT identical, refFileSize = %d, iTreeBuffer.Size() = %d ", refFileSize, iTreeBuffer.Size());
		//return KErrGeneral; not an issue, \n encoded differently by perforce... x0D x0A. (x0A only in generated ref file)
		}
		
	// read the file, and put it in a local buffer
	RBuf8 refBuf;
	refBuf.CreateL(refFileSize);
	ret = refFile.Read(0, refBuf, refFileSize);

	if(ret!=KErrNone)
		{
		RDebug::Printf("refFile.Read fails %d", ret);
		return ret;
		}
		
	// find occurences of \n now 
	RBuf8 copyRefBuf;
	copyRefBuf.CreateL(refFileSize);
	
	for(TInt iRefBuffer=0; iRefBuffer < refFileSize; iRefBuffer++)
		{
		if(refBuf[iRefBuffer] == 0x0D && iRefBuffer != refFileSize-1) // not the last byte
			{			
			if(refBuf[iRefBuffer+1] == 0x0A)
				{				
				copyRefBuf.Append(_L8("\n"));
				continue;
				}			
			}
		// previous is 0x0D, skip... 
		if( refBuf[iRefBuffer] == 0x0A && refBuf[iRefBuffer-1] == 0x0D)
			{
			continue;
			}			
		copyRefBuf.AppendFormat(_L8("%c"), refBuf[iRefBuffer]);				  
		}
	refBuf.Close();
	
	RDebug::Printf("copyRefBuf.Size %d", copyRefBuf.Size());
		

	// check size is identical, should be identical now
	if(copyRefBuf.Size() != iTreeBuffer.Size())
		{		
		RDebug::Printf("sizes are NOT identical, copyRefBuf.Size() = %d, iTreeBuffer.Size() = %d ", refFileSize, iTreeBuffer.Size());
		return KErrGeneral;
		}
	
	// now compare the 2 buffers		
    // Can only go as far as the smallest buffer
    TInt bufferSizeToCheck = Min(copyRefBuf.Size(), iTreeBuffer.Size());
	RDebug::Print(_L("bufferSizeToCheck = %d"), bufferSizeToCheck);

	aIsIdentical = ETrue;	
	for(TInt iRefBuffer=0; iRefBuffer < bufferSizeToCheck; iRefBuffer++)
		{
		if(iTreeBuffer[iRefBuffer] != copyRefBuf[iRefBuffer])
			{
			RDebug::Print(_L("Failed buffer comparison at position %d"), iRefBuffer);
            RDebug::Print(_L("Missmatching chars (%d %d) (%c %c)"), iTreeBuffer[iRefBuffer], copyRefBuf[iRefBuffer], iTreeBuffer[iRefBuffer], copyRefBuf[iRefBuffer]);
			aIsIdentical = EFalse;
			break;
			}			 	 	 
		}		

	RDebug::Print(_L("Finished Buffer comparison aIsIdentical=%d"), aIsIdentical); 

 	copyRefBuf.Close();
	
	return KErrNone;	
	}	

CBaseTestCase::~CBaseTestCase()
	{
	LOG_FUNC
	Cancel();
	iTimer.Close();
	iTreeBuffer.Close();
	iFs.Close();
	}

void CBaseTestCase::SelfComplete()
	{
	SelfComplete(KErrNone);
	}

void CBaseTestCase::SelfComplete(TInt aError)
	{
	TRequestStatus* s = &iStatus;
	iStatus = KRequestPending;
	User::RequestComplete(s,aError);
	SetActive();
	}


void CBaseTestCase::DoCancel()
	{
	LOG_FUNC
	iTimer.Cancel();
	if(iHost)
		{
		HostDoCancel();
		}
	else
		{
		DeviceDoCancel();
		}
	}

void CBaseTestCase::RunL()
	{
	if(iHost)
		{
		HostRunL();
		}
	else
		{
		DeviceRunL();
		}
	}

TInt CBaseTestCase::RunError(TInt aError)
	{
	LOG_FUNC
	RDebug::Printf("Test case C%lS::RunL left with %d",&iTestCaseId,aError);
	iTestPolicy->SignalTestComplete(aError);
	return KErrNone;
	}
	
TDesC& CBaseTestCase::TestCaseId()
	{
	return iTestCaseId;
	}
	
	
TInt CBaseTestCase::TestResult() const
	{
	return iTestResult;
	}
	
TBool CBaseTestCase::IsHostOnly() const
	{
	return iHostOnly;
	}
		
TBool CBaseTestCase::IsHost() const
	{
	return iHost;
	}
		
void CBaseTestCase::PerformTestL()
	{
	
	if(iHost)
		{
		iTreeBuffer.CreateL(KTreeBufferSize); //32k
		ExecuteHostTestCaseL();
		}
	else
		{
		ExecuteDeviceTestCaseL();
		}	
	}

void CBaseTestCase::SetTestPolicy(CBasicTestPolicy* aTestPolicy)
	{
	iTestPolicy = aTestPolicy;
	}

void CBaseTestCase::TestFailed(TInt aFailResult)
	{
	LOG_FUNC
	iTestResult = aFailResult;
	if(!iHostOnly)
		{
		RDebug::Printf("CActiveScheduler::Stop CBaseTestCase::TestFailed");
		CActiveScheduler::Stop();
		}		
	}
	
void CBaseTestCase::TestPassed()
	{
	LOG_FUNC
	iTestResult = KErrNone;	
	if(!iHostOnly)
		{
		RDebug::Printf("CActiveScheduler::Stop CBaseTestCase::TestPassed");
		CActiveScheduler::Stop();
		}
	}

CBasicTestPolicy& CBaseTestCase::TestPolicy()
	{
	return *iTestPolicy;
	}
	

/**
Gets the first endpoint address that satisfies the parameters
So caution when there are multiple endpoints on the interface setting
See method below for specifying the endpoint index if more than 
one endpoint of the given type exists on the interface setting
*/
TInt CBaseTestCase::GetEndpointAddress(RUsbInterface& aUsbInterface,TInt aInterfaceSetting,
		TUint8 aTransferType,TUint8 aDirection,TInt& aEndpointAddress)
	{
	LOG_FUNC
	
	return GetEndpointAddress(aUsbInterface, aInterfaceSetting, aTransferType, aDirection, 0, aEndpointAddress);
	}
	
/**
Gets the (aIndex+1)th endpoint address that satisfies the parameters
Allows the specification of the endpoint index (starting from ZERO)if more than 
one endpoint of the given type exists on the interface setting
*/
TInt CBaseTestCase::GetEndpointAddress(RUsbInterface& aUsbInterface,TInt aInterfaceSetting,
		TUint8 aTransferType,TUint8 aDirection,TUint8 aIndex,TInt& aEndpointAddress)
	{
	LOG_FUNC
		
	// Get the interface descriptor
	RDebug::Printf("Getting the interface descriptor for this alternate setting");

	TUsbInterfaceDescriptor alternateInterfaceDescriptor;
	TInt err = aUsbInterface.GetAlternateInterfaceDescriptor(aInterfaceSetting, alternateInterfaceDescriptor);

	if(err)
		{
		RDebug::Printf("<Error %d> Unable to get alternate interface (%d) descriptor",err,aInterfaceSetting);
		return err;
		}

	// Parse the descriptor tree from the interface 	
	RDebug::Printf("Search the child descriptors for matching endpoint attributes");
	
	TUsbGenericDescriptor* descriptor = alternateInterfaceDescriptor.iFirstChild;
	TUint8 indexCount = 0;
	while(descriptor)
		{
		RDebug::Printf("Check descriptor type for endpoint");

		// Cast the descriptor to an endpoint descriptor
		TUsbEndpointDescriptor* endpoint = TUsbEndpointDescriptor::Cast(descriptor);
		
		if(endpoint)
			{
			RDebug::Printf("Match attributes for transfer type");
			
			if( (endpoint->Attributes() & aTransferType) == aTransferType)
				{
				RDebug::Printf("Match attributes for endpoint direction");
				
				if( (endpoint->EndpointAddress() & aDirection) == aDirection) 
					{
					if(indexCount==aIndex)
						{
						aEndpointAddress = endpoint->EndpointAddress();
						RDebug::Printf("Endpoint address found");
						return KErrNone;
						}
					else
						{
						indexCount++;
						}
					}
				}
			}

		descriptor = descriptor->iNextPeer;
		}

	// Unable to find the endpoint address	
	RDebug::Printf("Unable to find endpoint address matching the specified attributes");
	
	return KErrNotFound;
	}
	
/*static*/ void CBaseTestCase::LogWithCondAndInfo(const TDesC& aCondition, const TDesC& aFileName, TInt aLine)
	{
	TBuf<256> buf;
 	buf.Format(KFailText, &aCondition, &aFileName, aLine);
 	RDebug::Print(buf); 
 	} 	
 
  	
/*static*/ void CBaseTestCase::PrintAndStoreTree(TUsbGenericDescriptor& aDesc, TInt aDepth)
	{ 
	
	TBuf8<20> buf;	
	for(TInt depth=aDepth;depth>=0;--depth)
		{
		buf.Append(_L8("  "));
		}
		
	//##==TBuf16<40> unicodeBuf;
	TBuf8<40> unicodeBuf;
	unicodeBuf.Copy(buf);	// Ideally this needs conversion to UNICODE
	if(aDesc.iRecognisedAndParsed == TUsbGenericDescriptor::ERecognised)
		{ 
		RDebug::Printf("%S+ length=%d, type=0x%02x", &unicodeBuf, aDesc.ibLength, aDesc.ibDescriptorType);
   		iTreeBuffer.AppendFormat(_L8("%S+ length=%d, type=0x%02x\n"), &buf, aDesc.ibLength, aDesc.ibDescriptorType);		
		}
	else
		{
		RDebug::Printf("%S- length=%d, type=0x%02x", &unicodeBuf, aDesc.ibLength, aDesc.ibDescriptorType);
		iTreeBuffer.AppendFormat(_L8("%S- length=%d, type=0x%02x\n"), &buf, aDesc.ibLength, aDesc.ibDescriptorType);
		} 		

		PrintAndStoreBlob(buf ,aDesc.iBlob);		
		
		if(aDesc.iFirstChild)    
		{
		RDebug::Printf("%S \\ ", &unicodeBuf);
		iTreeBuffer.AppendFormat(_L8("%S \\ \n"), &buf);		
		
		PrintAndStoreTree(*(aDesc.iFirstChild), aDepth+1);		
	
		RDebug::Printf("%S / ", &unicodeBuf);
		iTreeBuffer.AppendFormat(_L8("%S / \n"), &buf);
		}
	if(aDesc.iNextPeer)
		{
		PrintAndStoreTree(*(aDesc.iNextPeer), aDepth);
		}		
	} 
	   
void CBaseTestCase::PrintAndStoreBlob(TDes8& aBuf, TPtrC8& aBlob)
	{
	
	HBufC8* chunk = HBufC8::New(KChunkSize);
	
	TUint nbIter = aBlob.Length()/(KChunkSize/2);
	TUint remainderSize = aBlob.Length()%(KChunkSize/2);

	if(nbIter == 0)  
		{
		PrintAndStoreChunk(chunk, aBlob.Length() ,aBlob, 0, 0, aBuf );      
		}
	else
		{
		// print chunks
		TUint offset = 0;
		TInt i = 0;
		for(i=0;i<nbIter;++i)
			{
			PrintAndStoreChunk(chunk, (KChunkSize/2) ,aBlob, offset, i, aBuf); 
			offset+=(KChunkSize/2);
			} 
		// remainder
		PrintAndStoreChunk(chunk, remainderSize ,aBlob,offset, i ,aBuf);				
		}
	delete chunk;
	} 
	
void CBaseTestCase::PrintAndStoreChunk(HBufC8* aChunk, TUint aSize, TPtrC8& aBlob, TUint aOffset, TUint aIter, TDes8& aBuf)
	{	
	for(TInt i=0;i<aSize;++i)
		{
		aChunk->Des().AppendFormat(_L8("%02x"), aBlob[i+aOffset]);
		}
		
	TBuf16<40> unicodeBuf;	
	unicodeBuf.Copy(aBuf);
	TBuf16<256> unicodeChunk;	
	unicodeChunk.Copy(aChunk->Des());
			
	if(aIter ==0)
		{		
		RDebug::Printf("%S >%S", &unicodeBuf, &unicodeChunk);					
		iTreeBuffer.AppendFormat(_L8("%S >%S\n"), &aBuf, aChunk);	
		}
	else
		{	
		RDebug::Printf("%S  %S\n", &unicodeBuf, &unicodeChunk); 
		iTreeBuffer.AppendFormat(_L8("%S  %S\n"), &aBuf, aChunk);
		}
	aChunk->Des().Zero();		
	}	
	
TInt CBaseTestCase::CheckTree(TUsbGenericDescriptor& aDevDesc, TUsbGenericDescriptor& aConfigDesc, const TDesC& aFileName)
	{
	LOG_FUNC
	TInt ret = KErrNone;
	
	// flush buffer
	iTreeBuffer.Zero();
	
	// print and store tree from aDevDesc & aConfigDesc
	PrintAndStoreTree(aDevDesc);
	PrintAndStoreTree(aConfigDesc);
	
	// generate file if needed	
	#ifdef GENERATE_TREES
	GenerateRefFile(aFileName);	
	#endif // GENERATE_TREES
	
	// compare tree to ref.
	TBool isIdentical;
	if(KErrNone != CompareCurrentTreeToRef(aFileName, isIdentical))
		{ 
		RDebug::Printf("CompareCurrentTreeToRef error"); 
		ret = KErrGeneral;
		}	
	if(!isIdentical)
		{ 
		RDebug::Printf("!isIdentical"); 
		ret = KErrGeneral;
		}
	return ret;
	}
	
	
TInt CBaseTestCase::ParseConfigDescriptorAndCheckTree(TUsbDeviceDescriptor *devDesc, const TDesC8& configSet, TUint indexTest)
	{
	LOG_FUNC
	// Parse config. descriptor
	TUsbGenericDescriptor* parsed = NULL;
	TInt err = UsbDescriptorParser::Parse(configSet, parsed);
	if(err != KErrNone)
		{
		RDebug::Printf("parsing error : UsbDescriptorParser::Parse"); 
		return err;
		}
	TUsbConfigurationDescriptor* configDesc = TUsbConfigurationDescriptor::Cast(parsed);
	// checks 
	if(configDesc == 0)
		{
		RDebug::Printf("configDesc == 0");
		return KErrGeneral; 
		}
		
	// checking tree 
	TBuf<KMaxName> fname(iTestCaseId);
	fname.AppendFormat(_L("_%d"), indexTest);
	return CheckTree(*devDesc, *configDesc, fname); 
	}	
	
TInt CBaseTestCase::CheckTreeAfterDeviceInsertion(CUsbTestDevice& aTestDevice, const TDesC& aFileName)
	{
	LOG_FUNC
	TUsbGenericDescriptor deviceDesc = aTestDevice.DeviceDescriptor();
	TUsbGenericDescriptor configDesc = aTestDevice.ConfigurationDescriptor();	
	return CheckTree(deviceDesc, configDesc, aFileName); 	
	}	
	
	}