// rcomm.cpp

// 

// Copyright (c) 2007 - 2010 Accenture. All rights reserved.

// This component and the accompanying materials are made available

// under the terms of the "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:

// Accenture - Initial contribution

//

#include <fshell/ioutils.h>

#include <c32comm.h>

using namespace IoUtils;

class CCmdRcomm : public CCommandBase

	{

public:

	static CCommandBase* NewLC();

	~CCmdRcomm();

private:

	CCmdRcomm();

	void ReadL();

	void WriteL();

private: // From CCommandBase.

	virtual const TDesC& Name() const;

	virtual void DoRunL();

	virtual void ArgumentsL(RCommandArgumentList& aArguments);

	virtual void OptionsL(RCommandOptionList& aOptions);

private:

	HBufC* iCsyName;

	HBufC* iPortName;

	HBufC* iMode;

	TCommAccess iCommAccess;

	TUint iBufSize;

	TBool iRaw;

	TBool iVerbose;

	RCommServ iCommServ;

	RComm iComm;

	};

CCommandBase* CCmdRcomm::NewLC()

	{

	CCmdRcomm* self = new(ELeave) CCmdRcomm();

	CleanupStack::PushL(self);

	self->BaseConstructL();

	return self;

	}

CCmdRcomm::~CCmdRcomm()

	{

	delete iCsyName;

	delete iPortName;

	delete iMode;

	iComm.Close();

	iCommServ.Close();

	}

CCmdRcomm::CCmdRcomm()

	: iCommAccess(ECommShared), iBufSize(1024)

	{

	}

const TDesC& CCmdRcomm::Name() const

	{

	_LIT(KName, "rcomm");	

	return KName;

	}

void CCmdRcomm::DoRunL()

	{

	User::LeaveIfError(iCommServ.Connect());

	if (iCsyName)

		{

		TInt err = iCommServ.LoadCommModule(*iCsyName);

		if (err && (err != KErrAlreadyExists))

			{

			PrintError(err, _L("Unable to load comm module \'%S\'"), iCsyName);

			User::Leave(err);

			}

		}

	else

		{

		TInt numPorts;

		User::LeaveIfError(iCommServ.NumPorts(numPorts));

		for (TInt i = 0; i < numPorts; ++i)

			{

			TName moduleName;

			TSerialInfo serialInfo;

			TInt err = iCommServ.GetPortInfo(i, moduleName, serialInfo);

			if (err == KErrNone)

				{

				Printf(_L("%S:\r\n\tPort name: \'%S\'\r\n\tLow unit: %d\r\n\tHigh unit: %d\r\n\tDescription: %S\r\n\r\n"), &moduleName, &serialInfo.iName, serialInfo.iLowUnit, serialInfo.iHighUnit, &serialInfo.iDescription);

				}

			else

				{

				PrintWarning(_L("Couldn't get info for port %d, err=%d"), i, err);

				}

			}

		return;

		}

	if (iPortName)

		{

		TInt err = iComm.Open(iCommServ, *iPortName, iCommAccess);

		LeaveIfErr(err, _L("Unable to open port \'%S\'"), iPortName);

		}

	else

		{

		TSerialInfo serialInfo;

		_LIT(KCsy, ".csy");

		TPtrC csyName(*iCsyName);

		if (csyName.Right(KCsy().Length()).CompareF(KCsy) == 0)

			{

			// Remove the trailing '.csy' because otherwise C32 won't find the module.

			csyName.Set(csyName.Left(csyName.Length() - KCsy().Length()));

			}

		TInt err = iCommServ.GetPortInfo(csyName, serialInfo);

		LeaveIfErr(err, _L("Couldn't get port info for CSY '%S'"), &csyName);

		Printf(_L("Port name: \'%S\'\r\nLow unit: %d\r\nHigh unit: %d\r\nDescription: %S\r\n"), &serialInfo.iName, serialInfo.iLowUnit, serialInfo.iHighUnit, &serialInfo.iDescription);

		return;

		}

	_LIT(KModeRead, "read");

	_LIT(KModeWrite, "write");

	if (iMode)

		{

		if (*iMode == KModeRead)

			{

			ReadL();

			}

		else if (*iMode == KModeWrite)

			{

			WriteL();

			}

		else

			{

			PrintError(KErrArgument, _L("Invalid mode: \'%S\'"), iMode);

			}

		}

	}

void CCmdRcomm::ReadL()

	{

	if (iRaw)

		{

		HBufC8* readBuf = HBufC8::NewLC(iBufSize);

		TPtr8 readBufPtr(readBuf->Des());

		TInt err = KErrNone;

		while (err == KErrNone)

			{

			TRequestStatus status;

			iComm.ReadOneOrMore(status, readBufPtr);

			User::WaitForRequest(status);

			err = status.Int();

			if (err == KErrNone)

				{

				TPtrC writePtr((TUint16*)readBuf->Ptr(), readBuf->Length() / 2);

				Write(writePtr);

				}

			else if (iVerbose)

				{

				PrintWarning(_L("RComm::ReadOneOrMore failed with %d"), err);

				}

			}

		CleanupStack::PopAndDestroy(readBuf);

		}

	else

		{

		HBufC8* readBuf = HBufC8::NewLC(iBufSize);

		TPtr8 readBufPtr(readBuf->Des());

		HBufC* writeBuf = HBufC::NewLC(iBufSize / 2);

		TPtr writeBufPtr(writeBuf->Des());

		TInt err = KErrNone;

		while (err == KErrNone)

			{

			TRequestStatus status;

			iComm.ReadOneOrMore(status, readBufPtr);

			User::WaitForRequest(status);

			err = status.Int();

			if (err == KErrNone)

				{

				writeBufPtr.Copy(*readBuf);

				Write(*writeBuf);

				}

			else if (iVerbose)

				{

				PrintWarning(_L("RComm::ReadOneOrMore failed with %d"), err);

				}

			}

		CleanupStack::PopAndDestroy(2, readBuf);

		}

	}

void CCmdRcomm::WriteL()

	{

	RIoConsoleReadHandle& stdin = Stdin();

	User::LeaveIfError(stdin.SetReadMode(RIoReadHandle::EOneOrMore));

	if (iRaw)

		{

		HBufC* readBuf = HBufC::NewLC(iBufSize / 2);

		TPtr readBufPtr(readBuf->Des());

		TInt err = KErrNone;

		while (err == KErrNone)

			{

			err = stdin.Read(readBufPtr);

			if (err == KErrNone)

				{

				TPtrC8 writePtr((TUint8*)readBuf->Ptr(), readBuf->Size());

				TRequestStatus status;

				iComm.Write(status, writePtr);

				User::WaitForRequest(status);

				LeaveIfErr(status.Int(), _L("RComm::Write failed with %d"), status.Int());

				}

			}

		CleanupStack::PopAndDestroy(readBuf);

		}

	else

		{

		HBufC* readBuf = HBufC::NewLC(iBufSize / 2);

		TPtr readBufPtr(readBuf->Des());

		HBufC8* writeBuf = HBufC8::NewLC(iBufSize);

		TPtr8 writeBufPtr(writeBuf->Des());

		TInt err = KErrNone;

		while (err == KErrNone)

			{

			err = stdin.Read(readBufPtr);

			if (err == KErrNone)

				{

				writeBufPtr.Copy(*readBuf);

				TRequestStatus status;

				iComm.Write(status, *writeBuf);

				User::WaitForRequest(status);

				LeaveIfErr(status.Int(), _L("RComm::Write failed with %d"), status.Int());

				}

			}

		CleanupStack::PopAndDestroy(2, readBuf);

		}

	}

void CCmdRcomm::ArgumentsL(RCommandArgumentList& aArguments)

	{

	_LIT(KArg1, "csy_name");

	aArguments.AppendStringL(iCsyName, KArg1);

	_LIT(KArg2, "port_name");

	aArguments.AppendStringL(iPortName, KArg2);

	_LIT(KArg3, "mode");

	aArguments.AppendStringL(iMode, KArg3);

	}

void CCmdRcomm::OptionsL(RCommandOptionList& aOptions)

	{

	_LIT(KOption1, "access");

	aOptions.AppendUintL((TUint&)iCommAccess, KOption1);

	_LIT(KOption2, "buffer");

	aOptions.AppendUintL(iBufSize, KOption2);

	_LIT(KOption3, "raw");

	aOptions.AppendBoolL(iRaw, KOption3);

	_LIT(KOption4, "verbose");

	aOptions.AppendBoolL(iVerbose, KOption4);

	}

EXE_BOILER_PLATE(CCmdRcomm)