/*

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

#include "appuidmap.h"

#include "logger.h"

#include "stringconv.h"

#include "utils.h"

#include <iomanip>

EncDecContainerItem *AppUidListEntry::Factory()

{

	return new AppUidListEntry(AppUidMap::EnumEntries());

}

AppUidListEntry::AppUidListEntry(const EnumEntry *aEnumEntries)

	: EncDecContainerItem(), iUid("Application", aEnumEntries)

{

}

AppUidListEntry::~AppUidListEntry()

{

}

const char *AppUidListEntry::ItemType() const

{

	return 0; // n/a

}

void AppUidListEntry::Encode(REncodeWriteStream &aWriteStream)

{

	aWriteStream << iUid;

}

void AppUidListEntry::Decode(RDecodeReadStream &aReadStream)

{

	aReadStream >> iUid;

}

EncDecContainerItem *CertStoreEntry::Factory()

{

	return new CertStoreEntry;

}

static const EnumEntry enumDetailsForTBool[] =

{

    { "false", 0x00},

    { "true", 0x01},

    { "EFalse", false},

    { "ETrue", true},

	{ 0,0 }

};

CertStoreEntry::CertStoreEntry()

	: EncDecContainerItem(), 

	  iCertInfo(false),

	  iCertApps("ApplicationList", AppUidListEntry::Factory),

	  iTrusted("Trusted", enumDetailsForTBool), 

	  iReadDataStreamId("DataStreamId(read)", true),

	  iWriteDataStreamId("DataStreamId(write)", false),

	  iDataFileName("DataFileName"),

	  iCertData(),

	  iSwiMode(false)

{

	// We only need to initialise EncDecObject members which wrap non-class types

	iReadDataStreamId.Value() = 0;

	iWriteDataStreamId.Value() = 0;

}

CertStoreEntry::CertStoreEntry(bool aSwiMode)

	: EncDecContainerItem(), 

	  iCertInfo(aSwiMode),

	  iCertApps("ApplicationList", AppUidListEntry::Factory),

	  iTrusted("Trusted", enumDetailsForTBool), 

	  iReadDataStreamId("DataStreamId(read)", true),

	  iWriteDataStreamId("DataStreamId(write)", false),

	  iDataFileName("DataFileName"),

	  iCertData(),

	  iSwiMode(aSwiMode)

{

	// We only need to initialise EncDecObject members which wrap non-class types

	iReadDataStreamId.Value() = 0;

	iWriteDataStreamId.Value() = 0;

}

CertStoreEntry::~CertStoreEntry()

{

}

const TCertLabel &CertStoreEntry::Label() const

{

	return iCertInfo.Label();

}

CertInfo &CertStoreEntry::Info()

{

	return iCertInfo;

}

const CertInfo &CertStoreEntry::Info() const

{

	return iCertInfo;

}

const char *CertStoreEntry::ItemType() const

{

	return "Entry";

}

std::string CertStoreEntry::ItemName() const

{

	return stringFromUtf16(Label());

}

void CertStoreEntry::SetItemName(const std::string &aName)

{

	TInt outputWords;

	TText *outputBuf = utf16FromUtf8((const TUint8 *)aName.data(), aName.size(), outputWords);

	iCertInfo.Label() = TPtrC16(outputBuf, outputWords);

	delete [] outputBuf;

}

void CertStoreEntry::Encode(REncodeWriteStream &aWriteStream)

{

	iCertInfo.Encode(aWriteStream);

	aWriteStream << iCertApps;

	aWriteStream << iTrusted;

	if(aWriteStream.HumanReadable())

		{

		// Write data to a file

		// Generate a file name

		std::string certFileName = aWriteStream.CertFileName(iCertInfo.CertificateFormat(), iCertInfo.OutputCertificateId());

		iDataFileName.Value().Copy(TPtrC8((const TUint8*)certFileName.data(), certFileName.size()));

		// Write file name

		aWriteStream << iDataFileName;

		std::fstream certDataFile;

		OpenUtf8FStreamForWrite(certDataFile, certFileName.c_str());

		if(certDataFile.fail())

			{

			dbg << Log::Indent() << "Failed to open '" << certDataFile << "' for output!" << Log::Endl();

			FatalError();

			}

		if((iCertInfo.CertificateFormat() == EX509Certificate) && aWriteStream.PemOut())

			{

			std::string pemCert;

			Der2Pem(iCertData, pemCert);

			certDataFile.write(pemCert.data(), pemCert.size());

			}

		else

			{

			certDataFile.write(iCertData.data(), iCertData.size());

			}

		certDataFile.close();

		if(certDataFile.fail())

			{

			dbg << Log::Indent() << "Failed to write cert data to '" << certDataFile <<  Log::Endl();

			FatalError();

			}

		aWriteStream << iReadDataStreamId;

		}

	else

		{

		// Write to the store

		if(iCertData.size() != iCertInfo.CertSize())

			{

			dbg << Log::Indent() << "Internal error - cert data size does not match meta data" << Log::Endl();

			FatalError();

			}

		RStoreWriteStream dataStream;

		TStreamId dataStreamId = dataStream.CreateLC(*aWriteStream.StoreObject());

		prog << Log::Indent() << "Created store stream " << dataStreamId << " for certificate data" << Log::Endl();

		iWriteDataStreamId.Value() = dataStreamId;

		prog << Log::Indent() << "Writing " << iCertData.size() << " bytes of binary data" << Log::Endl();

		dataStream.WriteL((const TUint8 *)iCertData.data(), iCertData.size());

		CleanupStack::PopAndDestroy(&dataStream);

		aWriteStream << iWriteDataStreamId;

		}

}

void CertStoreEntry::Decode(RDecodeReadStream &aReadStream)

{

	iCertInfo.Decode(aReadStream);

	aReadStream >> iCertApps;

	if((!aReadStream.HumanReadable()) ||

	   (aReadStream.PeakToken() == iTrusted.Name()))

		{

		aReadStream >> iTrusted;

		}

	else

		{

		iTrusted.SetValue(true);

		}

	aReadStream >> iReadDataStreamId;

	if(aReadStream.HumanReadable())

		{

		aReadStream >> iDataFileName;

		// Read data from the specified file

		std::string nFileName = stringFromUtf16(iDataFileName.Value());

		std::fstream certDataFile;

		OpenUtf8FStreamForRead(certDataFile, nFileName.c_str());

		if(certDataFile.fail())

			{

			dbg << Log::Indent() << "Failed to open '" << nFileName << "' for input!" << Log::Endl();

			FatalError();

			}

		certDataFile.seekg(0, std::ios_base::end);

		TUint32 certSize = certDataFile.tellg();

		char *rawCertData = new char[certSize];

		certDataFile.seekg(0, std::ios_base::beg);

		certDataFile.read(rawCertData, certSize);

		certDataFile.close();

		if(certDataFile.fail())

			{

			dbg << Log::Indent() << "Failed to read cert data from '" << certDataFile << Log::Endl();

			FatalError();

			}

		iCertData.assign(rawCertData, certSize);

		delete [] rawCertData;

		if(iCertInfo.CertificateFormat() == EX509Certificate)

			{

			// It might be a PEM cert

			std::string derFromPem;

			if(Pem2Der(iCertData, derFromPem))

				{

				prog << Log::Indent() << "Converted PEM cert to DER" << Log::Endl();

				iCertData = derFromPem;

				certSize = iCertData.size();

				}

			}

		iCertInfo.SetCertSize(certSize);

		}

	else 

		{

		// Read data from the store

		RStoreReadStream dataStream;

		dataStream.OpenLC(*aReadStream.iStore, iReadDataStreamId.Value());

		TUint32 certSize = iCertInfo.CertSize();

		TUint8 * certData = new TUint8[certSize];

		prog << Log::Indent() << "Reading " << certSize << " byte certificate from store stream " << iReadDataStreamId.Value() << Log::Endl();

		dataStream.ReadL(certData, certSize);

		iCertData.assign((const char *)certData, certSize);

		CleanupStack::PopAndDestroy(&dataStream);

		}

	if(iCertInfo.CertificateFormat() == EX509Certificate)

		{

		TKeyIdentifier subjectKeyId;

		// nb. If processing a swicertstore we ignore any SubjectKeyId in the extension.

		if(X509SubjectKeyId((iSwiMode)?(KIgnoreCertificateExtension) : (KUseCertificateExtension), 

							false,

							iCertData,

							iCertSubject, subjectKeyId))

			{

			prog << Log::Indent() << "Subject = '" << iCertSubject << "'" << Log::Endl();

			prog << Log::Indent() << "Calculated SubjectKeyId is ";

			const TUint8 *p = subjectKeyId.Ptr();

			for(int i=0; i<subjectKeyId.Length(); ++i)

				{

				if(i) prog << ":";

				prog.Stream() << std::setfill('0') << std::setw(2) << int(p[i]);

				}

			prog.Stream() << std::setw(0);

			prog << Log::Endl();

			if(aReadStream.HumanReadable() && iCertInfo.SubjectKeyId().iAutoKey)

				{

				// Reading config file and auto set so copy generated

				// SubjectKeyId to value.

				prog << Log::Indent() << "Field set to auto so using calculated SubjectKeyId" << Log::Endl();;

				iCertInfo.SubjectKeyId().iHash = subjectKeyId;

				}

			else

				{

				// If the read value matches the calculated value then

				// set iAutoKey so we dump it as auto (with the value

				// as a comment).

				if(iCertInfo.SubjectKeyId().iHash == subjectKeyId)

					{

					prog << Log::Indent() << "Calculated SubjectKeyId matches value read from input so setting to auto" << Log::Endl();;

					iCertInfo.SubjectKeyId().iAutoKey = true;

					}

				else

					{

					prog << Log::Indent() << "Calculated SubjectKeyId does NOT match value read from input so setting to value read" << Log::Endl();;

					}

				}

			}

		}

}

CertStoreEntry& CertStoreEntry::operator= (const CertStoreEntry& aRhs)

{

	if(this == &aRhs) return *this; // handle self assignment

	EncDecContainerItem::operator=(*static_cast<const EncDecContainerItem *>(&aRhs));

	iCertInfo = aRhs.iCertInfo;

	iCertApps.reset();

	for(TUint32 i=0; i<aRhs.iCertApps.size(); ++i)

		{

		AppUidListEntry *newApp = new AppUidListEntry(AppUidMap::EnumEntries());

		const AppUidListEntry *oldApp = static_cast<const AppUidListEntry *>(&aRhs.iCertApps[i]);

		*newApp = *oldApp;

		iCertApps.push_back(newApp);

		}

	iTrusted = aRhs.iTrusted;

	iReadDataStreamId = aRhs.iReadDataStreamId;

	iWriteDataStreamId = aRhs.iWriteDataStreamId;

	iDataFileName = aRhs.iDataFileName;

	iCertData = aRhs.iCertData;

	iCertSubject = aRhs.iCertSubject;

	iSwiMode = aRhs.iSwiMode;

	return *this;

}

const TUint8 * CertStoreEntry::CertData() const

{

	return (const TUint8 *)iCertData.data();

}

const std::string &CertStoreEntry::CertSubject() const

{

	return iCertSubject;

}

// End of file