]> git.netwichtig.de Git - user/henk/code/inspircd.git/blobdiff - src/modules/extra/m_ssl_gnutls.cpp
Change Windows libraries to be dynamically linked
[user/henk/code/inspircd.git] / src / modules / extra / m_ssl_gnutls.cpp
index 7c5f7ee493d3246392d1e682ea39cf2cc7b42390..03673d7a06a4363aad55ca4ea5beb85644d20088 100644 (file)
-/*       +------------------------------------+
- *       | Inspire Internet Relay Chat Daemon |
- *       +------------------------------------+
+/*
+ * InspIRCd -- Internet Relay Chat Daemon
  *
- *  InspIRCd: (C) 2002-2009 InspIRCd Development Team
- * See: http://wiki.inspircd.org/Credits
+ *   Copyright (C) 2009-2010 Daniel De Graaf <danieldg@inspircd.org>
+ *   Copyright (C) 2008 John Brooks <john.brooks@dereferenced.net>
+ *   Copyright (C) 2006-2008 Craig Edwards <craigedwards@brainbox.cc>
+ *   Copyright (C) 2007 Dennis Friis <peavey@inspircd.org>
+ *   Copyright (C) 2006 Oliver Lupton <oliverlupton@gmail.com>
  *
- * This program is free but copyrighted software; see
- *         the file COPYING for details.
+ * This file is part of InspIRCd.  InspIRCd is free software: you can
+ * redistribute it and/or modify it under the terms of the GNU General Public
+ * License as published by the Free Software Foundation, version 2.
  *
- * ---------------------------------------------------
+ * This program is distributed in the hope that it will be useful, but WITHOUT
+ * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
+ * FOR A PARTICULAR PURPOSE.  See the GNU General Public License for more
+ * details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program.  If not, see <http://www.gnu.org/licenses/>.
  */
 
+
 #include "inspircd.h"
+#ifndef _WIN32
+#include <gcrypt.h>
+#endif
 #include <gnutls/gnutls.h>
 #include <gnutls/x509.h>
-#include "transport.h"
+#include "ssl.h"
 #include "m_cap.h"
 
-#ifdef WINDOWS
-#pragma comment(lib, "libgnutls-13.lib")
+#ifdef _WIN32
+# pragma comment(lib, "libgnutls-28.lib")
 #endif
 
 /* $ModDesc: Provides SSL support for clients */
-/* $CompileFlags: pkgconfincludes("gnutls","/gnutls/gnutls.h","") */
-/* $LinkerFlags: rpath("pkg-config --libs gnutls") pkgconflibs("gnutls","/libgnutls.so","-lgnutls") */
-/* $ModDep: transport.h */
-/* $CopyInstall: conf/key.pem $(CONPATH) */
-/* $CopyInstall: conf/cert.pem $(CONPATH) */
+/* $CompileFlags: pkgconfincludes("gnutls","/gnutls/gnutls.h","") exec("libgcrypt-config --cflags") */
+/* $LinkerFlags: rpath("pkg-config --libs gnutls") pkgconflibs("gnutls","/libgnutls.so","-lgnutls") exec("libgcrypt-config --libs") */
+/* $NoPedantic */
+
+#ifndef GNUTLS_VERSION_MAJOR
+#define GNUTLS_VERSION_MAJOR LIBGNUTLS_VERSION_MAJOR
+#define GNUTLS_VERSION_MINOR LIBGNUTLS_VERSION_MINOR
+#define GNUTLS_VERSION_PATCH LIBGNUTLS_VERSION_PATCH
+#endif
+
+// These don't exist in older GnuTLS versions
+#if ((GNUTLS_VERSION_MAJOR > 2) || (GNUTLS_VERSION_MAJOR == 2 && GNUTLS_VERSION_MINOR > 1) || (GNUTLS_VERSION_MAJOR == 2 && GNUTLS_VERSION_MINOR == 1 && GNUTLS_VERSION_PATCH >= 7))
+#define GNUTLS_NEW_PRIO_API
+#endif
+
+#if(GNUTLS_VERSION_MAJOR < 2)
+typedef gnutls_certificate_credentials_t gnutls_certificate_credentials;
+typedef gnutls_dh_params_t gnutls_dh_params;
+#endif
+
+#if (defined(_WIN32) && (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 };
 
-/** Represents an SSL user's extra data
- */
-class issl_session : public classbase
+static std::vector<gnutls_x509_crt_t> x509_certs;
+static gnutls_x509_privkey_t x509_key;
+#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;
+#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;
+#endif
+       st->ncerts = x509_certs.size();
+       st->cert.x509 = &x509_certs[0];
+       st->key.x509 = x509_key;
+       st->deinit_all = 0;
+
+       return 0;
+}
+
+class RandGen : public HandlerBase2<void, char*, size_t>
 {
-public:
-       issl_session()
+ public:
+       RandGen() {}
+       void Call(char* buffer, size_t len)
        {
-               sess = NULL;
+#ifdef GNUTLS_HAS_RND
+               gnutls_rnd(GNUTLS_RND_RANDOM, buffer, len);
+#else
+               gcry_randomize(buffer, len, GCRY_STRONG_RANDOM);
+#endif
        }
+};
 
+/** Represents an SSL user's extra data
+ */
+class issl_session
+{
+public:
+       StreamSocket* socket;
        gnutls_session_t sess;
        issl_status status;
-       std::string outbuf;
+       reference<ssl_cert> cert;
+
+       issl_session() : socket(NULL), sess(NULL) {}
 };
 
-class CommandStartTLS : public Command
+class CommandStartTLS : public SplitCommand
 {
-       Module* Caller;
  public:
-       CommandStartTLS (InspIRCd* Instance, Module* mod) : Command(Instance,"STARTTLS", 0, 0, true), Caller(mod)
+       bool enabled;
+       CommandStartTLS (Module* mod) : SplitCommand(mod, "STARTTLS")
        {
-               this->source = "m_ssl_gnutls.so";
+               enabled = true;
+               works_before_reg = true;
        }
 
-       CmdResult Handle (const std::vector<std::string> &parameters, User *user)
+       CmdResult HandleLocal(const std::vector<std::string> &parameters, LocalUser *user)
        {
-               /* changed from == REG_ALL to catch clients sending STARTTLS
-                * after NICK and USER but before OnUserConnect completes and
-                * give a proper error message (see bug #645) - dz
-                */
-               if (user->registered != REG_NONE)
+               if (!enabled)
                {
-                       user->WriteNumeric(691, "%s :STARTTLS is not permitted after client registration has started", user->nick.c_str());
+                       user->WriteNumeric(691, "%s :STARTTLS is not enabled", user->nick.c_str());
+                       return CMD_FAILURE;
+               }
+
+               if (user->registered == REG_ALL)
+               {
+                       user->WriteNumeric(691, "%s :STARTTLS is not permitted after client registration is complete", user->nick.c_str());
                }
                else
                {
-                       if (!user->GetIOHook())
+                       if (!user->eh.GetIOHook())
                        {
                                user->WriteNumeric(670, "%s :STARTTLS successful, go ahead with TLS handshake", user->nick.c_str());
-                               user->AddIOHook(Caller);
-                               Caller->OnRawSocketAccept(user->GetFd(), NULL, NULL);
+                               /* 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);
                        }
                        else
                                user->WriteNumeric(691, "%s :STARTTLS failure", user->nick.c_str());
@@ -82,155 +161,332 @@ class CommandStartTLS : public Command
 
 class ModuleSSLGnuTLS : public Module
 {
-       std::set<ListenSocketBase*> listenports;
-
        issl_session* sessions;
 
-       gnutls_certificate_credentials x509_cred;
-       gnutls_dh_params dh_params;
+       gnutls_certificate_credentials_t x509_cred;
+       gnutls_dh_params_t dh_params;
+       gnutls_digest_algorithm_t hash;
+       #ifdef GNUTLS_NEW_PRIO_API
+       gnutls_priority_t priority;
+       #endif
 
-       std::string keyfile;
-       std::string certfile;
-       std::string cafile;
-       std::string crlfile;
        std::string sslports;
        int dh_bits;
 
        bool cred_alloc;
+       bool dh_alloc;
 
+       RandGen randhandler;
        CommandStartTLS starttls;
 
+       GenericCap capHandler;
+       ServiceProvider iohook;
+
+       inline static const char* UnknownIfNULL(const char* str)
+       {
+               return str ? str : "UNKNOWN";
+       }
+
+       static ssize_t gnutls_pull_wrapper(gnutls_transport_ptr_t session_wrap, void* buffer, size_t size)
+       {
+               issl_session* session = reinterpret_cast<issl_session*>(session_wrap);
+               if (session->socket->GetEventMask() & FD_READ_WILL_BLOCK)
+               {
+#ifdef _WIN32
+                       gnutls_transport_set_errno(session->sess, EAGAIN);
+#else
+                       errno = EAGAIN;
+#endif
+                       return -1;
+               }
+
+               int rv = ServerInstance->SE->Recv(session->socket, reinterpret_cast<char *>(buffer), size, 0);
+
+#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
+
+               if (rv < (int)size)
+                       ServerInstance->SE->ChangeEventMask(session->socket, FD_READ_WILL_BLOCK);
+               return rv;
+       }
+
+       static ssize_t gnutls_push_wrapper(gnutls_transport_ptr_t session_wrap, const void* buffer, size_t size)
+       {
+               issl_session* session = reinterpret_cast<issl_session*>(session_wrap);
+               if (session->socket->GetEventMask() & FD_WRITE_WILL_BLOCK)
+               {
+#ifdef _WIN32
+                       gnutls_transport_set_errno(session->sess, EAGAIN);
+#else
+                       errno = EAGAIN;
+#endif
+                       return -1;
+               }
+
+               int rv = ServerInstance->SE->Send(session->socket, reinterpret_cast<const char *>(buffer), size, 0);
+
+#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
+
+               if (rv < (int)size)
+                       ServerInstance->SE->ChangeEventMask(session->socket, FD_WRITE_WILL_BLOCK);
+               return rv;
+       }
+
  public:
 
-       ModuleSSLGnuTLS(InspIRCd* Me)
-               : Module(Me), starttls(Me, this)
+       ModuleSSLGnuTLS()
+               : starttls(this), capHandler(this, "tls"), iohook(this, "ssl/gnutls", SERVICE_IOHOOK)
        {
-               ServerInstance->Modules->PublishInterface("BufferedSocketHook", this);
+#ifndef GNUTLS_HAS_RND
+               gcry_control (GCRYCTL_INITIALIZATION_FINISHED, 0);
+#endif
 
                sessions = new issl_session[ServerInstance->SE->GetMaxFds()];
 
                gnutls_global_init(); // This must be called once in the program
+               gnutls_x509_privkey_init(&x509_key);
+
+               #ifdef GNUTLS_NEW_PRIO_API
+               // Init this here so it's always initialized, avoids an extra boolean
+               gnutls_priority_init(&priority, "NORMAL", NULL);
+               #endif
 
                cred_alloc = false;
+               dh_alloc = false;
+       }
+
+       void init()
+       {
                // Needs the flag as it ignores a plain /rehash
                OnModuleRehash(NULL,"ssl");
 
+               ServerInstance->GenRandom = &randhandler;
+
                // Void return, guess we assume success
                gnutls_certificate_set_dh_params(x509_cred, dh_params);
-               Implementation eventlist[] = { I_On005Numeric, I_OnRawSocketConnect, I_OnRawSocketAccept,
-                       I_OnRawSocketClose, I_OnRawSocketRead, I_OnRawSocketWrite, I_OnCleanup,
-                       I_OnBufferFlushed, I_OnRequest, I_OnRehash, I_OnModuleRehash, I_OnPostConnect,
+               Implementation eventlist[] = { I_On005Numeric, I_OnRehash, I_OnModuleRehash, I_OnUserConnect,
                        I_OnEvent, I_OnHookIO };
                ServerInstance->Modules->Attach(eventlist, this, sizeof(eventlist)/sizeof(Implementation));
 
-               ServerInstance->AddCommand(&starttls);
+               ServerInstance->Modules->AddService(iohook);
+               ServerInstance->Modules->AddService(starttls);
        }
 
-       virtual void OnRehash(User* user)
+       void OnRehash(User* user)
        {
-               ConfigReader Conf(ServerInstance);
-
-               listenports.clear();
                sslports.clear();
 
-               for (size_t i = 0; i < ServerInstance->ports.size(); i++)
+               ConfigTag* Conf = ServerInstance->Config->ConfValue("gnutls");
+               starttls.enabled = Conf->getBool("starttls", true);
+
+               if (Conf->getBool("showports", true))
                {
-                       ListenSocketBase* port = ServerInstance->ports[i];
-                       std::string desc = port->GetDescription();
-                       if (desc != "gnutls")
-                               continue;
+                       sslports = Conf->getString("advertisedports");
+                       if (!sslports.empty())
+                               return;
 
-                       listenports.insert(port);
-                       std::string portid = port->GetBindDesc();
+                       for (size_t i = 0; i < ServerInstance->ports.size(); i++)
+                       {
+                               ListenSocket* port = ServerInstance->ports[i];
+                               if (port->bind_tag->getString("ssl") != "gnutls")
+                                       continue;
 
-                       ServerInstance->Logs->Log("m_ssl_gnutls", DEFAULT, "m_ssl_gnutls.so: Enabling SSL for port %s", portid.c_str());
-                       if (port->GetIP() != "127.0.0.1")
-                               sslports.append(portid).append(";");
-               }
+                               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());
 
-               if (!sslports.empty())
-                       sslports.erase(sslports.end() - 1);
+                               if (port->bind_tag->getString("type", "clients") == "clients" && port->bind_addr != "127.0.0.1")
+                               {
+                                       /*
+                                        * 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;
+                               }
+                       }
+               }
        }
 
-       virtual void OnModuleRehash(User* user, const std::string &param)
+       void OnModuleRehash(User* user, const std::string &param)
        {
                if(param != "ssl")
                        return;
 
+               std::string keyfile;
+               std::string certfile;
+               std::string cafile;
+               std::string crlfile;
                OnRehash(user);
 
-               ConfigReader Conf(ServerInstance);
-
-               std::string confdir(ServerInstance->ConfigFileName);
-               // +1 so we the path ends with a /
-               confdir = confdir.substr(0, confdir.find_last_of('/') + 1);
-
-               cafile = Conf.ReadValue("gnutls", "cafile", 0);
-               crlfile = Conf.ReadValue("gnutls", "crlfile", 0);
-               certfile = Conf.ReadValue("gnutls", "certfile", 0);
-               keyfile = Conf.ReadValue("gnutls", "keyfile", 0);
-               dh_bits = Conf.ReadInteger("gnutls", "dhbits", 0, false);
+               ConfigTag* Conf = ServerInstance->Config->ConfValue("gnutls");
 
-               // Set all the default values needed.
-               if (cafile.empty())
-                       cafile = "ca.pem";
+               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");
 
-               if (crlfile.empty())
-                       crlfile = "crl.pem";
-
-               if (certfile.empty())
-                       certfile = "cert.pem";
-
-               if (keyfile.empty())
-                       keyfile = "key.pem";
+               // 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");
 
                if((dh_bits != 768) && (dh_bits != 1024) && (dh_bits != 2048) && (dh_bits != 3072) && (dh_bits != 4096))
                        dh_bits = 1024;
 
-               // Prepend relative paths with the path to the config directory.
-               if ((cafile[0] != '/') && (!ServerInstance->Config->StartsWithWindowsDriveLetter(cafile)))
-                       cafile = confdir + cafile;
-
-               if ((crlfile[0] != '/') && (!ServerInstance->Config->StartsWithWindowsDriveLetter(crlfile)))
-                       crlfile = confdir + crlfile;
-
-               if ((certfile[0] != '/') && (!ServerInstance->Config->StartsWithWindowsDriveLetter(certfile)))
-                       certfile = confdir + certfile;
+               if (hashname == "md5")
+                       hash = GNUTLS_DIG_MD5;
+               else if (hashname == "sha1")
+                       hash = GNUTLS_DIG_SHA1;
+               else
+                       throw ModuleException("Unknown hash type " + hashname);
 
-               if ((keyfile[0] != '/') && (!ServerInstance->Config->StartsWithWindowsDriveLetter(keyfile)))
-                       keyfile = confdir + keyfile;
 
                int ret;
 
+               if (dh_alloc)
+               {
+                       gnutls_dh_params_deinit(dh_params);
+                       dh_alloc = false;
+                       dh_params = NULL;
+               }
+
                if (cred_alloc)
                {
                        // Deallocate the old credentials
-                       gnutls_dh_params_deinit(dh_params);
                        gnutls_certificate_free_credentials(x509_cred);
+
+                       for(unsigned int i=0; i < x509_certs.size(); i++)
+                               gnutls_x509_crt_deinit(x509_certs[i]);
+                       x509_certs.clear();
                }
-               else
-                       cred_alloc = true;
 
-               if((ret = gnutls_certificate_allocate_credentials(&x509_cred)) < 0)
+               ret = gnutls_certificate_allocate_credentials(&x509_cred);
+               cred_alloc = (ret >= 0);
+               if (!cred_alloc)
                        ServerInstance->Logs->Log("m_ssl_gnutls",DEBUG, "m_ssl_gnutls.so: Failed to allocate certificate credentials: %s", gnutls_strerror(ret));
 
-               if((ret = gnutls_dh_params_init(&dh_params)) < 0)
-                       ServerInstance->Logs->Log("m_ssl_gnutls",DEBUG, "m_ssl_gnutls.so: Failed to initialise DH parameters: %s", gnutls_strerror(ret));
-
                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((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));
 
-               if((ret = gnutls_certificate_set_x509_key_file (x509_cred, certfile.c_str(), keyfile.c_str(), GNUTLS_X509_FMT_PEM)) < 0)
+               FileReader reader;
+
+               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()) };
+
+               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()) };
+
+               // 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)
+               {
+                       // 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);
+               }
+
+               if (ret <= 0)
+               {
+                       // 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"));
+               }
+               x509_certs.resize(ret);
+
+               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
+               // It's safe to call this every time as we cannot have this uninitialized, see constructor and below.
+               gnutls_priority_deinit(priority);
+
+               // 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;
+
+               if ((ret = gnutls_priority_init(&priority, priocstr, &prioerror)) < 0)
                {
-                       // If this fails, no SSL port will work. At all. So, do the smart thing - throw a ModuleException
-                       throw ModuleException("Unable to load GnuTLS server certificate (" + certfile + ", key: " + keyfile + "): " + std::string(gnutls_strerror(ret)));
+                       // 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);
                }
 
-               // This may be on a large (once a day or week) timer eventually.
-               GenerateDHParams();
+               #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
+               ret = gnutls_dh_params_init(&dh_params);
+               dh_alloc = (ret >= 0);
+               if (!dh_alloc)
+               {
+                       ServerInstance->Logs->Log("m_ssl_gnutls",DEFAULT, "m_ssl_gnutls.so: Failed to initialise DH parameters: %s", gnutls_strerror(ret));
+                       return;
+               }
+
+               std::string dhfile = Conf->getString("dhfile");
+               if (!dhfile.empty())
+               {
+                       // 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()) };
+
+                       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();
+                       }
+               }
+               else
+               {
+                       GenerateDHParams();
+               }
        }
 
        void GenerateDHParams()
@@ -240,296 +496,205 @@ class ModuleSSLGnuTLS : public Module
                // once a day, once a week or once a month. Depending on the
                // security requirements.
 
+               if (!dh_alloc)
+                       return;
+
                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));
        }
 
-       virtual ~ModuleSSLGnuTLS()
+       ~ModuleSSLGnuTLS()
        {
-               gnutls_dh_params_deinit(dh_params);
-               gnutls_certificate_free_credentials(x509_cred);
+               for(unsigned int i=0; i < x509_certs.size(); i++)
+                       gnutls_x509_crt_deinit(x509_certs[i]);
+
+               gnutls_x509_privkey_deinit(x509_key);
+               #ifdef GNUTLS_NEW_PRIO_API
+               gnutls_priority_deinit(priority);
+               #endif
+
+               if (dh_alloc)
+                       gnutls_dh_params_deinit(dh_params);
+               if (cred_alloc)
+                       gnutls_certificate_free_credentials(x509_cred);
+
                gnutls_global_deinit();
-               ServerInstance->Modules->UnpublishInterface("BufferedSocketHook", this);
                delete[] sessions;
+               ServerInstance->GenRandom = &ServerInstance->HandleGenRandom;
        }
 
-       virtual void OnCleanup(int target_type, void* item)
+       void OnCleanup(int target_type, void* item)
        {
                if(target_type == TYPE_USER)
                {
-                       User* user = static_cast<User*>(item);
+                       LocalUser* user = IS_LOCAL(static_cast<User*>(item));
 
-                       if (user->GetIOHook() == this)
+                       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");
-                               user->DelIOHook();
-                       }
-                       if (user->GetExt("ssl_cert"))
-                       {
-                               ssl_cert* tofree;
-                               user->GetExt("ssl_cert", tofree);
-                               delete tofree;
-                               user->Shrink("ssl_cert");
                        }
                }
        }
 
-       virtual Version GetVersion()
+       Version GetVersion()
        {
-               return Version("$Id$", VF_VENDOR, API_VERSION);
+               return Version("Provides SSL support for clients", VF_VENDOR);
        }
 
 
-       virtual void On005Numeric(std::string &output)
+       void On005Numeric(std::string &output)
        {
                if (!sslports.empty())
                        output.append(" SSL=" + sslports);
-               output.append(" STARTTLS");
+               if (starttls.enabled)
+                       output.append(" STARTTLS");
        }
 
-       virtual void OnHookIO(EventHandler* user, ListenSocketBase* lsb)
+       void OnHookIO(StreamSocket* user, ListenSocket* lsb)
        {
-               if (!user->GetIOHook() && listenports.find(lsb) != listenports.end())
+               if (!user->GetIOHook() && lsb->bind_tag->getString("ssl") == "gnutls")
                {
                        /* Hook the user with our module */
                        user->AddIOHook(this);
                }
        }
 
-       virtual const char* OnRequest(Request* request)
+       void OnRequest(Request& request)
        {
-               ISHRequest* ISR = static_cast<ISHRequest*>(request);
-               if (strcmp("IS_NAME", request->GetId()) == 0)
-               {
-                       return "gnutls";
-               }
-               else if (strcmp("IS_HOOK", request->GetId()) == 0)
-               {
-                       const char* ret = "OK";
-                       try
-                       {
-                               ret = ISR->Sock->AddIOHook(this) ? "OK" : NULL;
-                       }
-                       catch (ModuleException &e)
-                       {
-                               return NULL;
-                       }
-                       return ret;
-               }
-               else if (strcmp("IS_UNHOOK", request->GetId()) == 0)
-               {
-                       return ISR->Sock->DelIOHook() ? "OK" : NULL;
-               }
-               else if (strcmp("IS_HSDONE", request->GetId()) == 0)
+               if (strcmp("GET_SSL_CERT", request.id) == 0)
                {
-                       if (ISR->Sock->GetFd() < 0)
-                               return "OK";
+                       SocketCertificateRequest& req = static_cast<SocketCertificateRequest&>(request);
+                       int fd = req.sock->GetFd();
+                       issl_session* session = &sessions[fd];
 
-                       issl_session* session = &sessions[ISR->Sock->GetFd()];
-                       return (session->status == ISSL_HANDSHAKING_READ || session->status == ISSL_HANDSHAKING_WRITE) ? NULL : "OK";
+                       req.cert = session->cert;
                }
-               else if (strcmp("IS_ATTACH", request->GetId()) == 0)
-               {
-                       if (ISR->Sock->GetFd() > -1)
-                       {
-                               issl_session* session = &sessions[ISR->Sock->GetFd()];
-                               if (session->sess)
-                               {
-                                       if (static_cast<Extensible*>(ServerInstance->SE->GetRef(ISR->Sock->GetFd())) == static_cast<Extensible*>(ISR->Sock))
-                                       {
-                                               VerifyCertificate(session, ISR->Sock);
-                                               return "OK";
-                                       }
-                               }
-                       }
-               }
-               else if (strcmp("GET_FP", request->GetId()) == 0)
-               {
-                       if (ISR->Sock->GetFd() > -1)
-                       {
-                               issl_session* session = &sessions[ISR->Sock->GetFd()];
-                               if (session->sess)
-                               {
-                                       Extensible* ext = ISR->Sock;
-                                       ssl_cert* certinfo;
-                                       if (ext->GetExt("ssl_cert",certinfo))
-                                               return certinfo->GetFingerprint().c_str();
-                               }
-                       }
-               }
-               return NULL;
        }
 
-
-       virtual void OnRawSocketAccept(int fd, irc::sockets::sockaddrs* client, irc::sockets::sockaddrs* server)
+       void InitSession(StreamSocket* user, bool me_server)
        {
-               /* Are there any possibilities of an out of range fd? Hope not, but lets be paranoid */
-               if ((fd < 0) || (fd > ServerInstance->SE->GetMaxFds() - 1))
-                       return;
-
-               issl_session* session = &sessions[fd];
-
-               /* For STARTTLS: Don't try and init a session on a socket that already has a session */
-               if (session->sess)
-                       return;
+               issl_session* session = &sessions[user->GetFd()];
 
-               gnutls_init(&session->sess, GNUTLS_SERVER);
+               gnutls_init(&session->sess, me_server ? GNUTLS_SERVER : GNUTLS_CLIENT);
+               session->socket = user;
 
-               gnutls_set_default_priority(session->sess); // Avoid calling all the priority functions, defaults are adequate.
+               #ifdef GNUTLS_NEW_PRIO_API
+               gnutls_priority_set(session->sess, priority);
+               #else
+               gnutls_set_default_priority(session->sess);
+               #endif
                gnutls_credentials_set(session->sess, GNUTLS_CRD_CERTIFICATE, 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);
 
-               gnutls_transport_set_ptr(session->sess, reinterpret_cast<gnutls_transport_ptr_t>(fd)); // Give gnutls the fd for the socket.
+               if (me_server)
+                       gnutls_certificate_server_set_request(session->sess, GNUTLS_CERT_REQUEST); // Request client certificate if any.
 
-               gnutls_certificate_server_set_request(session->sess, GNUTLS_CERT_REQUEST); // Request client certificate if any.
-
-               Handshake(session, fd);
+               Handshake(session, user);
        }
 
-       virtual void OnRawSocketConnect(int fd)
+       void OnStreamSocketAccept(StreamSocket* user, irc::sockets::sockaddrs* client, irc::sockets::sockaddrs* server)
        {
-               /* Are there any possibilities of an out of range fd? Hope not, but lets be paranoid */
-               if ((fd < 0) || (fd > ServerInstance->SE->GetMaxFds() - 1))
-                       return;
-
-               issl_session* session = &sessions[fd];
-
-               gnutls_init(&session->sess, GNUTLS_CLIENT);
+               issl_session* session = &sessions[user->GetFd()];
 
-               gnutls_set_default_priority(session->sess); // Avoid calling all the priority functions, defaults are adequate.
-               gnutls_credentials_set(session->sess, GNUTLS_CRD_CERTIFICATE, x509_cred);
-               gnutls_dh_set_prime_bits(session->sess, dh_bits);
-               gnutls_transport_set_ptr(session->sess, reinterpret_cast<gnutls_transport_ptr_t>(fd)); // Give gnutls the fd for the socket.
+               /* For STARTTLS: Don't try and init a session on a socket that already has a session */
+               if (session->sess)
+                       return;
 
-               Handshake(session, fd);
+               InitSession(user, true);
        }
 
-       virtual void OnRawSocketClose(int fd)
+       void OnStreamSocketConnect(StreamSocket* user)
        {
-               /* Are there any possibilities of an out of range fd? Hope not, but lets be paranoid */
-               if ((fd < 0) || (fd > ServerInstance->SE->GetMaxFds()))
-                       return;
-
-               CloseSession(&sessions[fd]);
-
-               EventHandler* user = ServerInstance->SE->GetRef(fd);
-
-               if ((user) && (user->GetExt("ssl_cert")))
-               {
-                       ssl_cert* tofree;
-                       user->GetExt("ssl_cert", tofree);
-                       delete tofree;
-                       user->Shrink("ssl_cert");
-               }
+               InitSession(user, false);
        }
 
-       virtual int OnRawSocketRead(int fd, char* buffer, unsigned int count, int &readresult)
+       void OnStreamSocketClose(StreamSocket* user)
        {
-               /* Are there any possibilities of an out of range fd? Hope not, but lets be paranoid */
-               if ((fd < 0) || (fd > ServerInstance->SE->GetMaxFds() - 1))
-                       return 0;
+               CloseSession(&sessions[user->GetFd()]);
+       }
 
-               issl_session* session = &sessions[fd];
+       int OnStreamSocketRead(StreamSocket* user, std::string& recvq)
+       {
+               issl_session* session = &sessions[user->GetFd()];
 
                if (!session->sess)
                {
-                       readresult = 0;
                        CloseSession(session);
-                       return 1;
+                       user->SetError("No SSL session");
+                       return -1;
                }
 
-               if (session->status == ISSL_HANDSHAKING_READ)
+               if (session->status == ISSL_HANDSHAKING_READ || session->status == ISSL_HANDSHAKING_WRITE)
                {
                        // The handshake isn't finished, try to finish it.
 
-                       if(!Handshake(session, fd))
+                       if(!Handshake(session, user))
                        {
-                               errno = session->status == ISSL_CLOSING ? EIO : EAGAIN;
-                               // Couldn't resume handshake.
+                               if (session->status != ISSL_CLOSING)
+                                       return 0;
                                return -1;
                        }
                }
-               else if (session->status == ISSL_HANDSHAKING_WRITE)
-               {
-                       errno = EAGAIN;
-                       MakePollWrite(fd);
-                       return -1;
-               }
 
                // If we resumed the handshake then session->status will be ISSL_HANDSHAKEN.
 
                if (session->status == ISSL_HANDSHAKEN)
                {
-                       unsigned int len = 0;
-                       while (len < count)
+                       char* buffer = ServerInstance->GetReadBuffer();
+                       size_t bufsiz = ServerInstance->Config->NetBufferSize;
+                       int ret = gnutls_record_recv(session->sess, buffer, bufsiz);
+                       if (ret > 0)
                        {
-                               int ret = gnutls_record_recv(session->sess, buffer + len, count - len);
-                               if (ret > 0)
-                               {
-                                       len += ret;
-                               }
-                               else if (ret == GNUTLS_E_AGAIN || ret == GNUTLS_E_INTERRUPTED)
-                               {
-                                       break;
-                               }
-                               else
-                               {
-                                       if (ret != 0)
-                                               ServerInstance->Logs->Log("m_ssl_gnutls", DEFAULT,
-                                                       "m_ssl_gnutls.so: Error while reading on fd %d: %s",
-                                                       fd, gnutls_strerror(ret));
-
-                                       // if ret == 0, client closed connection.
-                                       readresult = 0;
-                                       CloseSession(session);
-                                       return 1;
-                               }
+                               recvq.append(buffer, ret);
+                               return 1;
                        }
-                       readresult = len;
-                       if (len)
+                       else if (ret == GNUTLS_E_AGAIN || ret == GNUTLS_E_INTERRUPTED)
                        {
-                               return 1;
+                               return 0;
+                       }
+                       else if (ret == 0)
+                       {
+                               user->SetError("Connection closed");
+                               CloseSession(session);
+                               return -1;
                        }
                        else
                        {
-                               errno = EAGAIN;
+                               user->SetError(gnutls_strerror(ret));
+                               CloseSession(session);
                                return -1;
                        }
                }
                else if (session->status == ISSL_CLOSING)
-                       readresult = 0;
+                       return -1;
 
-               return 1;
+               return 0;
        }
 
-       virtual int OnRawSocketWrite(int fd, const char* buffer, int count)
+       int OnStreamSocketWrite(StreamSocket* user, std::string& sendq)
        {
-               /* Are there any possibilities of an out of range fd? Hope not, but lets be paranoid */
-               if ((fd < 0) || (fd > ServerInstance->SE->GetMaxFds() - 1))
-                       return 0;
-
-               issl_session* session = &sessions[fd];
-               const char* sendbuffer = buffer;
+               issl_session* session = &sessions[user->GetFd()];
 
                if (!session->sess)
                {
                        CloseSession(session);
-                       return 1;
+                       user->SetError("No SSL session");
+                       return -1;
                }
 
-               session->outbuf.append(sendbuffer, count);
-               sendbuffer = session->outbuf.c_str();
-               count = session->outbuf.size();
-
                if (session->status == ISSL_HANDSHAKING_WRITE || session->status == ISSL_HANDSHAKING_READ)
                {
                        // The handshake isn't finished, try to finish it.
-                       Handshake(session, fd);
-                       errno = session->status == ISSL_CLOSING ? EIO : EAGAIN;
+                       Handshake(session, user);
+                       if (session->status != ISSL_CLOSING)
+                               return 0;
                        return -1;
                }
 
@@ -537,42 +702,36 @@ class ModuleSSLGnuTLS : public Module
 
                if (session->status == ISSL_HANDSHAKEN)
                {
-                       ret = gnutls_record_send(session->sess, sendbuffer, count);
+                       ret = gnutls_record_send(session->sess, sendq.data(), sendq.length());
 
-                       if (ret == 0)
+                       if (ret == (int)sendq.length())
                        {
-                               CloseSession(session);
+                               ServerInstance->SE->ChangeEventMask(user, FD_WANT_NO_WRITE);
+                               return 1;
                        }
-                       else if (ret < 0)
+                       else if (ret > 0)
                        {
-                               if(ret != GNUTLS_E_AGAIN && ret != GNUTLS_E_INTERRUPTED)
-                               {
-                                       ServerInstance->Logs->Log("m_ssl_gnutls", DEFAULT,
-                                                       "m_ssl_gnutls.so: Error while writing to fd %d: %s",
-                                                       fd, gnutls_strerror(ret));
-                                       CloseSession(session);
-                               }
-                               else
-                               {
-                                       errno = EAGAIN;
-                               }
+                               sendq = sendq.substr(ret);
+                               ServerInstance->SE->ChangeEventMask(user, FD_WANT_SINGLE_WRITE);
+                               return 0;
                        }
-                       else
+                       else if (ret == GNUTLS_E_AGAIN || ret == GNUTLS_E_INTERRUPTED || ret == 0)
                        {
-                               session->outbuf = session->outbuf.substr(ret);
+                               ServerInstance->SE->ChangeEventMask(user, FD_WANT_SINGLE_WRITE);
+                               return 0;
+                       }
+                       else // (ret < 0)
+                       {
+                               user->SetError(gnutls_strerror(ret));
+                               CloseSession(session);
+                               return -1;
                        }
                }
 
-               if (!session->outbuf.empty())
-                       MakePollWrite(fd);
-
-               /* Who's smart idea was it to return 1 when we havent written anything?
-                * This fucks the buffer up in BufferedSocket :p
-                */
-               return ret < 1 ? 0 : ret;
+               return 0;
        }
 
-       bool Handshake(issl_session* session, int fd)
+       bool Handshake(issl_session* session, StreamSocket* user)
        {
                int ret = gnutls_handshake(session->sess);
 
@@ -586,20 +745,18 @@ class ModuleSSLGnuTLS : public Module
                                {
                                        // 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;
-                                       MakePollWrite(fd);
+                                       ServerInstance->SE->ChangeEventMask(user, FD_WANT_NO_READ | FD_WANT_SINGLE_WRITE);
                                }
                        }
                        else
                        {
-                               // Handshake failed.
-                               ServerInstance->Logs->Log("m_ssl_gnutls", DEFAULT,
-                                               "m_ssl_gnutls.so: Handshake failed on fd %d: %s",
-                                               fd, gnutls_strerror(ret));
+                               user->SetError("Handshake Failed - " + std::string(gnutls_strerror(ret)));
                                CloseSession(session);
                                session->status = ISSL_CLOSING;
                        }
@@ -608,80 +765,56 @@ class ModuleSSLGnuTLS : public Module
                }
                else
                {
-                       // Handshake complete.
-                       // This will do for setting the ssl flag...it could be done earlier if it's needed. But this seems neater.
-                       EventHandler *extendme = ServerInstance->SE->GetRef(fd);
-                       if (extendme)
-                       {
-                               extendme->Extend("ssl");
-                       }
-
                        // Change the seesion state
                        session->status = ISSL_HANDSHAKEN;
 
+                       VerifyCertificate(session,user);
+
                        // Finish writing, if any left
-                       MakePollWrite(fd);
+                       ServerInstance->SE->ChangeEventMask(user, FD_WANT_POLL_READ | FD_WANT_NO_WRITE | FD_ADD_TRIAL_WRITE);
 
                        return true;
                }
        }
 
-       virtual void OnPostConnect(User* user)
+       void OnUserConnect(LocalUser* user)
        {
-               // This occurs AFTER OnUserConnect so we can be sure the
-               // protocol module has propagated the NICK message.
-               if (user->GetIOHook() == this && (IS_LOCAL(user)))
+               if (user->eh.GetIOHook() == this)
                {
-                       ssl_cert* certdata = VerifyCertificate(&sessions[user->GetFd()],user);
-                       if (sessions[user->GetFd()].sess)
+                       if (sessions[user->eh.GetFd()].sess)
                        {
-                               std::string cipher = gnutls_kx_get_name(gnutls_kx_get(sessions[user->GetFd()].sess));
-                               cipher.append("-").append(gnutls_cipher_get_name(gnutls_cipher_get(sessions[user->GetFd()].sess))).append("-");
-                               cipher.append(gnutls_mac_get_name(gnutls_mac_get(sessions[user->GetFd()].sess)));
-                               user->WriteServ("NOTICE %s :*** You are connected using SSL cipher \"%s\"", user->nick.c_str(), cipher.c_str());
+                               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->PI->SendMetaData(user, "ssl", "ON");
-                       if (certdata)
-                               ServerInstance->PI->SendMetaData(user, "ssl_cert", certdata->GetMetaLine().c_str());
-               }
-       }
-
-       void MakePollWrite(int fd)
-       {
-               //OnRawSocketWrite(fd, NULL, 0);
-               EventHandler* eh = ServerInstance->SE->GetRef(fd);
-               if (eh)
-                       ServerInstance->SE->WantWrite(eh);
-       }
-
-       virtual void OnBufferFlushed(User* user)
-       {
-               if (user->GetIOHook() == this)
-               {
-                       issl_session* session = &sessions[user->GetFd()];
-                       if (session && session->outbuf.size())
-                               OnRawSocketWrite(user->GetFd(), NULL, 0);
                }
        }
 
        void CloseSession(issl_session* session)
        {
-               if(session->sess)
+               if (session->sess)
                {
                        gnutls_bye(session->sess, GNUTLS_SHUT_WR);
                        gnutls_deinit(session->sess);
                }
-
-               session->outbuf.clear();
+               session->socket = NULL;
                session->sess = NULL;
+               session->cert = NULL;
                session->status = ISSL_NONE;
        }
 
-       ssl_cert* VerifyCertificate(issl_session* session, Extensible* user)
+       void VerifyCertificate(issl_session* session, StreamSocket* user)
        {
                if (!session->sess || !user)
-                       return NULL;
+                       return;
 
                unsigned int status;
                const gnutls_datum_t* cert_list;
@@ -693,8 +826,7 @@ class ModuleSSLGnuTLS : public Module
                size_t digest_size = sizeof(digest);
                size_t name_size = sizeof(name);
                ssl_cert* certinfo = new ssl_cert;
-
-               user->Extend("ssl_cert",certinfo);
+               session->cert = certinfo;
 
                /* This verification function uses the trusted CAs in the credentials
                 * structure. So you must have installed one or more CA certificates.
@@ -704,7 +836,7 @@ class ModuleSSLGnuTLS : public Module
                if (ret < 0)
                {
                        certinfo->error = std::string(gnutls_strerror(ret));
-                       return certinfo;
+                       return;
                }
 
                certinfo->invalid = (status & GNUTLS_CERT_INVALID);
@@ -719,14 +851,14 @@ class ModuleSSLGnuTLS : public Module
                if (gnutls_certificate_type_get(session->sess) != GNUTLS_CRT_X509)
                {
                        certinfo->error = "No X509 keys sent";
-                       return certinfo;
+                       return;
                }
 
                ret = gnutls_x509_crt_init(&cert);
                if (ret < 0)
                {
                        certinfo->error = gnutls_strerror(ret);
-                       return certinfo;
+                       return;
                }
 
                cert_list_size = 0;
@@ -734,7 +866,7 @@ class ModuleSSLGnuTLS : public Module
                if (cert_list == NULL)
                {
                        certinfo->error = "No certificate was found";
-                       return certinfo;
+                       goto info_done_dealloc;
                }
 
                /* This is not a real world example, since we only check the first
@@ -745,7 +877,7 @@ class ModuleSSLGnuTLS : public Module
                if (ret < 0)
                {
                        certinfo->error = gnutls_strerror(ret);
-                       return certinfo;
+                       goto info_done_dealloc;
                }
 
                gnutls_x509_crt_get_dn(cert, name, &name_size);
@@ -754,7 +886,7 @@ class ModuleSSLGnuTLS : public Module
                gnutls_x509_crt_get_issuer_dn(cert, name, &name_size);
                certinfo->issuer = name;
 
-               if ((ret = gnutls_x509_crt_get_fingerprint(cert, GNUTLS_DIG_MD5, digest, &digest_size)) < 0)
+               if ((ret = gnutls_x509_crt_get_fingerprint(cert, hash, digest, &digest_size)) < 0)
                {
                        certinfo->error = gnutls_strerror(ret);
                }
@@ -770,20 +902,14 @@ class ModuleSSLGnuTLS : public Module
                        certinfo->error = "Not activated, or expired certificate";
                }
 
+info_done_dealloc:
                gnutls_x509_crt_deinit(cert);
-
-               return certinfo;
-       }
-
-       void OnEvent(Event* ev)
-       {
-               GenericCapHandler(ev, "tls", "tls");
        }
 
-       void Prioritize()
+       void OnEvent(Event& ev)
        {
-               Module* server = ServerInstance->Modules->Find("m_spanningtree.so");
-               ServerInstance->Modules->SetPriority(this, I_OnPostConnect, PRIORITY_AFTER, &server);
+               if (starttls.enabled)
+                       capHandler.HandleEvent(ev);
        }
 };