/*

* Copyright (c) 2005-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 "pkcs12safebag.h"

#include <pkcs12bags.h>

using namespace PKCS12;

CDecPkcs12SafeBag::CDecPkcs12SafeBag()

	{

	}

EXPORT_C CDecPkcs12SafeBag* CDecPkcs12SafeBag::NewL(const TDesC8& aSafeBagData)

	{

	TASN1DecGeneric seqGen(aSafeBagData);

	seqGen.InitL();

	// Check if this is a Sequence

	if (seqGen.Tag() != EASN1Sequence || seqGen.Class() != EUniversal)

		{

		User::Leave(KErrArgument);

		}

	TASN1DecSequence seq;

	CArrayPtrFlat<TASN1DecGeneric>* safeBagSequence = seq.DecodeDERLC(seqGen);

	// Check for BagId, BagId is an ObjectIdentifier

	const TASN1DecGeneric* safeBagSequenceAt0 = safeBagSequence->At(0);

	if (safeBagSequenceAt0->Tag() != EASN1ObjectIdentifier || safeBagSequenceAt0->Class() != EUniversal)

		{

		User::Leave(KErrArgument);

		}

	TASN1DecObjectIdentifier oid;

  	HBufC* bagId = oid.DecodeDERL(*safeBagSequenceAt0);

	CleanupStack::PushL(bagId);

	CDecPkcs12SafeBag* safeBag = NULL;

	// If BagType is a KeyBag.	

	if( *bagId == KPkcs12KeyBagOID )

		{

		safeBag = CDecPkcs12KeyBag::NewL(aSafeBagData);

		}

	// If BagType is a Shrouded KeyBag.

	else if( *bagId == KPkcs12ShroudedKeyBagOID )

		{

		safeBag = CDecPkcs12ShroudedKeyBag::NewL(aSafeBagData);

		}

	// If bagType is a CertBag.

	else if( *bagId == KPkcs12CertBagOID )

		{

		safeBag = CDecPkcs12CertBag::NewL(aSafeBagData);

		}

	// If Bag Type is a SafeContentsBag.

	else if( *bagId == KPkcs12SafeContentsBagOID )

		{

		safeBag = CDecPkcs12SafeContentsBag::NewL(aSafeBagData);

		}

	// If Bag Type is a CrlBag.

	else if ( *bagId == KPkcs12CrlBagOID )

		{

		safeBag = new(ELeave) CDecPkcs12SafeBag();

		CleanupStack::PushL(safeBag);

		safeBag->ConstructL(aSafeBagData);

		CleanupStack::Pop(safeBag);

		}

	// If Bag Type is a Secret Bag.

	else if ( *bagId == KPkcs12SecretBagOID )

		{

		safeBag = new(ELeave) CDecPkcs12SafeBag();

		CleanupStack::PushL(safeBag);

		safeBag->ConstructL(aSafeBagData);

		CleanupStack::Pop(safeBag);

		}

	else

		{

		User::Leave(KErrNotSupported);	

		}	

	CleanupStack::PopAndDestroy(2,safeBagSequence); // safeBagSequence, bagId.

	return safeBag;

	}

EXPORT_C CDecPkcs12SafeBag::~CDecPkcs12SafeBag()

	{

	iBagAttributes.ResetAndDestroy();

	iBagAttributes.Close();

	}

void CDecPkcs12SafeBag::ConstructL(const TDesC8& aSafeBagData)

	{

	// This is SafeBag Sequence

	TASN1DecGeneric seqGen(aSafeBagData);

	seqGen.InitL();

	// Check if this is a Sequence

	if (seqGen.Tag() != EASN1Sequence || seqGen.Class() != EUniversal)

		{

		User::Leave(KErrArgument);

		}

	TASN1DecSequence seq;

	CArrayPtrFlat<TASN1DecGeneric>* safeBagSequence = seq.DecodeDERLC(seqGen);

	const TASN1DecGeneric* safeBagSequenceAt0 = safeBagSequence->At(0);

	// Obtain the BagId from the SafeBag Sequence

	 if (safeBagSequenceAt0->Tag() != EASN1ObjectIdentifier || safeBagSequenceAt0->Class() != EUniversal)

		{

		User::Leave(KErrArgument);

		}

	TASN1DecObjectIdentifier oid;

	HBufC* bagId = oid.DecodeDERL(*(safeBagSequence->At(0)));

	// If BagType is a KeyBag	

	if(*bagId == KPkcs12KeyBagOID)

		{

		iBagId = EKeyBag;

		}

	// If BagType is a Shrouded KeyBag	

	else if( *bagId == KPkcs12ShroudedKeyBagOID )

		{

		iBagId = EShroudedKeyBag;

		}

	// If bagType is a CertBag

	else if( *bagId == KPkcs12CertBagOID )

		{

		iBagId = ECertBag;

		}

	else if ( *bagId == KPkcs12CrlBagOID )

		{

		iBagId = ECrlBag;

		}

	else if ( *bagId == KPkcs12SecretBagOID )

		{

		iBagId = ESecretBag;

		}

	// If Bag Type is a SafeContentsBag

	else if( *bagId == KPkcs12SafeContentsBagOID )

		{

		iBagId = ESafeContentsBag;

		}

	else

		{

		User::Leave(KErrNotSupported);	

		}	

	delete bagId;	 

	// Obtain the BagValue from the SafeBag Sequence. The SafeBag contains a KeyBag 

	// or a ShroudedKeyBag.

	const TASN1DecGeneric* safeBagSequenceAt1 = safeBagSequence->At(1);

	if (safeBagSequenceAt1->Tag() == EASN1EOC || safeBagSequenceAt1->Class() == EContextSpecific)

		{

		TASN1DecGeneric seqGen(safeBagSequence->At(1)->GetContentDER());

		seqGen.InitL();

		iBagValue.Set(seqGen.Encoding());

		}

   	else

		{

		User::Leave(KErrArgument);

		}

	// Obtain the BagAttribute from the SafeBag Sequence. ATTRIBUTES ARE OPTIONAL

	if (safeBagSequence->Count() == 3)

		{

		TASN1DecGeneric seqGen(*(safeBagSequence->At(2)));

	    seqGen.InitL();

		// Check if this is a Set

		if (seqGen.Tag() != EASN1Set || seqGen.Class() != EUniversal)

			{

			User::Leave(KErrArgument);

			}   

		// Set

		TASN1DecSet decSet;

        CArrayPtrFlat<TASN1DecGeneric>* attributeSet = decSet.NewDERLC((safeBagSequence->At(2)->Encoding()));

        TInt attributeSetCount = attributeSet->Count();

        for ( TInt index = 0; index < attributeSetCount; index++ )

        	{

        	const TDesC8& attribute(attributeSet->At(index)->Encoding());

        	CDecPkcs12Attribute* bagAttribute = CDecPkcs12Attribute::NewL(attribute);

        	CleanupStack::PushL(bagAttribute);

         	iBagAttributes.AppendL(bagAttribute);

         	CleanupStack::Pop(bagAttribute);

         	}

        CleanupStack::PopAndDestroy(attributeSet); // attributeSet

        }

	CleanupStack::PopAndDestroy(safeBagSequence); // safeBagSequence

	}

EXPORT_C CDecPkcs12SafeBag::TBagId CDecPkcs12SafeBag::BagID() const

	{

	return iBagId;

	}

EXPORT_C const TDesC8& CDecPkcs12SafeBag::BagValue() const

	{

	return iBagValue;

	}

EXPORT_C const RPointerArray<CDecPkcs12Attribute>& CDecPkcs12SafeBag::BagAttributes() const

	{

	return iBagAttributes;

	}