PKIFBCryptGuard.cpp

Go to the documentation of this file.

#include "pkif.h" 
#include "PKIFBCryptGuard.h"

#if !defined(WIN32)
#include <ltdl.h>
#endif

#include <cstring>

using namespace std;
class CPKIFBCryptGuardImpl
{
    public:
        CPKIFBCryptGuardImpl()
            :m_hBcrypt(0),m_bInited(false),m_lastError(0)
        {
            m_hBcrypt = LoadLibrary("bcrypt");
            if(m_hBcrypt) m_bInited = true;
        }
        ~CPKIFBCryptGuardImpl()
        {
            if(m_lastError)
            {
                free(m_lastError);
                m_lastError = 0;
            }
            if(m_hBcrypt) {
                FreeLibrary(m_hBcrypt);
            }
        }

        const char * GetLastError()
        {
            return m_lastError;
        }

        bool IsCNGAvailable()
        {
            return(m_hBcrypt != 0);
        }

    private:
        void SetLastError(const char * err)
        {
            if(m_lastError) {
                free(m_lastError);
                m_lastError = 0;
            }
            if(err) m_lastError = strdup(err);
        }
        HMODULE m_hBcrypt;

        char * m_lastError;
        bool m_bInited;
};

class BCryptGuardCleaner
{
    public:
        ~BCryptGuardCleaner()
        {
            CPKIFBCryptGuard::Shutdown();
        }
} BCryptGuardCleanerInstance;

CPKIFBCryptGuardImpl * CPKIFBCryptGuard::m_impl = 0;


CPKIFBCryptGuard::CPKIFBCryptGuard()
{
    if(!m_impl) {
        m_impl = new CPKIFBCryptGuardImpl();
    }
}
CPKIFBCryptGuard::~CPKIFBCryptGuard()
{
}


void CPKIFBCryptGuard::Shutdown()
{
    if(m_impl) {
        delete m_impl;
        m_impl = 0;
    }
}

bool CPKIFBCryptGuard::IsCNGAvailable()
{
    if(m_impl) {
        return m_impl->IsCNGAvailable();
    }
    return false;
}

const char * CPKIFBCryptGuard::GetLoaderError()
{
    if(m_impl) {
        return m_impl->GetLastError();
    }
    return 0;
}