136 lines
4.1 KiB
C++
136 lines
4.1 KiB
C++
#ifndef CRYPTOPP_OBJFACT_H
|
|
#define CRYPTOPP_OBJFACT_H
|
|
|
|
#include "cryptlib.h"
|
|
#include <map>
|
|
#include <vector>
|
|
|
|
NAMESPACE_BEGIN(CryptoPP)
|
|
|
|
//! _
|
|
template <class AbstractClass>
|
|
class ObjectFactory
|
|
{
|
|
public:
|
|
virtual AbstractClass * CreateObject() const =0;
|
|
};
|
|
|
|
//! _
|
|
template <class AbstractClass, class ConcreteClass>
|
|
class DefaultObjectFactory : public ObjectFactory<AbstractClass>
|
|
{
|
|
public:
|
|
AbstractClass * CreateObject() const
|
|
{
|
|
return new ConcreteClass;
|
|
}
|
|
|
|
};
|
|
|
|
//! _
|
|
template <class AbstractClass, int instance=0>
|
|
class ObjectFactoryRegistry
|
|
{
|
|
public:
|
|
class FactoryNotFound : public Exception
|
|
{
|
|
public:
|
|
FactoryNotFound(const char *name) : Exception(OTHER_ERROR, std::string("ObjectFactoryRegistry: could not find factory for algorithm ") + name) {}
|
|
};
|
|
|
|
~ObjectFactoryRegistry()
|
|
{
|
|
for (CPP_TYPENAME Map::iterator i = m_map.begin(); i != m_map.end(); ++i)
|
|
{
|
|
delete (ObjectFactory<AbstractClass> *)i->second;
|
|
i->second = NULL;
|
|
}
|
|
}
|
|
|
|
void RegisterFactory(const std::string &name, ObjectFactory<AbstractClass> *factory)
|
|
{
|
|
m_map[name] = factory;
|
|
}
|
|
|
|
const ObjectFactory<AbstractClass> * GetFactory(const char *name) const
|
|
{
|
|
CPP_TYPENAME Map::const_iterator i = m_map.find(name);
|
|
return i == m_map.end() ? NULL : (ObjectFactory<AbstractClass> *)i->second;
|
|
}
|
|
|
|
AbstractClass *CreateObject(const char *name) const
|
|
{
|
|
const ObjectFactory<AbstractClass> *factory = GetFactory(name);
|
|
if (!factory)
|
|
throw FactoryNotFound(name);
|
|
return factory->CreateObject();
|
|
}
|
|
|
|
// Return a vector containing the factory names. This is easier than returning an iterator.
|
|
// from Andrew Pitonyak
|
|
std::vector<std::string> GetFactoryNames() const
|
|
{
|
|
std::vector<std::string> names;
|
|
CPP_TYPENAME Map::const_iterator iter;
|
|
for (iter = m_map.begin(); iter != m_map.end(); ++iter)
|
|
names.push_back(iter->first);
|
|
return names;
|
|
}
|
|
|
|
CRYPTOPP_NOINLINE static ObjectFactoryRegistry<AbstractClass, instance> & Registry(CRYPTOPP_NOINLINE_DOTDOTDOT);
|
|
|
|
private:
|
|
// use void * instead of ObjectFactory<AbstractClass> * to save code size
|
|
typedef std::map<std::string, void *> Map;
|
|
Map m_map;
|
|
};
|
|
|
|
template <class AbstractClass, int instance>
|
|
ObjectFactoryRegistry<AbstractClass, instance> & ObjectFactoryRegistry<AbstractClass, instance>::Registry(CRYPTOPP_NOINLINE_DOTDOTDOT)
|
|
{
|
|
static ObjectFactoryRegistry<AbstractClass, instance> s_registry;
|
|
return s_registry;
|
|
}
|
|
|
|
template <class AbstractClass, class ConcreteClass, int instance = 0>
|
|
struct RegisterDefaultFactoryFor {
|
|
RegisterDefaultFactoryFor(const char *name=NULL)
|
|
{
|
|
// BCB2006 workaround
|
|
std::string n = name ? std::string(name) : std::string(ConcreteClass::StaticAlgorithmName());
|
|
ObjectFactoryRegistry<AbstractClass, instance>::Registry().
|
|
RegisterFactory(n, new DefaultObjectFactory<AbstractClass, ConcreteClass>);
|
|
}};
|
|
|
|
template <class SchemeClass>
|
|
void RegisterAsymmetricCipherDefaultFactories(const char *name=NULL, SchemeClass *dummy=NULL)
|
|
{
|
|
RegisterDefaultFactoryFor<PK_Encryptor, CPP_TYPENAME SchemeClass::Encryptor>((const char *)name);
|
|
RegisterDefaultFactoryFor<PK_Decryptor, CPP_TYPENAME SchemeClass::Decryptor>((const char *)name);
|
|
}
|
|
|
|
template <class SchemeClass>
|
|
void RegisterSignatureSchemeDefaultFactories(const char *name=NULL, SchemeClass *dummy=NULL)
|
|
{
|
|
RegisterDefaultFactoryFor<PK_Signer, CPP_TYPENAME SchemeClass::Signer>((const char *)name);
|
|
RegisterDefaultFactoryFor<PK_Verifier, CPP_TYPENAME SchemeClass::Verifier>((const char *)name);
|
|
}
|
|
|
|
template <class SchemeClass>
|
|
void RegisterSymmetricCipherDefaultFactories(const char *name=NULL, SchemeClass *dummy=NULL)
|
|
{
|
|
RegisterDefaultFactoryFor<SymmetricCipher, CPP_TYPENAME SchemeClass::Encryption, ENCRYPTION>((const char *)name);
|
|
RegisterDefaultFactoryFor<SymmetricCipher, CPP_TYPENAME SchemeClass::Decryption, DECRYPTION>((const char *)name);
|
|
}
|
|
|
|
template <class SchemeClass>
|
|
void RegisterAuthenticatedSymmetricCipherDefaultFactories(const char *name=NULL, SchemeClass *dummy=NULL)
|
|
{
|
|
RegisterDefaultFactoryFor<AuthenticatedSymmetricCipher, CPP_TYPENAME SchemeClass::Encryption, ENCRYPTION>((const char *)name);
|
|
RegisterDefaultFactoryFor<AuthenticatedSymmetricCipher, CPP_TYPENAME SchemeClass::Decryption, DECRYPTION>((const char *)name);
|
|
}
|
|
|
|
NAMESPACE_END
|
|
|
|
#endif
|