/*
* 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;
	}