/*

* Copyright (c) 2006-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: 

*

*/

#include <cmscontentinfo.h>

#include <asn1dec.h>

#include <asn1enc.h> 

#include "pkcs7asn1.h"

#include "cmsutils.h"

//

//Implementation of CMS ContentInfo Identity

//

EXPORT_C CCmsContentInfo* CCmsContentInfo::NewLC(const TDesC8& aRawData)

	{

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

	CleanupStack::PushL(self);

	self->ConstructL(aRawData);

	return self;			

	}

EXPORT_C CCmsContentInfo* CCmsContentInfo::NewL(const TDesC8& aRawData)

	{

	CCmsContentInfo* self = NewLC(aRawData);

	CleanupStack::Pop(self);

	return self;			

	}

EXPORT_C CCmsContentInfo* CCmsContentInfo::NewL(TCmsContentInfoType aContentInfoType, const TDesC8& aContentData)

	{

	CCmsContentInfo* self = NewLC(aContentInfoType, aContentData);

	CleanupStack::Pop(self);

	return self;					

	}

EXPORT_C CCmsContentInfo* CCmsContentInfo::NewLC(TCmsContentInfoType aContentInfoType, const TDesC8& aContentData)

	{

	CCmsContentInfo* self = new (ELeave) CCmsContentInfo(aContentInfoType, aContentData);

	CleanupStack::PushL(self);

	return self;					

	}

EXPORT_C CCmsContentInfo::~CCmsContentInfo()

	{

	}

CCmsContentInfo::CCmsContentInfo()

	{

	}

CCmsContentInfo::CCmsContentInfo(TCmsContentInfoType aContentInfoType, const TDesC8& aContentData)

:iContentType(aContentInfoType)

	{

	iContentData.Set(aContentData);

	}

void CCmsContentInfo::ConstructL(const TDesC8& aRawData)

	{

	const TInt minItems = 2;	// Must have OID

	const TInt maxItems = 2;	// Must have data

	CArrayPtr<TASN1DecGeneric>* contentInfo = PKCS7ASN1::DecodeSequenceLC(aRawData, minItems, maxItems);

	//Decode Content Type	

	iContentType=(TCmsContentInfoType)(CmsUtils::DecodeContentTypeL(contentInfo->At(0)));

	//Decode Content Data

	const TASN1DecGeneric* contentInfoAt1 = contentInfo->At(1);

	if ( contentInfoAt1->Tag() == 0 || contentInfoAt1->Class() == EContextSpecific )

		{

		TASN1DecGeneric decGen(contentInfoAt1->GetContentDER());

		decGen.InitL();

		if (iContentType==EContentTypeData)

			{

			if (decGen.Tag()!=EASN1OctetString || decGen.Class()!=EUniversal)

				{

				User::Leave(KErrArgument);	

				}

			else

				{

				iContentData.Set(decGen.GetContentDER());	

				}

			}

		else

			{

			iContentData.Set(decGen.Encoding());				

			}

		}

	else

		{

		User::Leave(KErrArgument);	

		}

	CleanupStack::PopAndDestroy(contentInfo);

	}

EXPORT_C CASN1EncSequence* CCmsContentInfo::EncodeASN1DERLC() const

	{

	// the root sequence contains the OID and the content data

	CASN1EncSequence* root = CASN1EncSequence::NewLC();

	//Encode the OID

	CASN1EncObjectIdentifier* oid = CmsUtils::EncodeContentTypeLC(iContentType);	 	

	root->AddAndPopChildL(oid);

	CASN1EncBase* enc(NULL);

	if (iContentType==EContentTypeData)

		{

		enc=CASN1EncOctetString::NewL(iContentData);			

		}

	else

		{

		//Encode the  Content

		//iContentData already encoded in sequence

		//so just rebuild the structure

		enc = CASN1EncEncoding::NewL(iContentData);

		}

	// Add [0] EXPLICT

	CASN1EncExplicitTag* Enc=CASN1EncExplicitTag::NewLC(enc, 0);

	root->AddAndPopChildL(Enc);					

	return root;

	}

EXPORT_C const TPtrC8 CCmsContentInfo::ContentData() const

	{

	return iContentData;	

	}

EXPORT_C TCmsContentInfoType CCmsContentInfo::ContentType() const

	{

	return iContentType;	

	}

//	

//Implementation of CMS EncapsulatedContentInfo Identity	

//

CEncapsulatedContentInfo* CEncapsulatedContentInfo::NewLC(const TDesC8& aRawData)

	{

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

	CleanupStack::PushL(self);

	self->ConstructL(aRawData);

	return self;			

	}

CEncapsulatedContentInfo* CEncapsulatedContentInfo::NewL(const TDesC8& aRawData)

	{

	CEncapsulatedContentInfo* self = NewLC(aRawData);

	CleanupStack::Pop(self);

	return self;			

	}

CEncapsulatedContentInfo* CEncapsulatedContentInfo::NewLC(TCmsContentInfoType aContentInfoType, TBool aIsEContentDataPresent, const TDesC8& aContentData)

	{

	CEncapsulatedContentInfo* self = new (ELeave) CEncapsulatedContentInfo(aContentInfoType, aIsEContentDataPresent, aContentData);

	CleanupStack::PushL(self);

	return self;					

	}

CEncapsulatedContentInfo* CEncapsulatedContentInfo::NewL(TCmsContentInfoType aContentInfoType, TBool aIsEContentDataPresent, const TDesC8& aContentData)

	{

	CEncapsulatedContentInfo* self = NewLC(aContentInfoType, aIsEContentDataPresent, aContentData);

	CleanupStack::Pop(self);

	return self;	

	}

CEncapsulatedContentInfo::CEncapsulatedContentInfo()

	{

	}

CEncapsulatedContentInfo::CEncapsulatedContentInfo(TCmsContentInfoType aContentInfoType, TBool aIsEContentDataPresent, const TDesC8& aContentData)

:iContentType(aContentInfoType),

 iIsContentDataPresent(aIsEContentDataPresent)

	{

	if (aIsEContentDataPresent)

		{

		iContentData.Set(aContentData);			

		}

	}

void CEncapsulatedContentInfo::ConstructL(const TDesC8& aRawData)

	{	

	const TInt minItems = 1;	// Must have OID

	const TInt maxItems = 2;	// Must have data

	CArrayPtr<TASN1DecGeneric>* contentInfo = PKCS7ASN1::DecodeSequenceLC(aRawData, minItems, maxItems);

	//Decode Content Type	

	iContentType=(TCmsContentInfoType)(CmsUtils::DecodeContentTypeL(contentInfo->At(0)));

	//Decode Content Data

	if(contentInfo->Count() == 2)

		{

		iIsContentDataPresent=ETrue;

		const TASN1DecGeneric* contentInfoAt1 = contentInfo->At(1);

		//Decode [0] Explicit

		if ( contentInfoAt1->Tag() == 0 || contentInfoAt1->Class() == EContextSpecific )

			{

			//Decode Wrapper Octet string	

			TASN1DecGeneric decGen(contentInfoAt1->GetContentDER());

			decGen.InitL();

			if(decGen.Tag() == EASN1OctetString && decGen.Class() == EUniversal)

				{

				iContentData.Set(decGen.GetContentDER());

				}

			else

				{

				//Wrapper is not an Octect String

				User::Leave(KErrArgument);

				}			

			}

		else

			{

			//Not [0] Explicit

			User::Leave(KErrArgument);

			}

		}

	else

		{

		//No optional data

		iContentData.Set(KNullDesC8());	

		}

	CleanupStack::PopAndDestroy(contentInfo);

	}	

CASN1EncSequence* CEncapsulatedContentInfo::EncodeASN1DERLC() const

	{

	// the root sequence contains the OID and the content data

	CASN1EncSequence* root = CASN1EncSequence::NewLC();

	//Encode the OID

	CASN1EncObjectIdentifier* oid = CmsUtils::EncodeContentTypeLC(iContentType);	 	

	root->AddAndPopChildL(oid);

	//Encode the Content

	if (iIsContentDataPresent)

		{

		//Wrapper Octect String

		CASN1EncOctetString* octetString=CASN1EncOctetString::NewL(iContentData);

		// Add [0] EXPLICT

		CASN1EncExplicitTag* Enc=CASN1EncExplicitTag::NewLC(octetString, 0);

		root->AddAndPopChildL(Enc);		

		}

	return root;

	}

EXPORT_C TBool CEncapsulatedContentInfo::IsContentDataPresent() const

	{

	return iIsContentDataPresent;	

	}

EXPORT_C const TPtrC8 CEncapsulatedContentInfo::ContentData() const

	{

	return iContentData;	

	}

EXPORT_C TCmsContentInfoType CEncapsulatedContentInfo::ContentType() const

	{

	return iContentType;	

	}