]> git.netwichtig.de Git - user/henk/code/inspircd.git/blobdiff - src/modules/extra/m_ssl_gnutls.cpp
Fix extras compilation under Windows
[user/henk/code/inspircd.git] / src / modules / extra / m_ssl_gnutls.cpp
index 8ec787465a80598bf524c3b9c8189dfdedb6e132..95638417bc9aa6cfc0c1c1c0bd8f5d6991737403 100644 (file)
@@ -2,7 +2,7 @@
  *       | Inspire Internet Relay Chat Daemon |
  *       +------------------------------------+
  *
- *  InspIRCd: (C) 2002-2009 InspIRCd Development Team
+ *  InspIRCd: (C) 2002-2010 InspIRCd Development Team
  * See: http://wiki.inspircd.org/Credits
  *
  * This program is free but copyrighted software; see
 #include "inspircd.h"
 #include <gnutls/gnutls.h>
 #include <gnutls/x509.h>
+#include <gcrypt.h>
 #include "ssl.h"
 #include "m_cap.h"
 
-#ifdef WINDOWS
-#pragma comment(lib, "libgnutls-13.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) */
 
 enum issl_status { ISSL_NONE, ISSL_HANDSHAKING_READ, ISSL_HANDSHAKING_WRITE, ISSL_HANDSHAKEN, ISSL_CLOSING, ISSL_CLOSED };
 
-static gnutls_x509_crt_t x509_cert;
+static std::vector<gnutls_x509_crt_t> x509_certs;
 static gnutls_x509_privkey_t x509_key;
 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;
-       st->ncerts = 1;
-       st->cert.x509 = &x509_cert;
+       st->ncerts = x509_certs.size();
+       st->cert.x509 = &x509_certs[0];
        st->key.x509 = x509_key;
        st->deinit_all = 0;
 
@@ -52,7 +46,7 @@ static ssize_t gnutls_pull_wrapper(gnutls_transport_ptr_t user_wrap, void* buffe
                errno = EAGAIN;
                return -1;
        }
-       int rv = recv(user->GetFd(), buffer, size, 0);
+       int rv = recv(user->GetFd(), reinterpret_cast<char *>(buffer), size, 0);
        if (rv < (int)size)
                ServerInstance->SE->ChangeEventMask(user, FD_READ_WILL_BLOCK);
        return rv;
@@ -66,12 +60,22 @@ static ssize_t gnutls_push_wrapper(gnutls_transport_ptr_t user_wrap, const void*
                errno = EAGAIN;
                return -1;
        }
-       int rv = send(user->GetFd(), buffer, size, 0);
+       int rv = send(user->GetFd(), reinterpret_cast<const char *>(buffer), size, 0);
        if (rv < (int)size)
                ServerInstance->SE->ChangeEventMask(user, FD_WRITE_WILL_BLOCK);
        return rv;
 }
 
+class RandGen : public HandlerBase2<void, char*, size_t>
+{
+ public:
+       RandGen() {}
+       void Call(char* buffer, size_t len)
+       {
+               gcry_randomize(buffer, len, GCRY_STRONG_RANDOM);
+       }
+};
+
 /** Represents an SSL user's extra data
  */
 class issl_session
@@ -86,26 +90,39 @@ public:
 class CommandStartTLS : public SplitCommand
 {
  public:
+       bool enabled;
        CommandStartTLS (Module* mod) : SplitCommand(mod, "STARTTLS")
        {
+               enabled = true;
                works_before_reg = true;
        }
 
        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 enabled", user->nick.c_str());
+                       return CMD_FAILURE;
+               }
+
+               if (user->registered == REG_ALL)
                {
-                       user->WriteNumeric(691, "%s :STARTTLS is not permitted after client registration has started", user->nick.c_str());
+                       user->WriteNumeric(691, "%s :STARTTLS is not permitted after client registration is complete", user->nick.c_str());
                }
                else
                {
                        if (!user->eh.GetIOHook())
                        {
                                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);
                        }
@@ -123,67 +140,73 @@ class ModuleSSLGnuTLS : public Module
 
        gnutls_certificate_credentials x509_cred;
        gnutls_dh_params dh_params;
+       gnutls_digest_algorithm_t hash;
 
-       std::string keyfile;
-       std::string certfile;
-
-       std::string cafile;
-       std::string crlfile;
        std::string sslports;
        int dh_bits;
 
        bool cred_alloc;
 
+       RandGen randhandler;
        CommandStartTLS starttls;
 
        GenericCap capHandler;
+       ServiceProvider iohook;
  public:
 
        ModuleSSLGnuTLS()
-               : starttls(this), capHandler(this, "tls")
+               : starttls(this), capHandler(this, "tls"), iohook(this, "ssl/gnutls", SERVICE_IOHOOK)
        {
-               ServerInstance->Modules->PublishInterface("BufferedSocketHook", this);
-
                sessions = new issl_session[ServerInstance->SE->GetMaxFds()];
 
                gnutls_global_init(); // This must be called once in the program
-               gnutls_x509_crt_init(&x509_cert);
                gnutls_x509_privkey_init(&x509_key);
 
                cred_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_OnRehash, I_OnModuleRehash, I_OnUserConnect,
                        I_OnEvent, I_OnHookIO };
                ServerInstance->Modules->Attach(eventlist, this, sizeof(eventlist)/sizeof(Implementation));
 
+               ServerInstance->Modules->AddService(iohook);
                ServerInstance->AddCommand(&starttls);
        }
 
        void OnRehash(User* user)
        {
-               ConfigReader Conf;
-
                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))
                {
-                       ListenSocket* port = ServerInstance->ports[i];
-                       if (port->bind_tag->getString("ssl") != "gnutls")
-                               continue;
+                       for (size_t i = 0; i < ServerInstance->ports.size(); i++)
+                       {
+                               ListenSocket* port = ServerInstance->ports[i];
+                               if (port->bind_tag->getString("ssl") != "gnutls")
+                                       continue;
 
-                       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());
+                               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 (port->bind_tag->getString("type", "clients") == "clients" && port->bind_addr != "127.0.0.1")
-                               sslports.append(portid).append(";");
-               }
+                               if (port->bind_tag->getString("type", "clients") == "clients" && port->bind_addr != "127.0.0.1")
+                                       sslports.append(portid).append(";");
+                       }
 
-               if (!sslports.empty())
-                       sslports.erase(sslports.end() - 1);
+                       if (!sslports.empty())
+                               sslports.erase(sslports.end() - 1);
+               }
        }
 
        void OnModuleRehash(User* user, const std::string &param)
@@ -191,32 +214,32 @@ class ModuleSSLGnuTLS : public Module
                if(param != "ssl")
                        return;
 
+               std::string keyfile;
+               std::string certfile;
+               std::string cafile;
+               std::string crlfile;
                OnRehash(user);
 
-               ConfigReader Conf;
-
-               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);
-
-               // Set all the default values needed.
-               if (cafile.empty())
-                       cafile = "conf/ca.pem";
+               ConfigTag* Conf = ServerInstance->Config->ConfValue("gnutls");
 
-               if (crlfile.empty())
-                       crlfile = "conf/crl.pem";
-
-               if (certfile.empty())
-                       certfile = "conf/cert.pem";
-
-               if (keyfile.empty())
-                       keyfile = "conf/key.pem";
+               cafile = Conf->getString("cafile", "conf/ca.pem");
+               crlfile = Conf->getString("crlfile", "conf/crl.pem");
+               certfile = Conf->getString("certfile", "conf/cert.pem");
+               keyfile = Conf->getString("keyfile", "conf/key.pem");
+               dh_bits = Conf->getInt("dhbits");
+               std::string hashname = Conf->getString("hash", "md5");
 
                if((dh_bits != 768) && (dh_bits != 1024) && (dh_bits != 2048) && (dh_bits != 3072) && (dh_bits != 4096))
                        dh_bits = 1024;
 
+               if (hashname == "md5")
+                       hash = GNUTLS_DIG_MD5;
+               else if (hashname == "sha1")
+                       hash = GNUTLS_DIG_SHA1;
+               else
+                       throw ModuleException("Unknown hash type " + hashname);
+
+
                int ret;
 
                if (cred_alloc)
@@ -224,6 +247,10 @@ class ModuleSSLGnuTLS : public Module
                        // 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;
@@ -248,13 +275,17 @@ class ModuleSSLGnuTLS : public Module
                gnutls_datum_t key_datum = { (unsigned char*)key_string.data(), key_string.length() };
 
                // If this fails, no SSL port will work. At all. So, do the smart thing - throw a ModuleException
-               if((ret = gnutls_x509_crt_import(x509_cert, &cert_datum, GNUTLS_X509_FMT_PEM)) < 0)
+               unsigned int certcount = Conf->getInt("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 < 0)
                        throw ModuleException("Unable to load GnuTLS server certificate (" + certfile + "): " + std::string(gnutls_strerror(ret)));
+               x509_certs.resize(certcount);
 
                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_cert, 1, x509_key)) < 0)
+               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)));
 
                gnutls_certificate_client_set_retrieve_function (x509_cred, cert_callback);
@@ -281,13 +312,18 @@ class ModuleSSLGnuTLS : public Module
 
        ~ModuleSSLGnuTLS()
        {
-               gnutls_x509_crt_deinit(x509_cert);
+               for(unsigned int i=0; i < x509_certs.size(); i++)
+                       gnutls_x509_crt_deinit(x509_certs[i]);
+               x509_certs.clear();
                gnutls_x509_privkey_deinit(x509_key);
-               gnutls_dh_params_deinit(dh_params);
-               gnutls_certificate_free_credentials(x509_cred);
+               if (cred_alloc)
+               {
+                       gnutls_dh_params_deinit(dh_params);
+                       gnutls_certificate_free_credentials(x509_cred);
+               }
                gnutls_global_deinit();
-               ServerInstance->Modules->UnpublishInterface("BufferedSocketHook", this);
                delete[] sessions;
+               ServerInstance->GenRandom = &ServerInstance->HandleGenRandom;
        }
 
        void OnCleanup(int target_type, void* item)
@@ -315,7 +351,8 @@ class ModuleSSLGnuTLS : public Module
        {
                if (!sslports.empty())
                        output.append(" SSL=" + sslports);
-               output.append(" STARTTLS");
+               if (starttls.enabled)
+                       output.append(" STARTTLS");
        }
 
        void OnHookIO(StreamSocket* user, ListenSocket* lsb)
@@ -479,17 +516,11 @@ class ModuleSSLGnuTLS : public Module
                                ServerInstance->SE->ChangeEventMask(user, FD_WANT_SINGLE_WRITE);
                                return 0;
                        }
-                       else if (ret == GNUTLS_E_AGAIN || ret == GNUTLS_E_INTERRUPTED)
+                       else if (ret == GNUTLS_E_AGAIN || ret == GNUTLS_E_INTERRUPTED || ret == 0)
                        {
                                ServerInstance->SE->ChangeEventMask(user, FD_WANT_SINGLE_WRITE);
                                return 0;
                        }
-                       else if (ret == 0)
-                       {
-                               CloseSession(session);
-                               user->SetError("SSL Connection closed");
-                               return -1;
-                       }
                        else // (ret < 0)
                        {
                                user->SetError(gnutls_strerror(ret));
@@ -551,13 +582,17 @@ class ModuleSSLGnuTLS : public Module
        {
                if (user->eh.GetIOHook() == this)
                {
-                       if (sessions[user->GetFd()].sess)
+                       if (sessions[user->eh.GetFd()].sess)
                        {
-                               SSLCertSubmission(user, this, ServerInstance->Modules->Find("m_sslinfo.so"), sessions[user->GetFd()].cert);
-                               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());
+                               ssl_cert* cert = sessions[user->eh.GetFd()].cert;
+                               std::string cipher = gnutls_kx_get_name(gnutls_kx_get(sessions[user->eh.GetFd()].sess));
+                               cipher.append("-").append(gnutls_cipher_get_name(gnutls_cipher_get(sessions[user->eh.GetFd()].sess))).append("-");
+                               cipher.append(gnutls_mac_get_name(gnutls_mac_get(sessions[user->eh.GetFd()].sess)));
+                               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());
                        }
                }
        }
@@ -649,7 +684,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);
                }
@@ -671,7 +706,8 @@ info_done_dealloc:
 
        void OnEvent(Event& ev)
        {
-               capHandler.HandleEvent(ev);
+               if (starttls.enabled)
+                       capHandler.HandleEvent(ev);
        }
 };