/*
* 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 <cmscertchoice.h>
#include "pkcs7asn1.h"
#include <asn1dec.h>
#include <x509cert.h>
#include <asn1enc.h>
#include "cmsutils.h"

//Implementation of CMS Certificate Choice

CCmsCertificateChoice* CCmsCertificateChoice::NewL(const TDesC8& aRawData)
	{
	CCmsCertificateChoice* self = new (ELeave) CCmsCertificateChoice();
	CleanupStack::PushL(self);
	self->ConstructL(aRawData);
	CleanupStack::Pop(self);
	return self;
	}

CCmsCertificateChoice::~CCmsCertificateChoice()
	{
	delete iCertificate;
	delete iEncodedAttributeCertificate;
	}

CCmsCertificateChoice::CCmsCertificateChoice()
	{
	}

void CCmsCertificateChoice::ConstructL(const TDesC8& aRawData)
	{
	TASN1DecGeneric decGen(aRawData);
	decGen.InitL();
		
	if (decGen.Tag()==EASN1Sequence && decGen.Class()==EUniversal)
		{
		// x509 certificates
		iCertificateType=ECertificateX509;
		iCertificate = CX509Certificate::NewL(aRawData);
		}
	else if (decGen.Tag()==1 && decGen.Class()==EContextSpecific)
			{
			// Attribute certificate
			iCertificateType=ECertificateAttribute;
			CASN1EncEncoding* enc=CASN1EncEncoding::NewLC(aRawData, EASN1Sequence, EUniversal);
			iEncodedAttributeCertificate=CmsUtils::CreateDEREncodingLC(*enc);
			CleanupStack::Pop(iEncodedAttributeCertificate);
			CleanupStack::PopAndDestroy(enc);
			}
		else if (decGen.Tag()==0 && decGen.Class()==EContextSpecific)
				{
				// extended certificates not supported
				iCertificateType=ECertificateExtendedCerificate;
				User::Leave(KErrNotSupported);
				}
			else
				{
				User::Leave(KErrArgument);	
				}		
	}

EXPORT_C const CX509Certificate& CCmsCertificateChoice::Certificate(void) const
	{
	return *iCertificate;
	}

CCmsCertificateChoice* CCmsCertificateChoice::NewLC(const CX509Certificate& aCertificate)
	{	
	CCmsCertificateChoice* self = new (ELeave) CCmsCertificateChoice();
	CleanupStack::PushL(self);
	self->ConstructL(aCertificate);
	return self;
	}

CCmsCertificateChoice* CCmsCertificateChoice::NewL(const CX509Certificate& aCertificate)
	{
	CCmsCertificateChoice* self = CCmsCertificateChoice::NewLC(aCertificate);
	CleanupStack::Pop();
	return self;
	}

	
CCmsCertificateChoice* CCmsCertificateChoice::NewLC(TCertificateType aCertType, const TDesC8& aEncodedCertificate)
	{		
	CCmsCertificateChoice* self = new (ELeave) CCmsCertificateChoice();
	CleanupStack::PushL(self);
	self->ConstructL(aCertType, aEncodedCertificate);
	return self;
	}

CCmsCertificateChoice* CCmsCertificateChoice::NewL(TCertificateType aCertType, const TDesC8& aEncodedCertificate)
	{
	CCmsCertificateChoice* self = CCmsCertificateChoice::NewLC(aCertType, aEncodedCertificate);
	CleanupStack::Pop();
	return self;
	}

void CCmsCertificateChoice::ConstructL(const CX509Certificate& aCertificate)
	{
	iCertificateType=ECertificateX509;
	iCertificate=CX509Certificate::NewL(aCertificate);
	}

void CCmsCertificateChoice::ConstructL(TCertificateType aCertType, const TDesC8& aEncodedCertificate)
	{
	iCertificateType=aCertType;
	if (aCertType==ECertificateAttribute)
		{
		iEncodedAttributeCertificate=aEncodedCertificate.AllocL();	
		}
	else if (aCertType==ECertificateX509)
			{
			iCertificate=CX509Certificate::NewL(aEncodedCertificate);	
			}
		 else
			 {
			 User::Leave(KErrNotSupported);	
			 }
	}
	

EXPORT_C const HBufC8* CCmsCertificateChoice::AttributeCertificate() const
	{
	return iEncodedAttributeCertificate;
	}

CASN1EncEncoding* CCmsCertificateChoice::EncodeASN1DERLC() const
	{
	CASN1EncEncoding* enc(NULL);
	if (iCertificateType==ECertificateX509)
		{
		enc=CASN1EncEncoding::NewLC(iCertificate->Encoding());
		}
	else if (iCertificateType==ECertificateAttribute)
			{
			TASN1DecGeneric decGen(*iEncodedAttributeCertificate);
			decGen.InitL();
			if (decGen.Tag()!=EASN1Sequence)
				{
				User::Leave(KErrArgument);	
				}
			enc=CASN1EncEncoding::NewLC(*iEncodedAttributeCertificate, 1, EContextSpecific);
			}
		else
			{
			//Extended Certificate
			User::Leave(KErrNotSupported);
			}
	return enc;
	}

EXPORT_C CCmsCertificateChoice::TCertificateType CCmsCertificateChoice::CertificateType()
	{
	return iCertificateType;
	}