/*

* 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 "pkcs12.h"

#include "pkcs7signedobject.h"

using namespace PKCS12;

CDecPkcs12::CDecPkcs12()

	{

	}

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

    {

    CDecPkcs12* self = NewLC(aRawData);

    CleanupStack::Pop(self);

    return self;

    }

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

    { 

    CDecPkcs12* self = new (ELeave) CDecPkcs12;

    CleanupStack::PushL(self);

    self->ConstructL(aRawData);

    return self;

    }

EXPORT_C CDecPkcs12* CDecPkcs12::NewL(RReadStream& aStream) 

    {

    CDecPkcs12* self = NewLC(aStream);

    CleanupStack::Pop(self);

    return self;

    }

EXPORT_C CDecPkcs12* CDecPkcs12::NewLC(RReadStream& aStream) 

    { 

    CDecPkcs12* self = new (ELeave) CDecPkcs12;

    CleanupStack::PushL(self);

    self->InternalizeL(aStream);

    return self;

    }

CDecPkcs12::~CDecPkcs12()

	{

	iContentInfos.ResetAndDestroy();

	iContentInfos.Close();

	delete iAuthenticatedSafeData;

	delete iMacData;	

	}

void CDecPkcs12::ConstructL(const TDesC8& aRawData)

	{	

	TASN1DecGeneric decGen(aRawData);

	decGen.InitL();

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

		{

		User::Leave(KErrArgument);

		}

	// aRawData conatins PFX Sequence

	TASN1DecSequence decSeq;

	CArrayPtr<TASN1DecGeneric>* seqContents = decSeq.DecodeDERLC(decGen);

    // Check if both the version and the authSafe are present, since macData is optional, 

	TInt pfxCount = seqContents->Count();

	if (pfxCount > 3 || pfxCount < 2)

		{

		User::Leave(KErrArgument);

		}

	// Decodes Version, Version is an Integer

	const TASN1DecGeneric* seqContentsAt0 = seqContents->At(0);

	if (seqContentsAt0->Tag() != EASN1Integer || seqContentsAt0->Class() != EUniversal)

		{

		User::Leave(KErrArgument);

		}

	TASN1DecInteger intDecoder;

	iVersion = intDecoder.DecodeDERShortL(*seqContentsAt0);

	if (iVersion != KPkcs12Version)

		{

		User::Leave(KErrArgument);

		}

	// Decodes the AuthSafe, AuthSafe is a Sequence

	// If AuthSafe is not present then leave

	const TASN1DecGeneric* seqContentsAt1 = seqContents->At(1);

	if(seqContentsAt1->Tag() != EASN1Sequence || seqContentsAt1->Class() != EUniversal)

		{

		User::Leave(KErrArgument);

		}

	TASN1DecSequence authSafeSeq;

	CArrayPtr<TASN1DecGeneric>* authSafeContents = authSafeSeq.DecodeDERLC(*seqContentsAt1);

	const TASN1DecGeneric* authSafeContentsAt0 = authSafeContents->At(0);

	if(authSafeContentsAt0->Tag() == EASN1ObjectIdentifier || authSafeContentsAt0->Class() == EUniversal)

		{

		TASN1DecObjectIdentifier oid;

		HBufC* objectId = oid.DecodeDERL(*authSafeContentsAt0);

		CleanupStack::PushL(objectId);

		if(*objectId == KPkcs7DataOID || *objectId == KPkcs7SignedDataOID)

			{

			// Decode this Sequence, If there is an OID for Data/Signed Data then use PKCS7 library

			iAuthenticatedSafeData = CPKCS7ContentInfo::NewL(seqContentsAt1->Encoding());

			}

		else

			{

			User::Leave(KErrArgument);

			}

		CleanupStack::PopAndDestroy(objectId);

		}

	// AuthSafe is absent

	else 

		{

		User::Leave(KErrArgument);

		}

	// Integrity Mode	

	if(iAuthenticatedSafeData->ContentType() == KPkcs7Data)

		{

		iMode = EPasswordIntegrityMode;	

		}

	else if(iAuthenticatedSafeData->ContentType() == KPkcs7SignedData)

		{

		iMode = EPublicKeyIntegrityMode;

		}

	TPtrC8 contentData(KNullDesC8);

	// In case of Password Integrity Mode, ContentType is Data

	if(iMode == EPasswordIntegrityMode)

		{

		// Check if MacData is present, Decodes MacData

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

  			{

  			const TASN1DecGeneric* seqContentsAt2 = seqContents->At(2);

    		if (seqContentsAt2->Tag() != EASN1Null || seqContentsAt2->Class() == EUniversal)

				{

				iMacData = CDecPkcs12MacData::NewL(seqContentsAt2->Encoding(),iAuthenticatedSafeData->ContentData());

			    } 

			}

		contentData.Set(iAuthenticatedSafeData->ContentData());

		}

    // Public-Key Integrity Mode, ContentType is SignedData

	else if(iMode == EPublicKeyIntegrityMode)

		{

		// Create a pkcs7signed data object

		CPKCS7SignedObject* signedData = CPKCS7SignedObject::NewL(*iAuthenticatedSafeData);

		CleanupStack::PushL(signedData);

		// Obtain the ContentInfo present in Signed Data

		const CPKCS7ContentInfo* contentInfo = &(signedData->ContentInfo());

		// Get the type of ContentInfo

		if (contentInfo->ContentType() != KPkcs7Data)

			{

			User::Leave(KErrNotSupported);

			}

		contentData.Set(contentInfo->ContentData());

		CleanupStack::PopAndDestroy(signedData);

		}

	else

		{

		User::Leave(KErrArgument);

		}

	TASN1DecGeneric decGen1(contentData);

	decGen1.InitL(); 					

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

		{

		User::Leave(KErrArgument);

		}

	// Desequence the ContentData present in ContentInfo which is in AuthSafe

    TASN1DecSequence seq;

    CArrayPtr<TASN1DecGeneric>* contentInfoSequences = seq.DecodeDERLC(decGen1);

    // The number of ContentInfos present         

    TInt contentInfoCount = contentInfoSequences->Count();

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

    	{

        CPKCS7ContentInfo* contentInfo = CPKCS7ContentInfo::NewL(contentInfoSequences->At(index)->Encoding());

        CleanupStack::PushL(contentInfo);

        iContentInfos.AppendL(contentInfo);

        CleanupStack::Pop(contentInfo);

        }

	CleanupStack::PopAndDestroy(3,seqContents); // seqContents, authSafeContents, contentInfoSequences

  	} 

void CDecPkcs12::InternalizeL(RReadStream& aStream)

	{

	MStreamBuf* streamBuf = aStream.Source();

	TInt pos = streamBuf->SizeL();

	streamBuf->SeekL(MStreamBuf::ERead, EStreamBeginning, 0);

	HBufC8* temp= HBufC8::NewLC(pos);	

	TPtr8 ptr = temp->Des();	

	aStream.ReadL(ptr,pos);	

    ConstructL(*temp);    

	CleanupStack::PopAndDestroy(temp); 

	}	

EXPORT_C TInt CDecPkcs12::Version() const

	{

	return iVersion;

   	}

EXPORT_C CDecPkcs12::TIntegrityMode CDecPkcs12::IntegrityMode() const

	{

	return iMode;

   	}

EXPORT_C const RPointerArray<CPKCS7ContentInfo>& CDecPkcs12::AuthenticatedSafeContents() const

	{

	return iContentInfos;

	}

EXPORT_C const CPKCS7ContentInfo& CDecPkcs12::AuthenticatedSafe() const

	{

	return *iAuthenticatedSafeData;

	}

EXPORT_C const CDecPkcs12MacData* CDecPkcs12::MacData() const

	{

	return iMacData;

	}