2
1
mirror of https://github.com/qpdf/qpdf.git synced 2024-11-13 16:26:28 +00:00
qpdf/libqpdf/QPDFCryptoProvider.cc
2019-11-09 09:53:42 -05:00

120 lines
2.8 KiB
C++

#include <qpdf/QPDFCryptoProvider.hh>
#include <qpdf/qpdf-config.h>
#include <qpdf/QUtil.hh>
#include <stdexcept>
#ifdef USE_CRYPTO_NATIVE
# include <qpdf/QPDFCrypto_native.hh>
#endif
#ifdef USE_CRYPTO_GNUTLS
# include <qpdf/QPDFCrypto_gnutls.hh>
#endif
std::shared_ptr<QPDFCryptoImpl>
QPDFCryptoProvider::getImpl()
{
QPDFCryptoProvider& p = getInstance();
if (p.m->default_provider.empty())
{
throw std::logic_error(
"QPDFCryptoProvider::getImpl called with no default provider.");
}
return p.getImpl_internal(p.m->default_provider);
}
std::shared_ptr<QPDFCryptoImpl>
QPDFCryptoProvider::getImpl(std::string const& name)
{
return getInstance().getImpl_internal(name);
}
template<typename T>
void
QPDFCryptoProvider::registerImpl(std::string const& name)
{
getInstance().registerImpl_internal<T>(name);
}
void
QPDFCryptoProvider::setDefaultProvider(std::string const& name)
{
getInstance().setDefaultProvider_internal(name);
}
QPDFCryptoProvider::QPDFCryptoProvider() :
m(std::make_shared<Members>())
{
#ifdef USE_CRYPTO_NATIVE
registerImpl_internal<QPDFCrypto_native>("native");
#endif
#ifdef USE_CRYPTO_GNUTLS
registerImpl_internal<QPDFCrypto_gnutls>("gnutls");
#endif
std::string default_crypto;
if (! QUtil::get_env("QPDF_CRYPTO_PROVIDER", &default_crypto))
{
default_crypto = DEFAULT_CRYPTO;
}
setDefaultProvider_internal(default_crypto);
}
QPDFCryptoProvider&
QPDFCryptoProvider::getInstance()
{
static QPDFCryptoProvider instance;
return instance;
}
std::shared_ptr<QPDFCryptoImpl>
QPDFCryptoProvider::getImpl_internal(std::string const& name) const
{
auto iter = this->m->providers.find(name);
if (iter == this->m->providers.end())
{
throw std::logic_error(
"QPDFCryptoProvider requested unknown implementation \"" +
name + "\"");
}
return this->m->providers[name]();
}
template<typename T>
void
QPDFCryptoProvider::registerImpl_internal(std::string const& name)
{
this->m->providers[name] = std::make_shared<T>;
}
void
QPDFCryptoProvider::setDefaultProvider_internal(std::string const& name)
{
if (! this->m->providers.count(name))
{
throw std::logic_error(
"QPDFCryptoProvider: request to set default"
" provider to unknown implementation \"" +
name + "\"");
}
this->m->default_provider = name;
}
std::set<std::string>
QPDFCryptoProvider::getRegisteredImpls()
{
std::set<std::string> result;
QPDFCryptoProvider& p = getInstance();
for (auto iter = p.m->providers.begin(); iter != p.m->providers.end();
++iter)
{
result.insert((*iter).first);
}
return result;
}
std::string
QPDFCryptoProvider::getDefaultProvider()
{
return getInstance().m->default_provider;
}