// Copyright (c) 2003-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 "cmultipartbiniterator.h"

EXPORT_C  CMultipartBinIterator* CMultipartBinIterator::NewL(CPushMessage& aPushMessage)

/** Allocates and creates a new binary multipart message iterator.

@param aPushMessage Binary multipart message to process

@return New binary multipart message iterator */

	{

	CMultipartBinIterator* self =  new (ELeave) CMultipartBinIterator(aPushMessage);

	CleanupStack::PushL(self);

	self->ConstructL(); 

	CleanupStack::Pop();

	return self;

	}

/**

 * CMultipartPushMessage is a base utility class to crack out parts of a 

 * multipart message.

 * @param aPushMessage: A message determined to be multipart. 

 */

CMultipartBinIterator::CMultipartBinIterator(CPushMessage& aPushMessage) : 

	CMultipartIteratorBase(aPushMessage)

	{

	}

/**

 * Validates & completes the construction of this class.

 */

void CMultipartBinIterator::ConstructL() 

	{

#ifdef _DEBUG

	_LIT(KMultipartBin, "application/vnd.wap.multipart.");

	TPtrC contentType;

	iPushMessage.GetContentType(contentType);

	__ASSERT_DEBUG( contentType.Find(KMultipartBin) == 0 , User::Panic(KPushInvalidContentType,0));

#endif

	TPtrC8 msgBody;

	if (!iPushMessage.GetMessageBody(msgBody))

		User::Leave(KErrNotFound);

	// get the nEntries field from the start of the binary multipart

	TWapBinCodex::ExtractUIntvarL(msgBody, 0, iMultiNumEntries);

	if (iMultiNumEntries.iValue == 0) // No multiparts

		User::Leave(KErrCorrupt);

	// move the iterator to start off after the multipart header

	iCurrentPartStart = iMultiNumEntries.iOctetSize;		

	}

EXPORT_C TBool CMultipartBinIterator::NextL()

/** Moves the iterator to the next message part.

@return ETrue if there is a next part, EFalse otherwise */

	{

	if (++iCurrentPart >= iMultiNumEntries.iValue)

		return EFalse;

	TWapBinCodex::TUintvar hdrSize, bodySize;

	TPtrC8 msgBody;

	if (!iPushMessage.GetMessageBody(msgBody))

		return EFalse;

	// Get the header size

	TWapBinCodex::ExtractUIntvarL(msgBody, iCurrentPartStart,  hdrSize);

	// move index over the header size data

	iCurrentPartStart += hdrSize.iOctetSize;

	// Get the body size

	TWapBinCodex::ExtractUIntvarL(msgBody, iCurrentPartStart, bodySize);

	// move index over the body size data

	iCurrentPartStart += bodySize.iOctetSize;

	// move index over the part data

	iCurrentPartStart += (hdrSize.iValue + bodySize.iValue);

	return ETrue;

	}

EXPORT_C void CMultipartBinIterator::FirstL()

/** Resets the iterator the beginning of the first part of the message. */

	{

	iCurrentPartStart = iMultiNumEntries.iOctetSize;	

	iCurrentPart = 0;

	}

EXPORT_C CPushMessage* CMultipartBinIterator::PartL()

/** Gets the message part currently referenced by the iterator.

@return Message part 

@leave KErrCorrupt The message is too short

@leave KErrNotFound Will leave if message body is empty

@leave TDesC8::AllocLC

@leave CPushMessage::NewL

*/

	{

	__ASSERT_DEBUG( iCurrentPart < iMultiNumEntries.iValue, User::Panic(KPushPartIndexOutOfRange,0));

	TPtrC8 msgBody;

	if (!iPushMessage.GetMessageBody(msgBody))

		User::Leave(KErrNotFound);

	// iCurrentPartStart is pointing to beginning of the message 

	// index will be used to get past the message headers

	TUint index = iCurrentPartStart;

	// Move over the headers

	TWapBinCodex::TUintvar hdrSize, bodySize;

	TWapBinCodex::ExtractUIntvarL(msgBody, index,  hdrSize);

	index += hdrSize.iOctetSize;

	TWapBinCodex::ExtractUIntvarL(msgBody, index, bodySize);

	index += bodySize.iOctetSize;

	// Now get the data

	TUint len = msgBody.Length();

	// validate that we have as much data as HEADER indicates

	if ( hdrSize.iValue < 0 || // No negative indices

		(len < (index + hdrSize.iValue)))	

		User::Leave(KErrCorrupt);

	// Get the Header

	HBufC8* header = msgBody.Mid(index, hdrSize.iValue).AllocLC();

	index += hdrSize.iValue;

	// validate that we have as much data as BODY indicates 

	if ( bodySize.iValue < 0 || // No negative indices

		(len < (index + bodySize.iValue)))	

		User::Leave(KErrCorrupt);

	// Get the Body

	HBufC8* body = msgBody.Mid(index, bodySize.iValue).AllocLC();

	CPushMessage* pm = CPushMessage::NewL( header,  body);

	CleanupStack::Pop(2, header);  // body, header

	return pm;

	}