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

//

#include <e32std.h>

#include "POPS.H"		//CImPop3Session

#include "POPSOP.H"		//CImPop3Operation

#include "POPS.PAN"		//IMPP panic codes

#include <imcvrecv.h>	//CImRecvConvert

#include <imcvutil.h>   //ImCheckDiskSpace utility class

#include <msvapi.h>     //MessageServer utility class

const TInt KTOPRequestSize = 18; // i.e. 'TOP nnnn 1\r\n' where nnnn can be a 32 bit integer

const TInt KTOPBatchLimit = 100; // Pipelining limit of 100 TOP requests at a time

// string constants used whilst parsing response from POP3 email server

_LIT8(KAuthMatchString,"*SASL *");

_LIT8(KAuthPlainMatchString,"* PLAIN*");

_LIT8(KAuthLoginMatchString,"* LOGIN*");

_LIT8(KAuthCramMD5MatchString,"* CRAM-MD5*");

// Special case for non-compliant Domino servers 

_LIT8(KAuthDominoMatchString,"*AUTH=*"); 

_LIT8(KAuthDominoPlainMatchString,"*=PLAIN*"); 

_LIT8(KAuthDominoLoginMatchString,"*=LOGIN*"); 

_LIT8(KAuthDominoCramMD5MatchString,"*=CRAM-MD5*"); 

// Utility functions declared in IMPP.CPP

GLREF_C TBool CommandAccepted(TDesC8& aResponse);

// IMPP Panic function

GLREF_C void Panic(TPopsPanic aPanic);

//

// CImPop3Operation - an Abstract base for all the specific POP3 operations

//

CImPop3Operation::CImPop3Operation()

	: CMsgActive(KImPopSessionPriority),iFullStopTerminator(_S8(".\r\n"))

	{

	_LIT(KPop3Operation,"CImPop3Operation");

	__DECLARE_NAME(KPop3Operation());

	}

CImPop3Operation::~CImPop3Operation()

	{

	Cancel();

	delete iImId;

	}

//

// 2nd stage of construction

//

void CImPop3Operation::Construct(CImPop3Session *aPopSession)

	{

	iPopSession=aPopSession;

	iTextServer=aPopSession->TextServerSession();

	CActiveScheduler::Add(this); 

	}

void CImPop3Operation::Construct(CImPop3Session *aPopSession, CImRecvConvert* aRecvConverter)

	{

	Construct(aPopSession);

	iRecvConverter=aRecvConverter;

	}

void CImPop3Operation::DoRunL()

	{

	iPopSession->SetOpNotPending();

	__ASSERT_ALWAYS(iTextServer->GetCurrentTextLine(iTextServerResponse)==ECRLFTerminated,User::Leave(EImppBufferNotTerminated));

	if(CommandAccepted(iTextServerResponse)==EFalse)

		{

		// Message has disapeared already

		}

	}

void CImPop3Operation::DoCancel()

	{

	iTextServer->Cancel();

	if (iRecvConverter != NULL)

		{

		iRecvConverter->Cancel();

		}

	CMsgActive::DoCancel();

	}

//

// Queue a read operation to get next line of text

//

void CImPop3Operation::QueueReadNext()

	{

	iTextServer->QueueReceiveNextTextLine(iStatus);

	SetActive();

	}

//

// Return the Pop3 mailbox index no of a message with give Babel msg id

//

TInt CImPop3Operation::MessageNo(TMsvId aMsgId)

	{

	TInt msgNo=iPopSession->MessageNo(aMsgId);

	return msgNo;

	}

//

// Check if there's a poperation outstainding if so panic

//

void CImPop3Operation::CheckIfPending()

	{

	iPopSession->IsPending() ? Panic(EImppOperationAlreadyPending) : iPopSession->SetPending();

	}

//

// Retrieve a full message or just the specified no of lines

//

void CImPop3Operation::RetrieveMessageL()

	{

	iPopSession->SetOpNotPending();

	iTextServer->GetCurrentTextLine(iTextServerResponse);

	if(iFirstLine)

		{ 

		iParsedHeader=EFalse;

		iCommandAccepted=CommandAccepted(iTextServerResponse);

		iFirstLine=EFalse;

		if(!iCommandAccepted)

			{

			return;

			}

		}

	else

		{ 

		TBool fakeDot=EFalse;

		if(iTextServerResponse.Compare(iFullStopTerminator)!=0)  // header + first n lines last line is a full stop

			{

			RemoveFirstDot();

			if (iRecvConverter->ParseNextField( iTextServerResponse ) == KErrNoMemory)

				{

				User::Leave(KErrNoMemory);

				}

			}

		else if(iParsedHeader==EFalse)

			{

			RemoveFirstDot();

			_LIT8(KCrLfStr,"\r\n");

			if (iRecvConverter->ParseNextField( KCrLfStr ) == KErrNoMemory)

				{

				User::Leave(KErrNoMemory);

				}    

			fakeDot=ETrue;

			}

		else

			return;

		// test for comp header

		if(iRecvConverter->ValidCompleteHeader())

			{

			iParsedHeader=ETrue;

			// test for valid headers

			const CImHeader& header = iRecvConverter->Header();

			iEmptyHeaders=(   (header.From().Length()==0)

							&&(header.ReplyTo().Length()==0)

							&&(header.ToRecipients().Count()==0)

							&&(header.CcRecipients().Count()==0)

							&&(header.BccRecipients().Count()==0));

			// store id of new entry created by IMCV

			iNewEntryId = iRecvConverter->EntryId();

			// store internet msg id

			HBufC8* imId = header.ImMsgId().AllocL();

			delete iImId;

			iImId = NULL;

			iImId = imId;

			}

		if(fakeDot!=EFalse)

			{

			return;

			}

		}

	QueueReadNext();

	}

//

// What to do on completion

//

void CImPop3Operation::DoComplete(TInt& /*aCompleteRequest*/)

	{

	}

TBool CImPop3Operation::PopCommandAccepted()

	{

	return iCommandAccepted;

	}

//

// Remove first dot of a pair

//

void CImPop3Operation::RemoveFirstDot()

	{

	if(iTextServerResponse.Length()>2)

		{

		if( (iTextServerResponse[0]=='.') && (iTextServerResponse[1]=='.'))

			{

			iTextServerResponse.Delete(0,1);

			}

		}

	}

TPtrC8 CImPop3Operation::ImMsgId()

	{ 

	return (iImId ? TPtrC8(*iImId) : TPtrC8()); 

	}

TBool CImPop3Operation::SetMessage(TMsvId anId)

	{

	iMessageNo=MessageNo(anId);

	return iMessageNo == KErrNotFound ? EFalse: ETrue;

	}

//

// CImPop3Stat return no of messages and the mailbox size

//

CImPop3Stat* CImPop3Stat::NewL(CImPop3Session *aPopSession)

	{

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

	self->Construct(aPopSession);

	return self;

	}

void CImPop3Stat::Start(TRequestStatus& aStatus, TUint &aNoMsg, TUint &aMboxSize)

	{

	CheckIfPending();

	iNoMessages=&aNoMsg;

	iMailboxSize=&aMboxSize;

	_LIT8(KStatCrLfStr,"STAT\r\n");

	iTextServer->SendQueueReceive(iStatus, KStatCrLfStr());

	SetActive();

	Queue(aStatus);

	}

// Got a response from the text server

void CImPop3Stat::DoRunL()

	{

	iPopSession->SetOpNotPending();

	__ASSERT_ALWAYS(iTextServer->GetCurrentTextLine(iTextServerResponse)==ECRLFTerminated,User::Leave(EImppBufferNotTerminated));

	if(CommandAccepted(iTextServerResponse))

		{

		TLex8 parseResponse(iTextServerResponse);

		parseResponse.SkipCharacters();

		parseResponse.SkipSpace();

		if(parseResponse.Val(*iNoMessages)==KErrNone)

			{

			parseResponse.SkipSpace();

			parseResponse.Val(*iMailboxSize);

			}

		else

			*iNoMessages=0;

		}

	}

//

// CImPop3List list a specific message no  and size or all messages and sizes

//

CImPop3List* CImPop3List::NewL(CImPop3Session *aPopSession)

	{

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

	self->Construct(aPopSession);

	return self;

	}

void CImPop3List::Start(TRequestStatus& aStatus, TUint* aSize)

	{

	CheckIfPending();

	iSize=aSize;

	iArrayPos=0;

	if(iMessageNo!=0)

		{

		_LIT8(KListArgCrLfStr,"LIST %d\r\n");

		iTextServer->SendQueueReceive(iStatus, KListArgCrLfStr, iMessageNo);

		}

	else

		{

		_LIT8(KListCrLfStr,"LIST\r\n");

		iTextServer->SendQueueReceive(iStatus,KListCrLfStr());

		iFirstLine=ETrue;

		}

	SetActive();

	Queue(aStatus);

	}

void CImPop3List::DoRunL()

	{

	iPopSession->SetOpNotPending();

	__ASSERT_ALWAYS(iTextServer->GetCurrentTextLine(iTextServerResponse)==ECRLFTerminated,User::Leave(EImppBufferNotTerminated));

	TLex8 parseResponse(iTextServerResponse);

	if(iNoResponses > 0)

		{

		if(CommandAccepted(iTextServerResponse))

		User::Leave(KErrNotFound);

		}

	if(iMessageNo!=0)

		{

		if(CommandAccepted(iTextServerResponse))

			{

			parseResponse.SkipCharacters();

			parseResponse.SkipSpace();

			GetIndexAndSize(&parseResponse);

			}

		else

			Panic(EImppBabelMsgDoesNotExist);

		}

	else

		{

		if(iFirstLine)

			{

			if(CommandAccepted(iTextServerResponse))

				{

				iFirstLine=EFalse;

				}

			else

				{

				return;

				}

			}

		else

			{

			if(iTextServerResponse.Compare(iFullStopTerminator)==0)  // list of message info last line is a full stop

				{

				return;

				}

			else

				{

				if(GetIndexAndSize(&parseResponse)==KErrNone)

					{

					iArrayPos++;

					iNoResponses++;

					}

				else

					{

					iSize[iArrayPos] = 0;

					}

				}

			}

		QueueReadNext();

		}

	}

//

// Get size of a specific message (folder view displays message sizes)

//

TInt CImPop3List::GetIndexAndSize(TLex8 *aLex)

	{

	aLex->SkipCharacters();

	aLex->SkipSpace();

	return(aLex->Val(iSize[iArrayPos]));

	}

//

// CImPop3Retr Retrieve a specific message

//

CImPop3Retr* CImPop3Retr::NewL(CImPop3Session *aPopSession, CImRecvConvert* aRecvConverter, RFs& aFs)

	{

	CImPop3Retr* self = new (ELeave) CImPop3Retr(aFs);

	CleanupStack::PushL(self);

	self->ConstructL(aPopSession, aRecvConverter);

	CleanupStack::Pop(self);

	return self;

	}

void CImPop3Retr::ConstructL(CImPop3Session *aPopSession, CImRecvConvert* aRecvConverter)

	{

	Construct(aPopSession,aRecvConverter);

	iCurrentDrive = MessageServer::CurrentDriveL(iFs);

	}

void CImPop3Retr::StartL(TRequestStatus& aStatus)

	{

	CheckIfPending();

	iFirstLine=ETrue;

	iNoBytesRetrieved=0;

	iEmptyHeaders = EFalse;

	ImCheckDiskSpace::LeaveIfLowDiskL(iFs, iCurrentDrive);

	iRecvConverter->ResetL();

	_LIT8(KRetrArgCrLfStr,"RETR %d\r\n");

	iTextServer->SendQueueReceive(iStatus, KRetrArgCrLfStr, iMessageNo);

	SetActive();

	Queue(aStatus);

	}

void CImPop3Retr::DoRunL()

	{

	RetrieveMessageL();

	if(iFirstLine==EFalse)

		{

		iNoBytesRetrieved+=iTextServerResponse.Length();

		}

	}

TUint CImPop3Retr::Progress()

	{

	return iNoBytesRetrieved;

	}

void CImPop3Retr::SetMessageIndex(TInt anIndex)

	{

	iMessageNo=anIndex;

	}

//

// CImPop3Dele Delete a specified message

//

CImPop3Dele* CImPop3Dele::NewL(CImPop3Session *aPopSession)

	{

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

	self->Construct(aPopSession);

	return self;

	}

void CImPop3Dele::Start(TRequestStatus& aStatus)

	{

	CheckIfPending();

	_LIT8(KDeleArgCrLfStr,"DELE %d\r\n");

	iTextServer->SendQueueReceive(iStatus, KDeleArgCrLfStr, iMessageNo);

	SetActive();

	Queue(aStatus);

	}

//

// CImPop3Noop No operation (but I'm still here)

//

CImPop3Noop* CImPop3Noop::NewL(CImPop3Session *aPopSession)

	{

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

	self->Construct(aPopSession);

	return self;

	}

void CImPop3Noop::Start(TRequestStatus& aStatus)

	{

	CheckIfPending();

	_LIT8(KNoopCrLfStr,"NOOP\r\n");

	iTextServer->SendQueueReceive(iStatus, KNoopCrLfStr());

	SetActive();

	Queue(aStatus);

	}

//

// CImPop3Rset Reset

//

CImPop3Rset* CImPop3Rset::NewL(CImPop3Session *aPopSession)

	{

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

	self->Construct(aPopSession);

	return self;

	}

void CImPop3Rset::Start(TRequestStatus& aStatus)

	{

	CheckIfPending();

	_LIT8(KRsetCrLfStr,"RSET\r\n");

	iTextServer->SendQueueReceive(iStatus, KRsetCrLfStr());

	SetActive();

	Queue(aStatus);

	}

//

// Return specified header and top n lines of message.

// If the POP server supports pipelining, TOP requests will be submitted in

// batches. The batch size limit is 100 requests.

//

CImPop3Top* CImPop3Top::NewL(CImPop3Session *aPopSession, CImRecvConvert* aRecvConverter,TBool aHeaderOnly)

	{

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

	self->ConstructL(aPopSession,aRecvConverter,aHeaderOnly);

	return self;

	}

void CImPop3Top::ConstructL(CImPop3Session *aPopSession, CImRecvConvert* aRecvConverter,TBool aHeaderOnly)

	{

	CImPop3Operation::Construct(aPopSession,aRecvConverter);

	iHeadersOnly = aHeaderOnly;

	}

void CImPop3Top::StartL(TRequestStatus& aStatus)

	{

	CheckIfPending();

	iFirstLine=ETrue;

	iNoBytesRetrieved=0;

	iEmptyHeaders = EFalse;

	if (iHeadersOnly)

		{

		iRecvConverter->ResetForHeadersL();

		}

	else

		{

		iRecvConverter->ResetL();

		}

	if(iPipeliningSupported)

		{

		if(iMessageNo <= iLastMulti)