]> git.netwichtig.de Git - user/henk/code/inspircd.git/blobdiff - src/modules/extra/m_ssl_gnutls.cpp
Merge insp20
[user/henk/code/inspircd.git] / src / modules / extra / m_ssl_gnutls.cpp
index 3b67a61802b51106ede6db1e20421340381bfa72..74b66d2de6bac523a23b9944b40a026d05a4b9d4 100644 (file)
 
 
 #include "inspircd.h"
+#include "modules/ssl.h"
+#include <memory>
+
+// Fix warnings about the use of commas at end of enumerator lists on C++03.
+#if defined __clang__
+# pragma clang diagnostic ignored "-Wc++11-extensions"
+#elif defined __GNUC__
+# pragma GCC diagnostic ignored "-pedantic"
+#endif
+
 #include <gnutls/gnutls.h>
 #include <gnutls/x509.h>
-#include "ssl.h"
-#include "m_cap.h"
+
+#if ((GNUTLS_VERSION_MAJOR > 2) || (GNUTLS_VERSION_MAJOR == 2 && GNUTLS_VERSION_MINOR > 9) || (GNUTLS_VERSION_MAJOR == 2 && GNUTLS_VERSION_MINOR == 9 && GNUTLS_VERSION_PATCH >= 8))
+#define GNUTLS_HAS_MAC_GET_ID
+#include <gnutls/crypto.h>
+#endif
+
+#if (GNUTLS_VERSION_MAJOR > 2 || GNUTLS_VERSION_MAJOR == 2 && GNUTLS_VERSION_MINOR > 12)
+# define GNUTLS_HAS_RND
+#else
+# include <gcrypt.h>
+#endif
 
 #ifdef _WIN32
 # pragma comment(lib, "libgnutls-28.lib")
 #endif
 
-/* $ModDesc: Provides SSL support for clients */
-/* $CompileFlags: pkgconfincludes("gnutls","/gnutls/gnutls.h","") iflt("pkg-config --modversion gnutls","2.12") exec("libgcrypt-config --cflags") */
-/* $LinkerFlags: rpath("pkg-config --libs gnutls") pkgconflibs("gnutls","/libgnutls.so","-lgnutls") iflt("pkg-config --modversion gnutls","2.12") exec("libgcrypt-config --libs") */
-/* $NoPedantic */
+/* $CompileFlags: pkgconfincludes("gnutls","/gnutls/gnutls.h","") eval("print `libgcrypt-config --cflags | tr -d \r` if `pkg-config --modversion gnutls 2>/dev/null | tr -d \r` lt '2.12'") */
+/* $LinkerFlags: rpath("pkg-config --libs gnutls") pkgconflibs("gnutls","/libgnutls.so","-lgnutls") eval("print `libgcrypt-config --libs | tr -d \r` if `pkg-config --modversion gnutls 2>/dev/null | tr -d \r` lt '2.12'") */
 
 #ifndef GNUTLS_VERSION_MAJOR
 #define GNUTLS_VERSION_MAJOR LIBGNUTLS_VERSION_MAJOR
@@ -52,87 +69,18 @@ typedef gnutls_certificate_credentials_t gnutls_certificate_credentials;
 typedef gnutls_dh_params_t gnutls_dh_params;
 #endif
 
-#if (GNUTLS_VERSION_MAJOR > 2 || (GNUTLS_VERSION_MAJOR == 2 && GNUTLS_VERSION_MINOR >= 12))
-# define GNUTLS_HAS_RND
-# include <gnutls/crypto.h>
-#else
-# include <gcrypt.h>
-#endif
-
 enum issl_status { ISSL_NONE, ISSL_HANDSHAKING_READ, ISSL_HANDSHAKING_WRITE, ISSL_HANDSHAKEN, ISSL_CLOSING, ISSL_CLOSED };
 
-struct SSLConfig : public refcountbase
-{
-       gnutls_certificate_credentials_t x509_cred;
-       std::vector<gnutls_x509_crt_t> x509_certs;
-       gnutls_x509_privkey_t x509_key;
-       gnutls_dh_params_t dh_params;
-#ifdef GNUTLS_NEW_PRIO_API
-       gnutls_priority_t priority;
-#endif
-
-       SSLConfig()
-               : x509_cred(NULL)
-               , x509_key(NULL)
-               , dh_params(NULL)
-#ifdef GNUTLS_NEW_PRIO_API
-               , priority(NULL)
-#endif
-       {
-       }
-
-       ~SSLConfig()
-       {
-               ServerInstance->Logs->Log("m_ssl_gnutls", DEBUG, "Destroying SSLConfig %p", (void*)this);
-
-               if (x509_cred)
-                       gnutls_certificate_free_credentials(x509_cred);
-
-               for (unsigned int i = 0; i < x509_certs.size(); i++)
-                       gnutls_x509_crt_deinit(x509_certs[i]);
-
-               if (x509_key)
-                       gnutls_x509_privkey_deinit(x509_key);
-
-               if (dh_params)
-                       gnutls_dh_params_deinit(dh_params);
-
-#ifdef GNUTLS_NEW_PRIO_API
-               if (priority)
-                       gnutls_priority_deinit(priority);
-#endif
-       }
-};
-
-static reference<SSLConfig> currconf;
-
-static SSLConfig* GetSessionConfig(gnutls_session_t session);
-
-#if(GNUTLS_VERSION_MAJOR < 2 || ( GNUTLS_VERSION_MAJOR == 2 && GNUTLS_VERSION_MINOR < 12 ) )
-static int cert_callback (gnutls_session_t session, const gnutls_datum_t * req_ca_rdn, int nreqs,
-       const gnutls_pk_algorithm_t * sign_algos, int sign_algos_length, gnutls_retr_st * st) {
-
-       st->type = GNUTLS_CRT_X509;
+#if (GNUTLS_VERSION_MAJOR > 2 || (GNUTLS_VERSION_MAJOR == 2 && GNUTLS_VERSION_MINOR >= 12))
+#define GNUTLS_NEW_CERT_CALLBACK_API
+typedef gnutls_retr2_st cert_cb_last_param_type;
 #else
-static int cert_callback (gnutls_session_t session, const gnutls_datum_t * req_ca_rdn, int nreqs,
-       const gnutls_pk_algorithm_t * sign_algos, int sign_algos_length, gnutls_retr2_st * st) {
-       st->cert_type = GNUTLS_CRT_X509;
-       st->key_type = GNUTLS_PRIVKEY_X509;
+typedef gnutls_retr_st cert_cb_last_param_type;
 #endif
-       SSLConfig* conf = GetSessionConfig(session);
-       std::vector<gnutls_x509_crt_t>& x509_certs = conf->x509_certs;
-       st->ncerts = x509_certs.size();
-       st->cert.x509 = &x509_certs[0];
-       st->key.x509 = conf->x509_key;
-       st->deinit_all = 0;
-
-       return 0;
-}
 
 class RandGen : public HandlerBase2<void, char*, size_t>
 {
  public:
-       RandGen() {}
        void Call(char* buffer, size_t len)
        {
 #ifdef GNUTLS_HAS_RND
@@ -143,563 +91,765 @@ class RandGen : public HandlerBase2<void, char*, size_t>
        }
 };
 
-/** Represents an SSL user's extra data
- */
-class issl_session
+namespace GnuTLS
 {
-public:
-       StreamSocket* socket;
-       gnutls_session_t sess;
-       issl_status status;
-       reference<ssl_cert> cert;
-       reference<SSLConfig> config;
-
-       issl_session() : socket(NULL), sess(NULL), status(ISSL_NONE) {}
-};
+       class Init
+       {
+        public:
+               Init() { gnutls_global_init(); }
+               ~Init() { gnutls_global_deinit(); }
+       };
 
-static SSLConfig* GetSessionConfig(gnutls_session_t sess)
-{
-       issl_session* session = reinterpret_cast<issl_session*>(gnutls_transport_get_ptr(sess));
-       return session->config;
-}
+       class Exception : public ModuleException
+       {
+        public:
+               Exception(const std::string& reason)
+                       : ModuleException(reason) { }
+       };
 
-class CommandStartTLS : public SplitCommand
-{
- public:
-       bool enabled;
-       CommandStartTLS (Module* mod) : SplitCommand(mod, "STARTTLS")
+       void ThrowOnError(int errcode, const char* msg)
        {
-               enabled = true;
-               works_before_reg = true;
+               if (errcode < 0)
+               {
+                       std::string reason = msg;
+                       reason.append(" :").append(gnutls_strerror(errcode));
+                       throw Exception(reason);
+               }
        }
 
-       CmdResult HandleLocal(const std::vector<std::string> &parameters, LocalUser *user)
+       /** Used to create a gnutls_datum_t* from a std::string
+        */
+       class Datum
+       {
+               gnutls_datum_t datum;
+
+        public:
+               Datum(const std::string& dat)
+               {
+                       datum.data = (unsigned char*)dat.data();
+                       datum.size = static_cast<unsigned int>(dat.length());
+               }
+
+               const gnutls_datum_t* get() const { return &datum; }
+       };
+
+       class Hash
        {
-               if (!enabled)
+               gnutls_digest_algorithm_t hash;
+
+        public:
+               // Nothing to deallocate, constructor may throw freely
+               Hash(const std::string& hashname)
                {
-                       user->WriteNumeric(691, "%s :STARTTLS is not enabled", user->nick.c_str());
-                       return CMD_FAILURE;
+                       // As older versions of gnutls can't do this, let's disable it where needed.
+#ifdef GNUTLS_HAS_MAC_GET_ID
+                       // As gnutls_digest_algorithm_t and gnutls_mac_algorithm_t are mapped 1:1, we can do this
+                       // There is no gnutls_dig_get_id() at the moment, but it may come later
+                       hash = (gnutls_digest_algorithm_t)gnutls_mac_get_id(hashname.c_str());
+                       if (hash == GNUTLS_DIG_UNKNOWN)
+                               throw Exception("Unknown hash type " + hashname);
+
+                       // Check if the user is giving us something that is a valid MAC but not digest
+                       gnutls_hash_hd_t is_digest;
+                       if (gnutls_hash_init(&is_digest, hash) < 0)
+                               throw Exception("Unknown hash type " + hashname);
+                       gnutls_hash_deinit(is_digest, NULL);
+#else
+                       if (hashname == "md5")
+                               hash = GNUTLS_DIG_MD5;
+                       else if (hashname == "sha1")
+                               hash = GNUTLS_DIG_SHA1;
+#ifdef INSPIRCD_GNUTLS_ENABLE_SHA256_FINGERPRINT
+                       else if (hashname == "sha256")
+                               hash = GNUTLS_DIG_SHA256;
+#endif
+                       else
+                               throw Exception("Unknown hash type " + hashname);
+#endif
                }
 
-               if (user->registered == REG_ALL)
+               gnutls_digest_algorithm_t get() const { return hash; }
+       };
+
+       class DHParams
+       {
+               gnutls_dh_params_t dh_params;
+
+               DHParams()
                {
-                       user->WriteNumeric(691, "%s :STARTTLS is not permitted after client registration is complete", user->nick.c_str());
+                       ThrowOnError(gnutls_dh_params_init(&dh_params), "gnutls_dh_params_init() failed");
                }
-               else
+
+        public:
+               /** Import */
+               static std::auto_ptr<DHParams> Import(const std::string& dhstr)
+               {
+                       std::auto_ptr<DHParams> dh(new DHParams);
+                       int ret = gnutls_dh_params_import_pkcs3(dh->dh_params, Datum(dhstr).get(), GNUTLS_X509_FMT_PEM);
+                       ThrowOnError(ret, "Unable to import DH params");
+                       return dh;
+               }
+
+               /** Generate */
+               static std::auto_ptr<DHParams> Generate(unsigned int bits)
+               {
+                       std::auto_ptr<DHParams> dh(new DHParams);
+                       ThrowOnError(gnutls_dh_params_generate2(dh->dh_params, bits), "Unable to generate DH params");
+                       return dh;
+               }
+
+               ~DHParams()
                {
-                       if (!user->eh.GetIOHook())
+                       gnutls_dh_params_deinit(dh_params);
+               }
+
+               const gnutls_dh_params_t& get() const { return dh_params; }
+       };
+
+       class X509Key
+       {
+               /** Ensure that the key is deinited in case the constructor of X509Key throws
+                */
+               class RAIIKey
+               {
+                public:
+                       gnutls_x509_privkey_t key;
+
+                       RAIIKey()
                        {
-                               user->WriteNumeric(670, "%s :STARTTLS successful, go ahead with TLS handshake", user->nick.c_str());
-                               /* We need to flush the write buffer prior to adding the IOHook,
-                                * otherwise we'll be sending this line inside the SSL session - which
-                                * won't start its handshake until the client gets this line. Currently,
-                                * we assume the write will not block here; this is usually safe, as
-                                * STARTTLS is sent very early on in the registration phase, where the
-                                * user hasn't built up much sendq. Handling a blocked write here would
-                                * be very annoying.
-                                */
-                               user->eh.DoWrite();
-                               user->eh.AddIOHook(creator);
-                               creator->OnStreamSocketAccept(&user->eh, NULL, NULL);
+                               ThrowOnError(gnutls_x509_privkey_init(&key), "gnutls_x509_privkey_init() failed");
                        }
-                       else
-                               user->WriteNumeric(691, "%s :STARTTLS failure", user->nick.c_str());
+
+                       ~RAIIKey()
+                       {
+                               gnutls_x509_privkey_deinit(key);
+                       }
+               } key;
+
+        public:
+               /** Import */
+               X509Key(const std::string& keystr)
+               {
+                       int ret = gnutls_x509_privkey_import(key.key, Datum(keystr).get(), GNUTLS_X509_FMT_PEM);
+                       ThrowOnError(ret, "Unable to import private key");
                }
 
-               return CMD_FAILURE;
-       }
-};
+               gnutls_x509_privkey_t& get() { return key.key; }
+       };
 
-class ModuleSSLGnuTLS : public Module
-{
-       issl_session* sessions;
+       class X509CertList
+       {
+               std::vector<gnutls_x509_crt_t> certs;
+
+        public:
+               /** Import */
+               X509CertList(const std::string& certstr)
+               {
+                       unsigned int certcount = 3;
+                       certs.resize(certcount);
+                       Datum datum(certstr);
 
-       gnutls_digest_algorithm_t hash;
+                       int ret = gnutls_x509_crt_list_import(raw(), &certcount, datum.get(), GNUTLS_X509_FMT_PEM, GNUTLS_X509_CRT_LIST_IMPORT_FAIL_IF_EXCEED);
+                       if (ret == GNUTLS_E_SHORT_MEMORY_BUFFER)
+                       {
+                               // the buffer wasn't big enough to hold all certs but gnutls changed certcount to the number of available certs,
+                               // try again with a bigger buffer
+                               certs.resize(certcount);
+                               ret = gnutls_x509_crt_list_import(raw(), &certcount, datum.get(), GNUTLS_X509_FMT_PEM, GNUTLS_X509_CRT_LIST_IMPORT_FAIL_IF_EXCEED);
+                       }
 
-       std::string sslports;
-       int dh_bits;
+                       ThrowOnError(ret, "Unable to load certificates");
 
-       RandGen randhandler;
-       CommandStartTLS starttls;
+                       // Resize the vector to the actual number of certs because we rely on its size being correct
+                       // when deallocating the certs
+                       certs.resize(certcount);
+               }
 
-       GenericCap capHandler;
-       ServiceProvider iohook;
+               ~X509CertList()
+               {
+                       for (std::vector<gnutls_x509_crt_t>::iterator i = certs.begin(); i != certs.end(); ++i)
+                               gnutls_x509_crt_deinit(*i);
+               }
 
-       inline static const char* UnknownIfNULL(const char* str)
-       {
-               return str ? str : "UNKNOWN";
-       }
+               gnutls_x509_crt_t* raw() { return &certs[0]; }
+               unsigned int size() const { return certs.size(); }
+       };
 
-       static ssize_t gnutls_pull_wrapper(gnutls_transport_ptr_t session_wrap, void* buffer, size_t size)
+       class X509CRL : public refcountbase
        {
-               issl_session* session = reinterpret_cast<issl_session*>(session_wrap);
-               if (session->socket->GetEventMask() & FD_READ_WILL_BLOCK)
+               class RAIICRL
                {
-#ifdef _WIN32
-                       gnutls_transport_set_errno(session->sess, EAGAIN);
-#else
-                       errno = EAGAIN;
-#endif
-                       return -1;
-               }
+                public:
+                       gnutls_x509_crl_t crl;
 
-               int rv = ServerInstance->SE->Recv(session->socket, reinterpret_cast<char *>(buffer), size, 0);
+                       RAIICRL()
+                       {
+                               ThrowOnError(gnutls_x509_crl_init(&crl), "gnutls_x509_crl_init() failed");
+                       }
 
-#ifdef _WIN32
-               if (rv < 0)
+                       ~RAIICRL()
+                       {
+                               gnutls_x509_crl_deinit(crl);
+                       }
+               } crl;
+
+        public:
+               /** Import */
+               X509CRL(const std::string& crlstr)
                {
-                       /* Windows doesn't use errno, but gnutls does, so check SocketEngine::IgnoreError()
-                        * and then set errno appropriately.
-                        * The gnutls library may also have a different errno variable than us, see
-                        * gnutls_transport_set_errno(3).
-                        */
-                       gnutls_transport_set_errno(session->sess, SocketEngine::IgnoreError() ? EAGAIN : errno);
+                       int ret = gnutls_x509_crl_import(get(), Datum(crlstr).get(), GNUTLS_X509_FMT_PEM);
+                       ThrowOnError(ret, "Unable to load certificate revocation list");
                }
-#endif
 
-               if (rv < (int)size)
-                       ServerInstance->SE->ChangeEventMask(session->socket, FD_READ_WILL_BLOCK);
-               return rv;
-       }
+               gnutls_x509_crl_t& get() { return crl.crl; }
+       };
 
-       static ssize_t gnutls_push_wrapper(gnutls_transport_ptr_t session_wrap, const void* buffer, size_t size)
+#ifdef GNUTLS_NEW_PRIO_API
+       class Priority
        {
-               issl_session* session = reinterpret_cast<issl_session*>(session_wrap);
-               if (session->socket->GetEventMask() & FD_WRITE_WILL_BLOCK)
+               gnutls_priority_t priority;
+
+        public:
+               Priority(const std::string& priorities)
                {
-#ifdef _WIN32
-                       gnutls_transport_set_errno(session->sess, EAGAIN);
-#else
-                       errno = EAGAIN;
-#endif
-                       return -1;
+                       // Try to set the priorities for ciphers, kex methods etc. to the user supplied string
+                       // If the user did not supply anything then the string is already set to "NORMAL"
+                       const char* priocstr = priorities.c_str();
+                       const char* prioerror;
+
+                       int ret = gnutls_priority_init(&priority, priocstr, &prioerror);
+                       if (ret < 0)
+                       {
+                               // gnutls did not understand the user supplied string
+                               throw Exception("Unable to initialize priorities to \"" + priorities + "\": " + gnutls_strerror(ret) + " Syntax error at position " + ConvToStr((unsigned int) (prioerror - priocstr)));
+                       }
                }
 
-               int rv = ServerInstance->SE->Send(session->socket, reinterpret_cast<const char *>(buffer), size, 0);
+               ~Priority()
+               {
+                       gnutls_priority_deinit(priority);
+               }
 
-#ifdef _WIN32
-               if (rv < 0)
+               void SetupSession(gnutls_session_t sess)
                {
-                       /* Windows doesn't use errno, but gnutls does, so check SocketEngine::IgnoreError()
-                        * and then set errno appropriately.
-                        * The gnutls library may also have a different errno variable than us, see
-                        * gnutls_transport_set_errno(3).
-                        */
-                       gnutls_transport_set_errno(session->sess, SocketEngine::IgnoreError() ? EAGAIN : errno);
+                       gnutls_priority_set(sess, priority);
+               }
+       };
+#else
+       /** Dummy class, used when gnutls_priority_set() is not available
+        */
+       class Priority
+       {
+        public:
+               Priority(const std::string& priorities)
+               {
+                       if (priorities != "NORMAL")
+                               throw Exception("You've set a non-default priority string, but GnuTLS lacks support for it");
+               }
+
+               static void SetupSession(gnutls_session_t sess)
+               {
+                       // Always set the default priorities
+                       gnutls_set_default_priority(sess);
                }
+       };
 #endif
 
-               if (rv < (int)size)
-                       ServerInstance->SE->ChangeEventMask(session->socket, FD_WRITE_WILL_BLOCK);
-               return rv;
-       }
+       class CertCredentials
+       {
+               /** DH parameters associated with these credentials
+                */
+               std::auto_ptr<DHParams> dh;
 
- public:
+        protected:
+               gnutls_certificate_credentials_t cred;
 
-       ModuleSSLGnuTLS()
-               : starttls(this), capHandler(this, "tls"), iohook(this, "ssl/gnutls", SERVICE_IOHOOK)
-       {
-#ifndef GNUTLS_HAS_RND
-               gcry_control (GCRYCTL_INITIALIZATION_FINISHED, 0);
-#endif
+        public:
+               CertCredentials()
+               {
+                       ThrowOnError(gnutls_certificate_allocate_credentials(&cred), "Cannot allocate certificate credentials");
+               }
 
-               sessions = new issl_session[ServerInstance->SE->GetMaxFds()];
+               ~CertCredentials()
+               {
+                       gnutls_certificate_free_credentials(cred);
+               }
 
-               gnutls_global_init(); // This must be called once in the program
-       }
+               /** Associates these credentials with the session
+                */
+               void SetupSession(gnutls_session_t sess)
+               {
+                       gnutls_credentials_set(sess, GNUTLS_CRD_CERTIFICATE, cred);
+               }
 
-       void init()
-       {
-               currconf = new SSLConfig;
-               InitSSLConfig(currconf);
+               /** Set the given DH parameters to be used with these credentials
+                */
+               void SetDH(std::auto_ptr<DHParams>& DH)
+               {
+                       dh = DH;
+                       gnutls_certificate_set_dh_params(cred, dh->get());
+               }
+       };
 
-               ServerInstance->GenRandom = &randhandler;
+       class X509Credentials : public CertCredentials
+       {
+               /** Private key
+                */
+               X509Key key;
 
-               Implementation eventlist[] = { I_On005Numeric, I_OnRehash, I_OnModuleRehash, I_OnUserConnect,
-                       I_OnEvent, I_OnHookIO };
-               ServerInstance->Modules->Attach(eventlist, this, sizeof(eventlist)/sizeof(Implementation));
+               /** Certificate list, presented to the peer
+                */
+               X509CertList certs;
 
-               ServerInstance->Modules->AddService(iohook);
-               ServerInstance->Modules->AddService(starttls);
-       }
+               /** Trusted CA, may be NULL
+                */
+               std::auto_ptr<X509CertList> trustedca;
 
-       void OnRehash(User* user)
-       {
-               sslports.clear();
+               /** Certificate revocation list, may be NULL
+                */
+               std::auto_ptr<X509CRL> crl;
 
-               ConfigTag* Conf = ServerInstance->Config->ConfValue("gnutls");
-               starttls.enabled = Conf->getBool("starttls", true);
+               static int cert_callback(gnutls_session_t session, const gnutls_datum_t* req_ca_rdn, int nreqs, const gnutls_pk_algorithm_t* sign_algos, int sign_algos_length, cert_cb_last_param_type* st);
 
-               if (Conf->getBool("showports", true))
+        public:
+               X509Credentials(const std::string& certstr, const std::string& keystr)
+                       : key(keystr)
+                       , certs(certstr)
                {
-                       sslports = Conf->getString("advertisedports");
-                       if (!sslports.empty())
-                               return;
+                       // Throwing is ok here, the destructor of Credentials is called in that case
+                       int ret = gnutls_certificate_set_x509_key(cred, certs.raw(), certs.size(), key.get());
+                       ThrowOnError(ret, "Unable to set cert/key pair");
 
-                       for (size_t i = 0; i < ServerInstance->ports.size(); i++)
-                       {
-                               ListenSocket* port = ServerInstance->ports[i];
-                               if (port->bind_tag->getString("ssl") != "gnutls")
-                                       continue;
+#ifdef GNUTLS_NEW_CERT_CALLBACK_API
+                       gnutls_certificate_set_retrieve_function(cred, cert_callback);
+#else
+                       gnutls_certificate_client_set_retrieve_function(cred, cert_callback);
+#endif
+               }
 
-                               const std::string& portid = port->bind_desc;
-                               ServerInstance->Logs->Log("m_ssl_gnutls", DEFAULT, "m_ssl_gnutls.so: Enabling SSL for port %s", portid.c_str());
+               /** Sets the trusted CA and the certificate revocation list
+                * to use when verifying certificates
+                */
+               void SetCA(std::auto_ptr<X509CertList>& certlist, std::auto_ptr<X509CRL>& CRL)
+               {
+                       // Do nothing if certlist is NULL
+                       if (certlist.get())
+                       {
+                               int ret = gnutls_certificate_set_x509_trust(cred, certlist->raw(), certlist->size());
+                               ThrowOnError(ret, "gnutls_certificate_set_x509_trust() failed");
 
-                               if (port->bind_tag->getString("type", "clients") == "clients" && port->bind_addr != "127.0.0.1")
+                               if (CRL.get())
                                {
-                                       /*
-                                        * Found an SSL port for clients that is not bound to 127.0.0.1 and handled by us, display
-                                        * the IP:port in ISUPPORT.
-                                        *
-                                        * We used to advertise all ports seperated by a ';' char that matched the above criteria,
-                                        * but this resulted in too long ISUPPORT lines if there were lots of ports to be displayed.
-                                        * To solve this by default we now only display the first IP:port found and let the user
-                                        * configure the exact value for the 005 token, if necessary.
-                                        */
-                                       sslports = portid;
-                                       break;
+                                       ret = gnutls_certificate_set_x509_crl(cred, &CRL->get(), 1);
+                                       ThrowOnError(ret, "gnutls_certificate_set_x509_crl() failed");
                                }
+
+                               trustedca = certlist;
+                               crl = CRL;
                        }
                }
-       }
+       };
 
-       void OnModuleRehash(User* user, const std::string &param)
+       class Profile : public refcountbase
        {
-               if(param != "ssl")
-                       return;
+               /** Name of this profile
+                */
+               const std::string name;
 
-               reference<SSLConfig> newconf = new SSLConfig;
-               try
+               /** X509 certificate(s) and key
+                */
+               X509Credentials x509cred;
+
+               /** The minimum length in bits for the DH prime to be accepted as a client
+                */
+               unsigned int min_dh_bits;
+
+               /** Hashing algorithm to use when generating certificate fingerprints
+                */
+               Hash hash;
+
+               /** Priorities for ciphers, compression methods, etc.
+                */
+               Priority priority;
+
+               Profile(const std::string& profilename, const std::string& certstr, const std::string& keystr,
+                               std::auto_ptr<DHParams>& DH, unsigned int mindh, const std::string& hashstr,
+                               const std::string& priostr, std::auto_ptr<X509CertList>& CA, std::auto_ptr<X509CRL>& CRL)
+                       : name(profilename)
+                       , x509cred(certstr, keystr)
+                       , min_dh_bits(mindh)
+                       , hash(hashstr)
+                       , priority(priostr)
                {
-                       InitSSLConfig(newconf);
+                       x509cred.SetDH(DH);
+                       x509cred.SetCA(CA, CRL);
                }
-               catch (ModuleException& ex)
+
+               static std::string ReadFile(const std::string& filename)
                {
-                       ServerInstance->Logs->Log("m_ssl_gnutls", DEFAULT, "m_ssl_gnutls: Not applying new config. %s", ex.GetReason());
-                       return;
+                       FileReader reader(filename);
+                       std::string ret = reader.GetString();
+                       if (ret.empty())
+                               throw Exception("Cannot read file " + filename);
+                       return ret;
                }
 
-               ServerInstance->Logs->Log("m_ssl_gnutls", DEFAULT, "m_ssl_gnutls: Applying new config, old config is in use by %d connection(s)", currconf->GetReferenceCount()-1);
-               currconf = newconf;
-       }
+        public:
+               static reference<Profile> Create(const std::string& profilename, ConfigTag* tag)
+               {
+                       std::string certstr = ReadFile(tag->getString("certfile", "cert.pem"));
+                       std::string keystr = ReadFile(tag->getString("keyfile", "key.pem"));
 
-       void InitSSLConfig(SSLConfig* config)
-       {
-               ServerInstance->Logs->Log("m_ssl_gnutls", DEBUG, "Initializing new SSLConfig %p", (void*)config);
+                       std::auto_ptr<DHParams> dh;
+                       int gendh = tag->getInt("gendh");
+                       if (gendh)
+                       {
+                               gendh = (gendh < 1024 ? 1024 : gendh);
+                               dh = DHParams::Generate(gendh);
+                       }
+                       else
+                               dh = DHParams::Import(ReadFile(tag->getString("dhfile", "dhparams.pem")));
+
+                       // Use default priority string if this tag does not specify one
+                       std::string priostr = tag->getString("priority", "NORMAL");
+                       unsigned int mindh = tag->getInt("mindhbits", 1024);
+                       std::string hashstr = tag->getString("hash", "md5");
+
+                       // Load trusted CA and revocation list, if set
+                       std::auto_ptr<X509CertList> ca;
+                       std::auto_ptr<X509CRL> crl;
+                       std::string filename = tag->getString("cafile");
+                       if (!filename.empty())
+                       {
+                               ca.reset(new X509CertList(ReadFile(filename)));
+
+                               filename = tag->getString("crlfile");
+                               if (!filename.empty())
+                                       crl.reset(new X509CRL(ReadFile(filename)));
+                       }
 
-               std::string keyfile;
-               std::string certfile;
-               std::string cafile;
-               std::string crlfile;
-               OnRehash(NULL);
+                       return new Profile(profilename, certstr, keystr, dh, mindh, hashstr, priostr, ca, crl);
+               }
 
-               ConfigTag* Conf = ServerInstance->Config->ConfValue("gnutls");
+               /** Set up the given session with the settings in this profile
+                */
+               void SetupSession(gnutls_session_t sess)
+               {
+                       priority.SetupSession(sess);
+                       x509cred.SetupSession(sess);
+                       gnutls_dh_set_prime_bits(sess, min_dh_bits);
+               }
 
-               cafile = Conf->getString("cafile", CONFIG_PATH "/ca.pem");
-               crlfile = Conf->getString("crlfile", CONFIG_PATH "/crl.pem");
-               certfile = Conf->getString("certfile", CONFIG_PATH "/cert.pem");
-               keyfile = Conf->getString("keyfile", CONFIG_PATH "/key.pem");
-               dh_bits = Conf->getInt("dhbits");
-               std::string hashname = Conf->getString("hash", "md5");
+               const std::string& GetName() const { return name; }
+               X509Credentials& GetX509Credentials() { return x509cred; }
+               gnutls_digest_algorithm_t GetHash() const { return hash.get(); }
+       };
+}
 
-               // The GnuTLS manual states that the gnutls_set_default_priority()
-               // call we used previously when initializing the session is the same
-               // as setting the "NORMAL" priority string.
-               // Thus if the setting below is not in the config we will behave exactly
-               // the same as before, when the priority setting wasn't available.
-               std::string priorities = Conf->getString("priority", "NORMAL");
+class GnuTLSIOHook : public SSLIOHook
+{
+ private:
+       gnutls_session_t sess;
+       issl_status status;
+       reference<GnuTLS::Profile> profile;
 
-               if((dh_bits != 768) && (dh_bits != 1024) && (dh_bits != 2048) && (dh_bits != 3072) && (dh_bits != 4096))
-                       dh_bits = 1024;
+       void InitSession(StreamSocket* user, bool me_server)
+       {
+               gnutls_init(&sess, me_server ? GNUTLS_SERVER : GNUTLS_CLIENT);
 
-               if (hashname == "md5")
-                       hash = GNUTLS_DIG_MD5;
-               else if (hashname == "sha1")
-                       hash = GNUTLS_DIG_SHA1;
-#ifdef INSPIRCD_GNUTLS_ENABLE_SHA256_FINGERPRINT
-               else if (hashname == "sha256")
-                       hash = GNUTLS_DIG_SHA256;
-#endif
-               else
-                       throw ModuleException("Unknown hash type " + hashname);
+               profile->SetupSession(sess);
+               gnutls_transport_set_ptr(sess, reinterpret_cast<gnutls_transport_ptr_t>(user));
+               gnutls_transport_set_push_function(sess, gnutls_push_wrapper);
+               gnutls_transport_set_pull_function(sess, gnutls_pull_wrapper);
 
+               if (me_server)
+                       gnutls_certificate_server_set_request(sess, GNUTLS_CERT_REQUEST); // Request client certificate if any.
+       }
 
-               int ret;
+       void CloseSession()
+       {
+               if (this->sess)
+               {
+                       gnutls_bye(this->sess, GNUTLS_SHUT_WR);
+                       gnutls_deinit(this->sess);
+               }
+               sess = NULL;
+               certificate = NULL;
+               status = ISSL_NONE;
+       }
 
-               gnutls_certificate_credentials_t& x509_cred = config->x509_cred;
+       bool Handshake(StreamSocket* user)
+       {
+               int ret = gnutls_handshake(this->sess);
 
-               ret = gnutls_certificate_allocate_credentials(&x509_cred);
                if (ret < 0)
                {
-                       // Set to NULL because we can't be sure what value is in it and we must not try to
-                       // deallocate it in case of an error
-                       x509_cred = NULL;
-                       throw ModuleException("Failed to allocate certificate credentials: " + std::string(gnutls_strerror(ret)));
-               }
+                       if(ret == GNUTLS_E_AGAIN || ret == GNUTLS_E_INTERRUPTED)
+                       {
+                               // Handshake needs resuming later, read() or write() would have blocked.
 
-               if((ret =gnutls_certificate_set_x509_trust_file(x509_cred, cafile.c_str(), GNUTLS_X509_FMT_PEM)) < 0)
-                       ServerInstance->Logs->Log("m_ssl_gnutls",DEBUG, "m_ssl_gnutls.so: Failed to set X.509 trust file '%s': %s", cafile.c_str(), gnutls_strerror(ret));
+                               if (gnutls_record_get_direction(this->sess) == 0)
+                               {
+                                       // gnutls_handshake() wants to read() again.
+                                       this->status = ISSL_HANDSHAKING_READ;
+                                       SocketEngine::ChangeEventMask(user, FD_WANT_POLL_READ | FD_WANT_NO_WRITE);
+                               }
+                               else
+                               {
+                                       // gnutls_handshake() wants to write() again.
+                                       this->status = ISSL_HANDSHAKING_WRITE;
+                                       SocketEngine::ChangeEventMask(user, FD_WANT_NO_READ | FD_WANT_SINGLE_WRITE);
+                               }
+                       }
+                       else
+                       {
+                               user->SetError("Handshake Failed - " + std::string(gnutls_strerror(ret)));
+                               CloseSession();
+                               this->status = ISSL_CLOSING;
+                       }
+
+                       return false;
+               }
+               else
+               {
+                       // Change the seesion state
+                       this->status = ISSL_HANDSHAKEN;
 
-               if((ret = gnutls_certificate_set_x509_crl_file (x509_cred, crlfile.c_str(), GNUTLS_X509_FMT_PEM)) < 0)
-                       ServerInstance->Logs->Log("m_ssl_gnutls",DEBUG, "m_ssl_gnutls.so: Failed to set X.509 CRL file '%s': %s", crlfile.c_str(), gnutls_strerror(ret));
+                       VerifyCertificate();
 
-               FileReader reader;
+                       // Finish writing, if any left
+                       SocketEngine::ChangeEventMask(user, FD_WANT_POLL_READ | FD_WANT_NO_WRITE | FD_ADD_TRIAL_WRITE);
 
-               reader.LoadFile(certfile);
-               std::string cert_string = reader.Contents();
-               gnutls_datum_t cert_datum = { (unsigned char*)cert_string.data(), static_cast<unsigned int>(cert_string.length()) };
+                       return true;
+               }
+       }
 
-               reader.LoadFile(keyfile);
-               std::string key_string = reader.Contents();
-               gnutls_datum_t key_datum = { (unsigned char*)key_string.data(), static_cast<unsigned int>(key_string.length()) };
+       void VerifyCertificate()
+       {
+               unsigned int certstatus;
+               const gnutls_datum_t* cert_list;
+               int ret;
+               unsigned int cert_list_size;
+               gnutls_x509_crt_t cert;
+               char str[512];
+               unsigned char digest[512];
+               size_t digest_size = sizeof(digest);
+               size_t name_size = sizeof(str);
+               ssl_cert* certinfo = new ssl_cert;
+               this->certificate = certinfo;
 
-               std::vector<gnutls_x509_crt_t>& x509_certs = config->x509_certs;
+               /* This verification function uses the trusted CAs in the credentials
+                * structure. So you must have installed one or more CA certificates.
+                */
+               ret = gnutls_certificate_verify_peers2(this->sess, &certstatus);
 
-               // If this fails, no SSL port will work. At all. So, do the smart thing - throw a ModuleException
-               unsigned int certcount = 3;
-               x509_certs.resize(certcount);
-               ret = gnutls_x509_crt_list_import(&x509_certs[0], &certcount, &cert_datum, GNUTLS_X509_FMT_PEM, GNUTLS_X509_CRT_LIST_IMPORT_FAIL_IF_EXCEED);
-               if (ret == GNUTLS_E_SHORT_MEMORY_BUFFER)
+               if (ret < 0)
                {
-                       // the buffer wasn't big enough to hold all certs but gnutls updated certcount to the number of available certs, try again with a bigger buffer
-                       x509_certs.resize(certcount);
-                       ret = gnutls_x509_crt_list_import(&x509_certs[0], &certcount, &cert_datum, GNUTLS_X509_FMT_PEM, GNUTLS_X509_CRT_LIST_IMPORT_FAIL_IF_EXCEED);
+                       certinfo->error = std::string(gnutls_strerror(ret));
+                       return;
                }
 
-               if (ret <= 0)
+               certinfo->invalid = (certstatus & GNUTLS_CERT_INVALID);
+               certinfo->unknownsigner = (certstatus & GNUTLS_CERT_SIGNER_NOT_FOUND);
+               certinfo->revoked = (certstatus & GNUTLS_CERT_REVOKED);
+               certinfo->trusted = !(certstatus & GNUTLS_CERT_SIGNER_NOT_CA);
+
+               /* Up to here the process is the same for X.509 certificates and
+                * OpenPGP keys. From now on X.509 certificates are assumed. This can
+                * be easily extended to work with openpgp keys as well.
+                */
+               if (gnutls_certificate_type_get(this->sess) != GNUTLS_CRT_X509)
                {
-                       // clear the vector so we won't call gnutls_x509_crt_deinit() on the (uninited) certs later
-                       x509_certs.clear();
-                       throw ModuleException("Unable to load GnuTLS server certificate (" + certfile + "): " + ((ret < 0) ? (std::string(gnutls_strerror(ret))) : "No certs could be read"));
+                       certinfo->error = "No X509 keys sent";
+                       return;
                }
-               x509_certs.resize(ret);
 
-               gnutls_x509_privkey_t& x509_key = config->x509_key;
-               if (gnutls_x509_privkey_init(&x509_key) < 0)
+               ret = gnutls_x509_crt_init(&cert);
+               if (ret < 0)
                {
-                       // Make sure the destructor does not try to deallocate this, see above
-                       x509_key = NULL;
-                       throw ModuleException("Unable to initialize private key");
+                       certinfo->error = gnutls_strerror(ret);
+                       return;
                }
 
-               if((ret = gnutls_x509_privkey_import(x509_key, &key_datum, GNUTLS_X509_FMT_PEM)) < 0)
-                       throw ModuleException("Unable to load GnuTLS server private key (" + keyfile + "): " + std::string(gnutls_strerror(ret)));
-
-               if((ret = gnutls_certificate_set_x509_key(x509_cred, &x509_certs[0], certcount, x509_key)) < 0)
-                       throw ModuleException("Unable to set GnuTLS cert/key pair: " + std::string(gnutls_strerror(ret)));
-
-               #ifdef GNUTLS_NEW_PRIO_API
-               // Try to set the priorities for ciphers, kex methods etc. to the user supplied string
-               // If the user did not supply anything then the string is already set to "NORMAL"
-               const char* priocstr = priorities.c_str();
-               const char* prioerror;
-
-               gnutls_priority_t& priority = config->priority;
-               if ((ret = gnutls_priority_init(&priority, priocstr, &prioerror)) < 0)
+               cert_list_size = 0;
+               cert_list = gnutls_certificate_get_peers(this->sess, &cert_list_size);
+               if (cert_list == NULL)
                {
-                       // gnutls did not understand the user supplied string, log and fall back to the default priorities
-                       ServerInstance->Logs->Log("m_ssl_gnutls",DEFAULT, "m_ssl_gnutls.so: Failed to set priorities to \"%s\": %s Syntax error at position %u, falling back to default (NORMAL)", priorities.c_str(), gnutls_strerror(ret), (unsigned int) (prioerror - priocstr));
-                       gnutls_priority_init(&priority, "NORMAL", NULL);
+                       certinfo->error = "No certificate was found";
+                       goto info_done_dealloc;
                }
 
-               #else
-               if (priorities != "NORMAL")
-                       ServerInstance->Logs->Log("m_ssl_gnutls",DEFAULT, "m_ssl_gnutls.so: You've set <gnutls:priority> to a value other than the default, but this is only supported with GnuTLS v2.1.7 or newer. Your GnuTLS version is older than that so the option will have no effect.");
-               #endif
-
-               #if(GNUTLS_VERSION_MAJOR < 2 || ( GNUTLS_VERSION_MAJOR == 2 && GNUTLS_VERSION_MINOR < 12 ) )
-               gnutls_certificate_client_set_retrieve_function (x509_cred, cert_callback);
-               #else
-               gnutls_certificate_set_retrieve_function (x509_cred, cert_callback);
-               #endif
+               /* This is not a real world example, since we only check the first
+                * certificate in the given chain.
+                */
 
-               gnutls_dh_params_t& dh_params = config->dh_params;
-               ret = gnutls_dh_params_init(&dh_params);
+               ret = gnutls_x509_crt_import(cert, &cert_list[0], GNUTLS_X509_FMT_DER);
                if (ret < 0)
                {
-                       // Make sure the destructor does not try to deallocate this, see above
-                       dh_params = NULL;
-                       ServerInstance->Logs->Log("m_ssl_gnutls",DEFAULT, "m_ssl_gnutls.so: Failed to initialise DH parameters: %s", gnutls_strerror(ret));
-                       return;
+                       certinfo->error = gnutls_strerror(ret);
+                       goto info_done_dealloc;
                }
 
-               std::string dhfile = Conf->getString("dhfile");
-               if (!dhfile.empty())
+               if (gnutls_x509_crt_get_dn(cert, str, &name_size) == 0)
                {
-                       // Try to load DH params from file
-                       reader.LoadFile(dhfile);
-                       std::string dhstring = reader.Contents();
-                       gnutls_datum_t dh_datum = { (unsigned char*)dhstring.data(), static_cast<unsigned int>(dhstring.length()) };
+                       std::string& dn = certinfo->dn;
+                       dn = str;
+                       // Make sure there are no chars in the string that we consider invalid
+                       if (dn.find_first_of("\r\n") != std::string::npos)
+                               dn.clear();
+               }
 
-                       if ((ret = gnutls_dh_params_import_pkcs3(dh_params, &dh_datum, GNUTLS_X509_FMT_PEM)) < 0)
-                       {
-                               // File unreadable or GnuTLS was unhappy with the contents, generate the DH primes now
-                               ServerInstance->Logs->Log("m_ssl_gnutls", DEFAULT, "m_ssl_gnutls.so: Generating DH parameters because I failed to load them from file '%s': %s", dhfile.c_str(), gnutls_strerror(ret));
-                               GenerateDHParams(dh_params);
-                       }
+               name_size = sizeof(str);
+               if (gnutls_x509_crt_get_issuer_dn(cert, str, &name_size) == 0)
+               {
+                       std::string& issuer = certinfo->issuer;
+                       issuer = str;
+                       if (issuer.find_first_of("\r\n") != std::string::npos)
+                               issuer.clear();
+               }
+
+               if ((ret = gnutls_x509_crt_get_fingerprint(cert, profile->GetHash(), digest, &digest_size)) < 0)
+               {
+                       certinfo->error = gnutls_strerror(ret);
+               }
+               else
+               {
+                       certinfo->fingerprint = BinToHex(digest, digest_size);
                }
-               else
+
+               /* Beware here we do not check for errors.
+                */
+               if ((gnutls_x509_crt_get_expiration_time(cert) < ServerInstance->Time()) || (gnutls_x509_crt_get_activation_time(cert) > ServerInstance->Time()))
                {
-                       GenerateDHParams(dh_params);
+                       certinfo->error = "Not activated, or expired certificate";
                }
 
-               gnutls_certificate_set_dh_params(x509_cred, dh_params);
+info_done_dealloc:
+               gnutls_x509_crt_deinit(cert);
        }
 
-       void GenerateDHParams(gnutls_dh_params_t dh_params)
+       static const char* UnknownIfNULL(const char* str)
        {
-               // Generate Diffie Hellman parameters - for use with DHE
-               // kx algorithms. These should be discarded and regenerated
-               // once a day, once a week or once a month. Depending on the
-               // security requirements.
-
-               int ret;
-
-               if((ret = gnutls_dh_params_generate2(dh_params, dh_bits)) < 0)
-                       ServerInstance->Logs->Log("m_ssl_gnutls",DEFAULT, "m_ssl_gnutls.so: Failed to generate DH parameters (%d bits): %s", dh_bits, gnutls_strerror(ret));
+               return str ? str : "UNKNOWN";
        }
 
-       ~ModuleSSLGnuTLS()
+       static ssize_t gnutls_pull_wrapper(gnutls_transport_ptr_t session_wrap, void* buffer, size_t size)
        {
-               currconf = NULL;
-
-               gnutls_global_deinit();
-               delete[] sessions;
-               ServerInstance->GenRandom = &ServerInstance->HandleGenRandom;
-       }
+               StreamSocket* sock = reinterpret_cast<StreamSocket*>(session_wrap);
+#ifdef _WIN32
+               GnuTLSIOHook* session = static_cast<GnuTLSIOHook*>(sock->GetIOHook());
+#endif
 
-       void OnCleanup(int target_type, void* item)
-       {
-               if(target_type == TYPE_USER)
+               if (sock->GetEventMask() & FD_READ_WILL_BLOCK)
                {
-                       LocalUser* user = IS_LOCAL(static_cast<User*>(item));
-
-                       if (user && user->eh.GetIOHook() == this)
-                       {
-                               // User is using SSL, they're a local user, and they're using one of *our* SSL ports.
-                               // Potentially there could be multiple SSL modules loaded at once on different ports.
-                               ServerInstance->Users->QuitUser(user, "SSL module unloading");
-                       }
+#ifdef _WIN32
+                       gnutls_transport_set_errno(session->sess, EAGAIN);
+#else
+                       errno = EAGAIN;
+#endif
+                       return -1;
                }
-       }
-
-       Version GetVersion()
-       {
-               return Version("Provides SSL support for clients", VF_VENDOR);
-       }
-
 
-       void On005Numeric(std::string &output)
-       {
-               if (!sslports.empty())
-                       output.append(" SSL=" + sslports);
-               if (starttls.enabled)
-                       output.append(" STARTTLS");
-       }
+               int rv = SocketEngine::Recv(sock, reinterpret_cast<char *>(buffer), size, 0);
 
-       void OnHookIO(StreamSocket* user, ListenSocket* lsb)
-       {
-               if (!user->GetIOHook() && lsb->bind_tag->getString("ssl") == "gnutls")
+#ifdef _WIN32
+               if (rv < 0)
                {
-                       /* Hook the user with our module */
-                       user->AddIOHook(this);
+                       /* Windows doesn't use errno, but gnutls does, so check SocketEngine::IgnoreError()
+                        * and then set errno appropriately.
+                        * The gnutls library may also have a different errno variable than us, see
+                        * gnutls_transport_set_errno(3).
+                        */
+                       gnutls_transport_set_errno(session->sess, SocketEngine::IgnoreError() ? EAGAIN : errno);
                }
+#endif
+
+               if (rv < (int)size)
+                       SocketEngine::ChangeEventMask(sock, FD_READ_WILL_BLOCK);
+               return rv;
        }
 
-       void OnRequest(Request& request)
+       static ssize_t gnutls_push_wrapper(gnutls_transport_ptr_t session_wrap, const void* buffer, size_t size)
        {
-               if (strcmp("GET_SSL_CERT", request.id) == 0)
-               {
-                       SocketCertificateRequest& req = static_cast<SocketCertificateRequest&>(request);
-                       int fd = req.sock->GetFd();
-                       issl_session* session = &sessions[fd];
+               StreamSocket* sock = reinterpret_cast<StreamSocket*>(session_wrap);
+#ifdef _WIN32
+               GnuTLSIOHook* session = static_cast<GnuTLSIOHook*>(sock->GetIOHook());
+#endif
 
-                       req.cert = session->cert;
-               }
-               else if (!strcmp("GET_RAW_SSL_SESSION", request.id))
+               if (sock->GetEventMask() & FD_WRITE_WILL_BLOCK)
                {
-                       SSLRawSessionRequest& req = static_cast<SSLRawSessionRequest&>(request);
-                       if ((req.fd >= 0) && (req.fd < ServerInstance->SE->GetMaxFds()))
-                               req.data = reinterpret_cast<void*>(sessions[req.fd].sess);
+#ifdef _WIN32
+                       gnutls_transport_set_errno(session->sess, EAGAIN);
+#else
+                       errno = EAGAIN;
+#endif
+                       return -1;
                }
-       }
-
-       void InitSession(StreamSocket* user, bool me_server)
-       {
-               issl_session* session = &sessions[user->GetFd()];
-
-               gnutls_init(&session->sess, me_server ? GNUTLS_SERVER : GNUTLS_CLIENT);
-               session->socket = user;
-               session->config = currconf;
-
-               #ifdef GNUTLS_NEW_PRIO_API
-               gnutls_priority_set(session->sess, currconf->priority);
-               #else
-               gnutls_set_default_priority(session->sess);
-               #endif
-               gnutls_credentials_set(session->sess, GNUTLS_CRD_CERTIFICATE, currconf->x509_cred);
-               gnutls_dh_set_prime_bits(session->sess, dh_bits);
-               gnutls_transport_set_ptr(session->sess, reinterpret_cast<gnutls_transport_ptr_t>(session));
-               gnutls_transport_set_push_function(session->sess, gnutls_push_wrapper);
-               gnutls_transport_set_pull_function(session->sess, gnutls_pull_wrapper);
 
-               if (me_server)
-                       gnutls_certificate_server_set_request(session->sess, GNUTLS_CERT_REQUEST); // Request client certificate if any.
-
-               Handshake(session, user);
-       }
-
-       void OnStreamSocketAccept(StreamSocket* user, irc::sockets::sockaddrs* client, irc::sockets::sockaddrs* server)
-       {
-               issl_session* session = &sessions[user->GetFd()];
+               int rv = SocketEngine::Send(sock, reinterpret_cast<const char *>(buffer), size, 0);
 
-               /* For STARTTLS: Don't try and init a session on a socket that already has a session */
-               if (session->sess)
-                       return;
+#ifdef _WIN32
+               if (rv < 0)
+               {
+                       /* Windows doesn't use errno, but gnutls does, so check SocketEngine::IgnoreError()
+                        * and then set errno appropriately.
+                        * The gnutls library may also have a different errno variable than us, see
+                        * gnutls_transport_set_errno(3).
+                        */
+                       gnutls_transport_set_errno(session->sess, SocketEngine::IgnoreError() ? EAGAIN : errno);
+               }
+#endif
 
-               InitSession(user, true);
+               if (rv < (int)size)
+                       SocketEngine::ChangeEventMask(sock, FD_WRITE_WILL_BLOCK);
+               return rv;
        }
 
-       void OnStreamSocketConnect(StreamSocket* user)
+ public:
+       GnuTLSIOHook(IOHookProvider* hookprov, StreamSocket* sock, bool outbound, const reference<GnuTLS::Profile>& sslprofile)
+               : SSLIOHook(hookprov)
+               , sess(NULL)
+               , status(ISSL_NONE)
+               , profile(sslprofile)
        {
-               InitSession(user, false);
+               InitSession(sock, outbound);
+               sock->AddIOHook(this);
+               Handshake(sock);
        }
 
-       void OnStreamSocketClose(StreamSocket* user)
+       void OnStreamSocketClose(StreamSocket* user) CXX11_OVERRIDE
        {
-               CloseSession(&sessions[user->GetFd()]);
+               CloseSession();
        }
 
-       int OnStreamSocketRead(StreamSocket* user, std::string& recvq)
+       int OnStreamSocketRead(StreamSocket* user, std::string& recvq) CXX11_OVERRIDE
        {
-               issl_session* session = &sessions[user->GetFd()];
-
-               if (!session->sess)
+               if (!this->sess)
                {
-                       CloseSession(session);
+                       CloseSession();
                        user->SetError("No SSL session");
                        return -1;
                }
 
-               if (session->status == ISSL_HANDSHAKING_READ || session->status == ISSL_HANDSHAKING_WRITE)
+               if (this->status == ISSL_HANDSHAKING_READ || this->status == ISSL_HANDSHAKING_WRITE)
                {
                        // The handshake isn't finished, try to finish it.
 
-                       if(!Handshake(session, user))
+                       if (!Handshake(user))
                        {
-                               if (session->status != ISSL_CLOSING)
+                               if (this->status != ISSL_CLOSING)
                                        return 0;
                                return -1;
                        }
                }
 
-               // If we resumed the handshake then session->status will be ISSL_HANDSHAKEN.
+               // If we resumed the handshake then this->status will be ISSL_HANDSHAKEN.
 
-               if (session->status == ISSL_HANDSHAKEN)
+               if (this->status == ISSL_HANDSHAKEN)
                {
                        char* buffer = ServerInstance->GetReadBuffer();
                        size_t bufsiz = ServerInstance->Config->NetBufferSize;
-                       int ret = gnutls_record_recv(session->sess, buffer, bufsiz);
+                       int ret = gnutls_record_recv(this->sess, buffer, bufsiz);
                        if (ret > 0)
                        {
                                recvq.append(buffer, ret);
@@ -712,68 +862,66 @@ class ModuleSSLGnuTLS : public Module
                        else if (ret == 0)
                        {
                                user->SetError("Connection closed");
-                               CloseSession(session);
+                               CloseSession();
                                return -1;
                        }
                        else
                        {
                                user->SetError(gnutls_strerror(ret));
-                               CloseSession(session);
+                               CloseSession();
                                return -1;
                        }
                }
-               else if (session->status == ISSL_CLOSING)
+               else if (this->status == ISSL_CLOSING)
                        return -1;
 
                return 0;
        }
 
-       int OnStreamSocketWrite(StreamSocket* user, std::string& sendq)
+       int OnStreamSocketWrite(StreamSocket* user, std::string& sendq) CXX11_OVERRIDE
        {
-               issl_session* session = &sessions[user->GetFd()];
-
-               if (!session->sess)
+               if (!this->sess)
                {
-                       CloseSession(session);
+                       CloseSession();
                        user->SetError("No SSL session");
                        return -1;
                }
 
-               if (session->status == ISSL_HANDSHAKING_WRITE || session->status == ISSL_HANDSHAKING_READ)
+               if (this->status == ISSL_HANDSHAKING_WRITE || this->status == ISSL_HANDSHAKING_READ)
                {
                        // The handshake isn't finished, try to finish it.
-                       Handshake(session, user);
-                       if (session->status != ISSL_CLOSING)
+                       Handshake(user);
+                       if (this->status != ISSL_CLOSING)
                                return 0;
                        return -1;
                }
 
                int ret = 0;
 
-               if (session->status == ISSL_HANDSHAKEN)
+               if (this->status == ISSL_HANDSHAKEN)
                {
-                       ret = gnutls_record_send(session->sess, sendq.data(), sendq.length());
+                       ret = gnutls_record_send(this->sess, sendq.data(), sendq.length());
 
                        if (ret == (int)sendq.length())
                        {
-                               ServerInstance->SE->ChangeEventMask(user, FD_WANT_NO_WRITE);
+                               SocketEngine::ChangeEventMask(user, FD_WANT_NO_WRITE);
                                return 1;
                        }
                        else if (ret > 0)
                        {
                                sendq = sendq.substr(ret);
-                               ServerInstance->SE->ChangeEventMask(user, FD_WANT_SINGLE_WRITE);
+                               SocketEngine::ChangeEventMask(user, FD_WANT_SINGLE_WRITE);
                                return 0;
                        }
                        else if (ret == GNUTLS_E_AGAIN || ret == GNUTLS_E_INTERRUPTED || ret == 0)
                        {
-                               ServerInstance->SE->ChangeEventMask(user, FD_WANT_SINGLE_WRITE);
+                               SocketEngine::ChangeEventMask(user, FD_WANT_SINGLE_WRITE);
                                return 0;
                        }
                        else // (ret < 0)
                        {
                                user->SetError(gnutls_strerror(ret));
-                               CloseSession(session);
+                               CloseSession();
                                return -1;
                        }
                }
@@ -781,198 +929,198 @@ class ModuleSSLGnuTLS : public Module
                return 0;
        }
 
-       bool Handshake(issl_session* session, StreamSocket* user)
+       void TellCiphersAndFingerprint(LocalUser* user)
        {
-               int ret = gnutls_handshake(session->sess);
-
-               if (ret < 0)
+               if (sess)
                {
-                       if(ret == GNUTLS_E_AGAIN || ret == GNUTLS_E_INTERRUPTED)
-                       {
-                               // Handshake needs resuming later, read() or write() would have blocked.
+                       std::string text = "*** You are connected using SSL cipher '";
 
-                               if(gnutls_record_get_direction(session->sess) == 0)
-                               {
-                                       // gnutls_handshake() wants to read() again.
-                                       session->status = ISSL_HANDSHAKING_READ;
-                                       ServerInstance->SE->ChangeEventMask(user, FD_WANT_POLL_READ | FD_WANT_NO_WRITE);
-                               }
-                               else
-                               {
-                                       // gnutls_handshake() wants to write() again.
-                                       session->status = ISSL_HANDSHAKING_WRITE;
-                                       ServerInstance->SE->ChangeEventMask(user, FD_WANT_NO_READ | FD_WANT_SINGLE_WRITE);
-                               }
-                       }
-                       else
-                       {
-                               user->SetError("Handshake Failed - " + std::string(gnutls_strerror(ret)));
-                               CloseSession(session);
-                               session->status = ISSL_CLOSING;
-                       }
+                       text += UnknownIfNULL(gnutls_kx_get_name(gnutls_kx_get(sess)));
+                       text.append("-").append(UnknownIfNULL(gnutls_cipher_get_name(gnutls_cipher_get(sess)))).append("-");
+                       text.append(UnknownIfNULL(gnutls_mac_get_name(gnutls_mac_get(sess)))).append("'");
 
-                       return false;
+                       if (!certificate->fingerprint.empty())
+                               text += " and your SSL certificate fingerprint is " + certificate->fingerprint;
+
+                       user->WriteNotice(text);
                }
-               else
-               {
-                       // Change the seesion state
-                       session->status = ISSL_HANDSHAKEN;
+       }
 
-                       VerifyCertificate(session,user);
+       GnuTLS::Profile* GetProfile() { return profile; }
+};
 
-                       // Finish writing, if any left
-                       ServerInstance->SE->ChangeEventMask(user, FD_WANT_POLL_READ | FD_WANT_NO_WRITE | FD_ADD_TRIAL_WRITE);
+int GnuTLS::X509Credentials::cert_callback(gnutls_session_t sess, const gnutls_datum_t* req_ca_rdn, int nreqs, const gnutls_pk_algorithm_t* sign_algos, int sign_algos_length, cert_cb_last_param_type* st)
+{
+#ifndef GNUTLS_NEW_CERT_CALLBACK_API
+       st->type = GNUTLS_CRT_X509;
+#else
+       st->cert_type = GNUTLS_CRT_X509;
+       st->key_type = GNUTLS_PRIVKEY_X509;
+#endif
+       StreamSocket* sock = reinterpret_cast<StreamSocket*>(gnutls_transport_get_ptr(sess));
+       GnuTLS::X509Credentials& cred = static_cast<GnuTLSIOHook*>(sock->GetIOHook())->GetProfile()->GetX509Credentials();
 
-                       return true;
-               }
-       }
+       st->ncerts = cred.certs.size();
+       st->cert.x509 = cred.certs.raw();
+       st->key.x509 = cred.key.get();
+       st->deinit_all = 0;
+
+       return 0;
+}
 
-       void OnUserConnect(LocalUser* user)
+class GnuTLSIOHookProvider : public refcountbase, public IOHookProvider
+{
+       reference<GnuTLS::Profile> profile;
+
+ public:
+       GnuTLSIOHookProvider(Module* mod, reference<GnuTLS::Profile>& prof)
+               : IOHookProvider(mod, "ssl/" + prof->GetName(), IOHookProvider::IOH_SSL)
+               , profile(prof)
        {
-               if (user->eh.GetIOHook() == this)
-               {
-                       if (sessions[user->eh.GetFd()].sess)
-                       {
-                               const gnutls_session_t& sess = sessions[user->eh.GetFd()].sess;
-                               std::string cipher = UnknownIfNULL(gnutls_kx_get_name(gnutls_kx_get(sess)));
-                               cipher.append("-").append(UnknownIfNULL(gnutls_cipher_get_name(gnutls_cipher_get(sess)))).append("-");
-                               cipher.append(UnknownIfNULL(gnutls_mac_get_name(gnutls_mac_get(sess))));
-
-                               ssl_cert* cert = sessions[user->eh.GetFd()].cert;
-                               if (cert->fingerprint.empty())
-                                       user->WriteServ("NOTICE %s :*** You are connected using SSL cipher \"%s\"", user->nick.c_str(), cipher.c_str());
-                               else
-                                       user->WriteServ("NOTICE %s :*** You are connected using SSL cipher \"%s\""
-                                               " and your SSL fingerprint is %s", user->nick.c_str(), cipher.c_str(), cert->fingerprint.c_str());
-                       }
-               }
+               ServerInstance->Modules->AddService(*this);
        }
 
-       void CloseSession(issl_session* session)
+       ~GnuTLSIOHookProvider()
        {
-               if (session->sess)
-               {
-                       gnutls_bye(session->sess, GNUTLS_SHUT_WR);
-                       gnutls_deinit(session->sess);
-               }
-               session->socket = NULL;
-               session->sess = NULL;
-               session->cert = NULL;
-               session->status = ISSL_NONE;
-               session->config = NULL;
+               ServerInstance->Modules->DelService(*this);
        }
 
-       void VerifyCertificate(issl_session* session, StreamSocket* user)
+       void OnAccept(StreamSocket* sock, irc::sockets::sockaddrs* client, irc::sockets::sockaddrs* server) CXX11_OVERRIDE
        {
-               if (!session->sess || !user)
-                       return;
+               new GnuTLSIOHook(this, sock, true, profile);
+       }
 
-               unsigned int status;
-               const gnutls_datum_t* cert_list;
-               int ret;
-               unsigned int cert_list_size;
-               gnutls_x509_crt_t cert;
-               char name[MAXBUF];
-               unsigned char digest[MAXBUF];
-               size_t digest_size = sizeof(digest);
-               size_t name_size = sizeof(name);
-               ssl_cert* certinfo = new ssl_cert;
-               session->cert = certinfo;
+       void OnConnect(StreamSocket* sock) CXX11_OVERRIDE
+       {
+               new GnuTLSIOHook(this, sock, false, profile);
+       }
+};
 
-               /* This verification function uses the trusted CAs in the credentials
-                * structure. So you must have installed one or more CA certificates.
-                */
-               ret = gnutls_certificate_verify_peers2(session->sess, &status);
+class ModuleSSLGnuTLS : public Module
+{
+       typedef std::vector<reference<GnuTLSIOHookProvider> > ProfileList;
 
-               if (ret < 0)
-               {
-                       certinfo->error = std::string(gnutls_strerror(ret));
-                       return;
-               }
+       // First member of the class, gets constructed first and destructed last
+       GnuTLS::Init libinit;
+       RandGen randhandler;
+       ProfileList profiles;
 
-               certinfo->invalid = (status & GNUTLS_CERT_INVALID);
-               certinfo->unknownsigner = (status & GNUTLS_CERT_SIGNER_NOT_FOUND);
-               certinfo->revoked = (status & GNUTLS_CERT_REVOKED);
-               certinfo->trusted = !(status & GNUTLS_CERT_SIGNER_NOT_CA);
+       void ReadProfiles()
+       {
+               // First, store all profiles in a new, temporary container. If no problems occur, swap the two
+               // containers; this way if something goes wrong we can go back and continue using the current profiles,
+               // avoiding unpleasant situations where no new SSL connections are possible.
+               ProfileList newprofiles;
 
-               /* Up to here the process is the same for X.509 certificates and
-                * OpenPGP keys. From now on X.509 certificates are assumed. This can
-                * be easily extended to work with openpgp keys as well.
-                */
-               if (gnutls_certificate_type_get(session->sess) != GNUTLS_CRT_X509)
+               ConfigTagList tags = ServerInstance->Config->ConfTags("sslprofile");
+               if (tags.first == tags.second)
                {
-                       certinfo->error = "No X509 keys sent";
-                       return;
-               }
+                       // No <sslprofile> tags found, create a profile named "gnutls" from settings in the <gnutls> block
+                       const std::string defname = "gnutls";
+                       ConfigTag* tag = ServerInstance->Config->ConfValue(defname);
+                       ServerInstance->Logs->Log(MODNAME, LOG_DEFAULT, "No <sslprofile> tags found; using settings from the <gnutls> tag");
 
-               ret = gnutls_x509_crt_init(&cert);
-               if (ret < 0)
-               {
-                       certinfo->error = gnutls_strerror(ret);
-                       return;
+                       try
+                       {
+                               reference<GnuTLS::Profile> profile(GnuTLS::Profile::Create(defname, tag));
+                               newprofiles.push_back(new GnuTLSIOHookProvider(this, profile));
+                       }
+                       catch (CoreException& ex)
+                       {
+                               throw ModuleException("Error while initializing the default SSL profile - " + ex.GetReason());
+                       }
                }
 
-               cert_list_size = 0;
-               cert_list = gnutls_certificate_get_peers(session->sess, &cert_list_size);
-               if (cert_list == NULL)
+               for (ConfigIter i = tags.first; i != tags.second; ++i)
                {
-                       certinfo->error = "No certificate was found";
-                       goto info_done_dealloc;
-               }
+                       ConfigTag* tag = i->second;
+                       if (tag->getString("provider") != "gnutls")
+                               continue;
 
-               /* This is not a real world example, since we only check the first
-                * certificate in the given chain.
-                */
+                       std::string name = tag->getString("name");
+                       if (name.empty())
+                       {
+                               ServerInstance->Logs->Log(MODNAME, LOG_DEFAULT, "Ignoring <sslprofile> tag without name at " + tag->getTagLocation());
+                               continue;
+                       }
 
-               ret = gnutls_x509_crt_import(cert, &cert_list[0], GNUTLS_X509_FMT_DER);
-               if (ret < 0)
-               {
-                       certinfo->error = gnutls_strerror(ret);
-                       goto info_done_dealloc;
-               }
+                       reference<GnuTLS::Profile> profile;
+                       try
+                       {
+                               profile = GnuTLS::Profile::Create(name, tag);
+                       }
+                       catch (CoreException& ex)
+                       {
+                               throw ModuleException("Error while initializing SSL profile \"" + name + "\" at " + tag->getTagLocation() + " - " + ex.GetReason());
+                       }
 
-               if (gnutls_x509_crt_get_dn(cert, name, &name_size) == 0)
-               {
-                       std::string& dn = certinfo->dn;
-                       dn = name;
-                       // Make sure there are no chars in the string that we consider invalid
-                       if (dn.find_first_of("\r\n") != std::string::npos)
-                               dn.clear();
+                       newprofiles.push_back(new GnuTLSIOHookProvider(this, profile));
                }
 
-               name_size = sizeof(name);
-               if (gnutls_x509_crt_get_issuer_dn(cert, name, &name_size) == 0)
-               {
-                       std::string& issuer = certinfo->issuer;
-                       issuer = name;
-                       if (issuer.find_first_of("\r\n") != std::string::npos)
-                               issuer.clear();
-               }
+               // New profiles are ok, begin using them
+               // Old profiles are deleted when their refcount drops to zero
+               profiles.swap(newprofiles);
+       }
+
+ public:
+       ModuleSSLGnuTLS()
+       {
+#ifndef GNUTLS_HAS_RND
+               gcry_control (GCRYCTL_INITIALIZATION_FINISHED, 0);
+#endif
+       }
+
+       void init() CXX11_OVERRIDE
+       {
+               ReadProfiles();
+               ServerInstance->GenRandom = &randhandler;
+       }
 
-               if ((ret = gnutls_x509_crt_get_fingerprint(cert, hash, digest, &digest_size)) < 0)
+       void OnModuleRehash(User* user, const std::string &param) CXX11_OVERRIDE
+       {
+               if(param != "ssl")
+                       return;
+
+               try
                {
-                       certinfo->error = gnutls_strerror(ret);
+                       ReadProfiles();
                }
-               else
+               catch (ModuleException& ex)
                {
-                       certinfo->fingerprint = irc::hex(digest, digest_size);
+                       ServerInstance->Logs->Log(MODNAME, LOG_DEFAULT, ex.GetReason() + " Not applying settings.");
                }
+       }
 
-               /* Beware here we do not check for errors.
-                */
-               if ((gnutls_x509_crt_get_expiration_time(cert) < ServerInstance->Time()) || (gnutls_x509_crt_get_activation_time(cert) > ServerInstance->Time()))
+       ~ModuleSSLGnuTLS()
+       {
+               ServerInstance->GenRandom = &ServerInstance->HandleGenRandom;
+       }
+
+       void OnCleanup(int target_type, void* item) CXX11_OVERRIDE
+       {
+               if(target_type == TYPE_USER)
                {
-                       certinfo->error = "Not activated, or expired certificate";
+                       LocalUser* user = IS_LOCAL(static_cast<User*>(item));
+
+                       if (user && user->eh.GetIOHook() && user->eh.GetIOHook()->prov->creator == this)
+                       {
+                               // User is using SSL, they're a local user, and they're using one of *our* SSL ports.
+                               // Potentially there could be multiple SSL modules loaded at once on different ports.
+                               ServerInstance->Users->QuitUser(user, "SSL module unloading");
+                       }
                }
+       }
 
-info_done_dealloc:
-               gnutls_x509_crt_deinit(cert);
+       Version GetVersion() CXX11_OVERRIDE
+       {
+               return Version("Provides SSL support for clients", VF_VENDOR);
        }
 
-       void OnEvent(Event& ev)
+       void OnUserConnect(LocalUser* user) CXX11_OVERRIDE
        {
-               if (starttls.enabled)
-                       capHandler.HandleEvent(ev);
+               IOHook* hook = user->eh.GetIOHook();
+               if (hook && hook->prov->creator == this)
+                       static_cast<GnuTLSIOHook*>(hook)->TellCiphersAndFingerprint(user);
        }
 };