CAPICertUpdate2.cpp

Go to the documentation of this file.

#include "CAPICertUpdate2.h"
#include "CAPIUtils.h" 
#include "ToolkitUtils.h"
#include "components.h"
#include "PKIFCacheErrors.h"
#include "Buffer.h"
#include "Certificate.h"
#include "PKIFCacheException.h"

#include <atlbase.h>
#include <sstream>


struct CPKIFCAPICertUpdate2Impl
{
    HCERTSTORE m_hSto;
    int m_nSysStoRegLoc;
    char* m_szStore;
};

CPKIFCAPICertUpdate2::CPKIFCAPICertUpdate2(
    int sysStoRegLoc,
    const char* store)
    :m_impl (new CPKIFCAPICertUpdate2Impl), IPKIFCAPISource(sysStoRegLoc, store)
{
    LOG_STRING_DEBUG("CPKIFCAPICertUpdate2::CPKIFCAPICertUpdate2(void)", TOOLKIT_SR_CAPICERTUPDATE, 0, this);

    m_impl->m_hSto = NULL;

    m_impl->m_nSysStoRegLoc = sysStoRegLoc;

    m_impl->m_szStore = NULL;
    size_t len = 0;
    if(store)
    {
        len = strlen(store);
        m_impl->m_szStore = new char[len + 1];

        //reviewed 4/24
        strcpy(m_impl->m_szStore, store);
    }
}
CPKIFCAPICertUpdate2::~CPKIFCAPICertUpdate2(void)
{
    LOG_STRING_DEBUG("CPKIFCAPICertUpdate2::~CPKIFCAPICertUpdate2(void)", TOOLKIT_SR_CAPICERTUPDATE, 0, this);

    if(m_impl->m_szStore)
        delete[] m_impl->m_szStore;

    if(NULL != m_impl->m_hSto)
    {
        CertCloseStore(m_impl->m_hSto, 0); m_impl->m_hSto = NULL;
    }

    delete m_impl;
    m_impl = NULL;
}
void CPKIFCAPICertUpdate2::Initialize(void)
{
    LOG_STRING_DEBUG("CPKIFCAPICertUpdate2::Initialize(void)", TOOLKIT_SR_CAPICERTUPDATE, 0, this);

    if(NULL != m_impl->m_hSto)
    {
        LOG_STRING_WARN("Skipping initialization - CPKIFCAPICertUpdate2 instance already initialized", TOOLKIT_SR_CAPIREPOSITORY, COMMON_ALREADY_INITIALIZED, this);
        return; //already initialized - just log it, return and don't bother with exception
    }

    USES_CONVERSION;
    m_impl->m_hSto =  CertOpenStore(CERT_STORE_PROV_SYSTEM, X509_ASN_ENCODING, NULL, 
                                CERT_STORE_OPEN_EXISTING_FLAG | m_impl->m_nSysStoRegLoc , T2OLE(m_impl->m_szStore));
    if(NULL == m_impl->m_hSto)
    {
        std::ostringstream os;
        os << "CertOpenStore failed: " << GetLastError();
        RAISE_CACHE_EXCEPTION(os.str().c_str(), thisComponent, CACHE_CERT_STORE_OPEN_FAILED, this)
    }
}

void CPKIFCAPICertUpdate2::AddCertificate(CertType certType, const CPKIFCertificatePtr& cert)
{
    LOG_STRING_DEBUG("CPKIFCAPICertUpdate2::AddCertificate(CertType certType, const CPKIFCertificatePtr& cert)", TOOLKIT_SR_CAPICERTUPDATE, 0, this);

    if(NULL == m_impl->m_hSto)
    {
        RAISE_CACHE_EXCEPTION("CPKIFCAPICertUpdate2 instance not initialized.", thisComponent, COMMON_NOT_INITIALIZED, this)
    }

    CPKIFBufferPtr certBuf = cert->Encoded();
    if(!CertAddEncodedCertificateToStore(m_impl->m_hSto, X509_ASN_ENCODING, certBuf->GetBuffer(), certBuf->GetLength(),CERT_STORE_ADD_NEWER_INHERIT_PROPERTIES, NULL))
    {
        //this is for debugging only -  we don't really care if we failed to add it to the cache
        std::ostringstream os;
        os << "CertAddEncodedCertificateToStore failed: " << GetLastError();
        LOG_STRING_ERROR(os.str().c_str(), thisComponent, CACHE_UPDATE_FAILED, this);
    }
    else
    {
        LOG_STRING_DEBUG("Successfully added certificate to CAPI store", thisComponent, 0, this);
    }
}