/*

* Copyright (c) 2008 Nokia Corporation and/or its subsidiary(-ies).

* All rights reserved.

* This component and the accompanying materials are made available

* under the terms of "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:  JavaCertStore

*

*/

#include "commsendpoint.h"

#include "comms.h"

#include "commsmessage.h"

#include "logger.h"

#include "javaoslayer.h"

#include "coreinterface.h"

#include "javacertstore.h"

#include "securitycommsmessagedefs.h"

#include "javacommonutils.h"

#include <sys/stat.h>

#include <sys/types.h>

#include <errno.h>

#include <dirent.h>

#include <stdlib.h>

#include <string.h>

#include <algorithm>

#include "trustedcertificate.h"

#include "javasmartcardcertificatesreader.h"

#include "metadatafilehandler.h"

#ifdef __SYMBIAN32__

java::captain::ExtensionPluginInterface* getExtensionPlugin()

{

#else

extern "C" java::captain::ExtensionPluginInterface* getExtensionPlugin()

{

#endif

    return new java::captain::JavaCertStore();

}

using namespace std;

using namespace java::security;

using namespace java::util;

using namespace java::comms;

namespace java

{

namespace captain

{

JavaCertStore::JavaCertStore() : iCore(0)

{

    JELOG2(EJavaCaptain);

}

JavaCertStore::~JavaCertStore()

{

    JELOG2(EJavaCaptain);

}

void JavaCertStore::startPlugin(CoreInterface* core)

{

    JELOG2(EJavaCaptain);

    iCore = core;

    iCore->getComms()->registerListener(java::comms::PLUGIN_ID_JAVA_CERT_STORE_EXTENSION_C, this);

    loadCertsMetadata();

}

void JavaCertStore::stopPlugin()

{

    JELOG2(EJavaCaptain);

    iCore->getComms()->unregisterListener(java::comms::PLUGIN_ID_JAVA_CERT_STORE_EXTENSION_C, this);

    iCore = 0;

    unloadCertsMetadata();

}

java::comms::CommsListener* JavaCertStore::getCommsListener()

{

    JELOG2(EJavaCaptain);

    return this;

}

void JavaCertStore::processMessage(java::comms::CommsMessage& aMessage)

{

    JELOG2(EJavaCaptain);

    switch (aMessage.getMessageId())

    {

    case JAVA_CERT_STORE_MSG_ID_REQUEST:

    {

        int operation = 0;

        aMessage >> operation;

        switch (operation)

        {

        case JAVA_CERT_STORE_OPERATION_QUERY_CERTS:

            queryCerts(aMessage);

            break;

        case JAVA_CERT_STORE_OPERATION_REFRESH_CERTS:

            /*if (!aMessage.hasPermission(MANAGE_CERTIFICATES)) {

              break;

            }*/

            unloadCertsMetadata();

            loadCertsMetadata();

            break;

        case JAVA_CERT_STORE_OPERATION_DELETE_CERT:

        case JAVA_CERT_STORE_OPERATION_DISABLE_CERT:

        case JAVA_CERT_STORE_OPERATION_ENABLE_CERT:

            // check that the client is trusted

            if (!aMessage.hasPermission(MANAGE_CERTIFICATES))

            {

                break;

            }

            std::string id;

            aMessage >> id;

            int state;

            switch (operation)

            {

            case JAVA_CERT_STORE_OPERATION_DELETE_CERT:

                state = STATE_DELETED;

                break;

            case JAVA_CERT_STORE_OPERATION_DISABLE_CERT:

                state = STATE_DISABLED;

                break;

            case JAVA_CERT_STORE_OPERATION_ENABLE_CERT:

                state = STATE_ENABLED;

                break;

            }

            manageCert(id, state);

            break;

        }

    }

    break;

    }

}

void JavaCertStore::queryCerts(java::comms::CommsMessage& aMessage)

{

    JELOG2(EJavaCaptain);

    int id = 0;

    std::string hash;

    int state = 0;

    aMessage >> id;

    int query = 0;

    // process the filter

    while (id != 0)

    {

        switch (id)

        {

        case JAVA_CERT_STORE_FILTER_ID_HASH:

            aMessage >> hash;

            break;

        case JAVA_CERT_STORE_FILTER_ID_STATE:

            int tmp;

            aMessage >> tmp;

            state = state | tmp;

            break;

        case JAVA_CERT_STORE_QUERY_ID_PROTECTION_DOMAIN:

            query = JAVA_CERT_STORE_QUERY_ID_PROTECTION_DOMAIN;

            break;

        case JAVA_CERT_STORE_QUERY_ID_FULL_DETAILS:

            query = JAVA_CERT_STORE_QUERY_ID_FULL_DETAILS;

            break;

        case JAVA_CERT_STORE_QUERY_ID_CERT_CONTENT:

            query = JAVA_CERT_STORE_QUERY_ID_CERT_CONTENT;

            break;

        case JAVA_CERT_STORE_QUERY_ID_CERT_CONTENT_PEM:

            query = JAVA_CERT_STORE_QUERY_ID_CERT_CONTENT_PEM;

            break;

        case JAVA_CERT_STORE_QUERY_ID_STATE:

            query = JAVA_CERT_STORE_QUERY_ID_STATE;

            break;

        }

        aMessage >> id;

    }

    java::comms::CommsMessage replyMsg;

    replyMsg.setReceiver(aMessage.getSender());

    replyMsg.setMessageRef(aMessage.getMessageRef());

    bool queryEnabled = (state & JAVA_CERT_STORE_STATE_ENABLED);

    bool queryDisabled = (state & JAVA_CERT_STORE_STATE_DISABLED);

    bool queryDeleted = (state & JAVA_CERT_STORE_STATE_DELETED);

    bool queryState = (query == JAVA_CERT_STORE_QUERY_ID_STATE);

    bool queryResponseSet = false;

    transform(hash.begin(), hash.end(), hash.begin(), (int(*)(int)) tolower);

    for (int i=0; i<no_certs; i++)

    {

        if ((state == 0 ||

                (queryEnabled && iCertsMetadata[i]->state == STATE_ENABLED)

                || (queryDisabled && iCertsMetadata[i]->state == STATE_DISABLED)

                || (queryDeleted && iCertsMetadata[i]->state == STATE_DELETED))

                && (hash.compare("") == 0

        {

            // what we return back depends on what was queried

            switch (query)

            {

            case JAVA_CERT_STORE_QUERY_ID_PROTECTION_DOMAIN:

                replyMsg << iCertsMetadata[i]->prot_domain_name;

                replyMsg << iCertsMetadata[i]->prot_domain_category;

                break;

            case JAVA_CERT_STORE_QUERY_ID_FULL_DETAILS:

                // return content, ID (which is the actual hash)

                // and the encoded info about the state

                if (replyWithContent(replyMsg, *iCertsMetadata[i]))

                {

                    // add also the ID and the state

                    replyMsg << encodeState(iCertsMetadata[i]->disposable,

                                            iCertsMetadata[i]->disablable,

                                            iCertsMetadata[i]->state);

                }

                break;

            case JAVA_CERT_STORE_QUERY_ID_CERT_CONTENT:

                // return the length and the content of the certificate

                replyWithContent(replyMsg, *iCertsMetadata[i]);

                break;

            case JAVA_CERT_STORE_QUERY_ID_CERT_CONTENT_PEM:

                // return the length and the content of the certificate in PEM format

                replyWithContent(replyMsg, *iCertsMetadata[i], PEM_FORMAT);

                break;

            case JAVA_CERT_STORE_QUERY_ID_STATE:

                if (iCertsMetadata[i]->state == STATE_ENABLED)

                {

                    replyMsg << JAVA_CERT_STORE_STATE_ENABLED;

                }

                else if (iCertsMetadata[i]->state == STATE_DISABLED)

                {

                    replyMsg << JAVA_CERT_STORE_STATE_DISABLED;

                }

                else if (iCertsMetadata[i]->state == STATE_DELETED)

                {

                    replyMsg << JAVA_CERT_STORE_STATE_DELETED;

                }

                else

                {

                    replyMsg << JAVA_CERT_STORE_STATE_UNKNOWN;

                }

                break;

            default:

                // if nothing specifically was queried,

                // return the full_path

                replyMsg << iCertsMetadata[i]->full_path;

            }

            queryResponseSet = true;

        }

    }

    if (queryState && !queryResponseSet)

    {

        replyMsg << JAVA_CERT_STORE_STATE_NOT_PRESENT;

    }

    int r = iCore->getComms()->send(replyMsg);

    if (r != 0)

    {

        ELOG1(EJavaCaptain,

              "[JavaCertStore] - error replying to JavaCaptain %s",

              java::util::JavaCommonUtils::intToString(r).c_str());

    }

}

void JavaCertStore::loadCertsMetadata()

{

    // loads the metadata of the certificates stored on device

    iPrimaryCertsPath = "Z";

    java::util::JavaOsLayer::getJavaCaptainRoot(iPrimaryCertsPath, true);

    iPrimaryCertsPath.append(KJavaCertsDir);

    iPrimaryCertsPath.append(1, KFileSeparator);

    iSecondaryCertsPath = "C";

    java::util::JavaOsLayer::getJavaCaptainRoot(iSecondaryCertsPath, true);

    iSecondaryCertsPath.append(KJavaCertsDir);

    iSecondaryCertsPath.append(1, KFileSeparator);

    iCertsMetadataPath = "C";

    java::util::JavaOsLayer::getJavaCaptainRoot(iCertsMetadataPath, true);

    iCertsMetadataPath.append(KJavaCertsStateDir);

    iCertsMetadataPath.append(1, KFileSeparator);

    no_certs = 0;

    loadCertsMetadata(iPrimaryCertsPath, true);

    loadCertsMetadata(iSecondaryCertsPath, false);

    // loads the metadata of the smart card certificates

    vector<TrustedCertificate> trustedCerts;

    JavaSmartCardCertificatesReader::retrieveTrustedCertificates(trustedCerts);

    for (int i=0; i<trustedCerts.size(); i++)

    {

        CERT_METADATA * metadata = new CERT_METADATA();

        metadata->disposable = false;

        metadata->disablable = false;

        metadata->state = STATE_ENABLED;

        metadata->changes = false;

        metadata->hash = trustedCerts[i].getId();

        transform(metadata->hash.begin(), metadata->hash.end(), metadata->hash.begin(), (int(*)(int)) tolower);

        metadata->full_path = "Smart Card";

        metadata->len = trustedCerts[i].getData().size();

        metadata->data = trustedCerts[i].getData();

        assignProtectionDomain(trustedCerts[i].getTrustedUsage(), metadata);

        addCertMetadataToCache(metadata, true /* overwrite */);

    }

    trustedCerts.clear();

}

void JavaCertStore::unloadCertsMetadata()

{

    for (int i=0; i<no_certs; i++)

    {

        // free the memory

        CERT_METADATA * metadata = iCertsMetadata[i];

        writeMetadataIntoFile(metadata);

        delete metadata;

    }

    // clear the cache

    iCertsMetadata.clear();

    no_certs = 0;

}

void JavaCertStore::writeMetadataIntoFile(CERT_METADATA * metadata)

{

    if (!metadata->changes)

    {

        // no changes

        return;

    }

    if (JavaCertStoreMetadataFileHandler::writeState(

                metadata->file_name, metadata->state))

    {

        // reset the changes flag

        metadata->changes = false;

    }

}

bool JavaCertStore::readMetadataFromFiles(const std::string& aCertsLocation, const std::string& cert_file_name, CERT_METADATA * metadata)

{

    // there are two types of metadata:

    // 1) read-only metadata (this metadata resides into file with the same name than

    //    certificate file, but with KMetadataSuffix as extension)

    // 2) read-write metadata (this metadata resides into file with the same name than

    //    certificate file, but with KStateSuffix as extension)

    bool readingSucceeded = false;

    size_t ext = cert_file_name.rfind('.');

    if (ext != string::npos)

    {

        std::string file_name_without_extension = string(cert_file_name, 0, ext);

        // read-only metadata

        std::string read_only_metadata_file_name = aCertsLocation

                + file_name_without_extension

                + KMetadataSuffix;

        std::string read_write_metadata_file_name = iCertsMetadataPath

                + file_name_without_extension

                + KStateSuffix;

        // read the read-only metadata file */

        FILE * read_only_metadata_file = fopen(read_only_metadata_file_name.c_str(),"r");

        if (read_only_metadata_file != NULL)

        {

            // save the name of the metadata_file for later use

            metadata->file_name = read_write_metadata_file_name;

            // identifier for the metadata value being read

            const int READ_DOMAIN_NAME = 2;

            const int READ_DOMAIN_CATEGORY = 3;

            const int READ_HASH = 4;

            const int READ_REMOVABLE = 5;

            const int READ_DISABLABLE = 6;

            // domain_name_info

            int domain_name_index = 0;

            char domain_name[50];

            // domain_category info

            int domain_category_index = 0;

            char domain_category[50];

            // hash info

            int hash_index = 0;

            char hash[50];

            // removable&disablable

            bool removable = false;

            bool disablable = false;

            // start&end separators for the metadata's keys

            bool key_ss = true;

            bool key_es = false;

            int retval;

            // identifier for the operation being performed

            int op = 0;

            while ((int)(retval = getc(read_only_metadata_file))!= EOF)

            {

                if (key_ss)

                {

                    // start of reading something new

                    switch ((char)retval)

                    {

                    case 'n':

                        op = READ_DOMAIN_NAME;

                        key_ss = false;

                        break;

                    case 'c':

                        op = READ_DOMAIN_CATEGORY;

                        key_ss = false;

                        break;

                    case 'h':

                        op = READ_HASH;

                        key_ss = false;

                        break;

                    case 'r':

                        op = READ_REMOVABLE;

                        key_ss = false;

                        break;

                    case 'd':

                        op = READ_DISABLABLE;

                        key_ss = false;

                        break;

                    }

                }

                else

                {

                    if ((char)retval == '=')

                    {

                        // end separator

                        key_es = true;

                    }

                    else if (retval == 10 || retval == 13 /* CR or LF */)

                    {

                        key_ss = true;

                        key_es = false;

                    }

                    else if (key_es)

                    {

                        switch (op)

                        {

                        case READ_DOMAIN_NAME:

                            domain_name[domain_name_index] = (char)retval;

                            domain_name_index++;

                            domain_name[domain_name_index] = '\0';

                            break;

                        case READ_DOMAIN_CATEGORY:

                            domain_category[domain_category_index] = (char)retval;

                            domain_category_index++;

                            domain_category[domain_category_index] = '\0';

                            break;

                        case READ_HASH:

                            hash[hash_index] = (char)retval;

                            hash_index++;

                            hash[hash_index] = '\0';

                            break;

                        case READ_REMOVABLE:

                            if ((char)retval == '1')

                            {

                                removable = true;

                            }

                            break;

                        case READ_DISABLABLE:

                            if ((char)retval == '1')

                            {

                                disablable = true;

                            }

                            break;

                        }

                    }

                }

            }

            if (domain_name_index > 0

                    && domain_category_index > 0

                    && hash_index > 0)

            {

                metadata->prot_domain_name = string(domain_name, domain_name_index);

                metadata->prot_domain_category = string(domain_category, domain_category_index);

                metadata->hash = string(hash, hash_index);

                transform(metadata->hash.begin(), metadata->hash.end(), metadata->hash.begin(), (int(*)(int)) tolower);

                metadata->disposable = removable;

                metadata->disablable = disablable;

                fclose(read_only_metadata_file);

                // read the read-write metadata file

                int state = JavaCertStoreMetadataFileHandler::readState(

                                read_write_metadata_file_name);

                if (state == STATE_UNDEFINED)

                {

                    // create the file and initialize it with

                    metadata->state = STATE_ENABLED;

                    // create the directory (if it doesn't exist)

                    std::string certsStateBaseDir = "C";

                    java::util::JavaOsLayer::getJavaCaptainRoot(certsStateBaseDir, true);

                    if (mkDirAll(KJavaCertsStateDir, certsStateBaseDir))

                    {

                        // force the writing

                        metadata->changes = true;

                        writeMetadataIntoFile(metadata);

                    }

                }

                else