src/modules/extra/m_ssl_gnutls.cpp

   1 /*
   2  * InspIRCd -- Internet Relay Chat Daemon
   3  *
   4  *   Copyright (C) 2009-2010 Daniel De Graaf <danieldg@inspircd.org>
   5  *   Copyright (C) 2008 John Brooks <john.brooks@dereferenced.net>
   6  *   Copyright (C) 2006-2008 Craig Edwards <craigedwards@brainbox.cc>
   7  *   Copyright (C) 2007 Dennis Friis <peavey@inspircd.org>
   8  *   Copyright (C) 2006 Oliver Lupton <oliverlupton@gmail.com>
   9  *
  10  * This file is part of InspIRCd.  InspIRCd is free software: you can
  11  * redistribute it and/or modify it under the terms of the GNU General Public
  12  * License as published by the Free Software Foundation, version 2.
  13  *
  14  * This program is distributed in the hope that it will be useful, but WITHOUT
  15  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
  16  * FOR A PARTICULAR PURPOSE.  See the GNU General Public License for more
  17  * details.
  18  *
  19  * You should have received a copy of the GNU General Public License
  20  * along with this program.  If not, see <http://www.gnu.org/licenses/>.
  21  */
  22
  23
  24 #include "inspircd.h"
  25 #include "modules/ssl.h"
  26 #include <memory>
  27
  28 // Fix warnings about the use of commas at end of enumerator lists on C++03.
  29 #if defined __clang__
  30 # pragma clang diagnostic ignored "-Wc++11-extensions"
  31 #elif defined __GNUC__
  32 # pragma GCC diagnostic ignored "-pedantic"
  33 #endif
  34
  35 #include <gnutls/gnutls.h>
  36 #include <gnutls/x509.h>
  37
  38 #ifndef GNUTLS_VERSION_NUMBER
  39 #define GNUTLS_VERSION_NUMBER LIBGNUTLS_VERSION_NUMBER
  40 #define GNUTLS_VERSION LIBGNUTLS_VERSION
  41 #endif
  42
  43 // Check if the GnuTLS library is at least version major.minor.patch
  44 #define INSPIRCD_GNUTLS_HAS_VERSION(major, minor, patch) (GNUTLS_VERSION_NUMBER >= ((major << 16) | (minor << 8) | patch))
  45
  46 #if INSPIRCD_GNUTLS_HAS_VERSION(2, 9, 8)
  47 #define GNUTLS_HAS_MAC_GET_ID
  48 #include <gnutls/crypto.h>
  49 #endif
  50
  51 #if INSPIRCD_GNUTLS_HAS_VERSION(2, 12, 0)
  52 # define GNUTLS_HAS_RND
  53 #else
  54 # include <gcrypt.h>
  55 #endif
  56
  57 #ifdef _WIN32
  58 # pragma comment(lib, "libgnutls-28.lib")
  59 #endif
  60
  61 /* $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'") */
  62 /* $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'") */
  63
  64 // These don't exist in older GnuTLS versions
  65 #if INSPIRCD_GNUTLS_HAS_VERSION(2, 1, 7)
  66 #define GNUTLS_NEW_PRIO_API
  67 #endif
  68
  69 #if (!INSPIRCD_GNUTLS_HAS_VERSION(2, 0, 0))
  70 typedef gnutls_certificate_credentials_t gnutls_certificate_credentials;
  71 typedef gnutls_dh_params_t gnutls_dh_params;
  72 #endif
  73
  74 enum issl_status { ISSL_NONE, ISSL_HANDSHAKING, ISSL_HANDSHAKEN };
  75
  76 #if INSPIRCD_GNUTLS_HAS_VERSION(2, 12, 0)
  77 #define INSPIRCD_GNUTLS_HAS_VECTOR_PUSH
  78 #define GNUTLS_NEW_CERT_CALLBACK_API
  79 typedef gnutls_retr2_st cert_cb_last_param_type;
  80 #else
  81 typedef gnutls_retr_st cert_cb_last_param_type;
  82 #endif
  83
  84 #if INSPIRCD_GNUTLS_HAS_VERSION(3, 3, 5)
  85 #define INSPIRCD_GNUTLS_HAS_RECV_PACKET
  86 #endif
  87
  88 #if INSPIRCD_GNUTLS_HAS_VERSION(2, 99, 0)
  89 // The second parameter of gnutls_init() has changed in 2.99.0 from gnutls_connection_end_t to unsigned int
  90 // (it became a general flags parameter) and the enum has been deprecated and generates a warning on use.
  91 typedef unsigned int inspircd_gnutls_session_init_flags_t;
  92 #else
  93 typedef gnutls_connection_end_t inspircd_gnutls_session_init_flags_t;
  94 #endif
  95
  96 #if INSPIRCD_GNUTLS_HAS_VERSION(3, 1, 9)
  97 #define INSPIRCD_GNUTLS_HAS_CORK
  98 #endif
  99
 100 class RandGen : public HandlerBase2<void, char*, size_t>
 101 {
 102  public:
 103         void Call(char* buffer, size_t len)
 104         {
 105 #ifdef GNUTLS_HAS_RND
 106                 gnutls_rnd(GNUTLS_RND_RANDOM, buffer, len);
 107 #else
 108                 gcry_randomize(buffer, len, GCRY_STRONG_RANDOM);
 109 #endif
 110         }
 111 };
 112
 113 namespace GnuTLS
 114 {
 115         class Init
 116         {
 117          public:
 118                 Init() { gnutls_global_init(); }
 119                 ~Init() { gnutls_global_deinit(); }
 120         };
 121
 122         class Exception : public ModuleException
 123         {
 124          public:
 125                 Exception(const std::string& reason)
 126                         : ModuleException(reason) { }
 127         };
 128
 129         void ThrowOnError(int errcode, const char* msg)
 130         {
 131                 if (errcode < 0)
 132                 {
 133                         std::string reason = msg;
 134                         reason.append(" :").append(gnutls_strerror(errcode));
 135                         throw Exception(reason);
 136                 }
 137         }
 138
 139         /** Used to create a gnutls_datum_t* from a std::string
 140          */
 141         class Datum
 142         {
 143                 gnutls_datum_t datum;
 144
 145          public:
 146                 Datum(const std::string& dat)
 147                 {
 148                         datum.data = (unsigned char*)dat.data();
 149                         datum.size = static_cast<unsigned int>(dat.length());
 150                 }
 151
 152                 const gnutls_datum_t* get() const { return &datum; }
 153         };
 154
 155         class Hash
 156         {
 157                 gnutls_digest_algorithm_t hash;
 158
 159          public:
 160                 // Nothing to deallocate, constructor may throw freely
 161                 Hash(const std::string& hashname)
 162                 {
 163                         // As older versions of gnutls can't do this, let's disable it where needed.
 164 #ifdef GNUTLS_HAS_MAC_GET_ID
 165                         // As gnutls_digest_algorithm_t and gnutls_mac_algorithm_t are mapped 1:1, we can do this
 166                         // There is no gnutls_dig_get_id() at the moment, but it may come later
 167                         hash = (gnutls_digest_algorithm_t)gnutls_mac_get_id(hashname.c_str());
 168                         if (hash == GNUTLS_DIG_UNKNOWN)
 169                                 throw Exception("Unknown hash type " + hashname);
 170
 171                         // Check if the user is giving us something that is a valid MAC but not digest
 172                         gnutls_hash_hd_t is_digest;
 173                         if (gnutls_hash_init(&is_digest, hash) < 0)
 174                                 throw Exception("Unknown hash type " + hashname);
 175                         gnutls_hash_deinit(is_digest, NULL);
 176 #else
 177                         if (hashname == "md5")
 178                                 hash = GNUTLS_DIG_MD5;
 179                         else if (hashname == "sha1")
 180                                 hash = GNUTLS_DIG_SHA1;
 181 #ifdef INSPIRCD_GNUTLS_ENABLE_SHA256_FINGERPRINT
 182                         else if (hashname == "sha256")
 183                                 hash = GNUTLS_DIG_SHA256;
 184 #endif
 185                         else
 186                                 throw Exception("Unknown hash type " + hashname);
 187 #endif
 188                 }
 189
 190                 gnutls_digest_algorithm_t get() const { return hash; }
 191         };
 192
 193         class DHParams
 194         {
 195                 gnutls_dh_params_t dh_params;
 196
 197                 DHParams()
 198                 {
 199                         ThrowOnError(gnutls_dh_params_init(&dh_params), "gnutls_dh_params_init() failed");
 200                 }
 201
 202          public:
 203                 /** Import */
 204                 static std::auto_ptr<DHParams> Import(const std::string& dhstr)
 205                 {
 206                         std::auto_ptr<DHParams> dh(new DHParams);
 207                         int ret = gnutls_dh_params_import_pkcs3(dh->dh_params, Datum(dhstr).get(), GNUTLS_X509_FMT_PEM);
 208                         ThrowOnError(ret, "Unable to import DH params");
 209                         return dh;
 210                 }
 211
 212                 /** Generate */
 213                 static std::auto_ptr<DHParams> Generate(unsigned int bits)
 214                 {
 215                         std::auto_ptr<DHParams> dh(new DHParams);
 216                         ThrowOnError(gnutls_dh_params_generate2(dh->dh_params, bits), "Unable to generate DH params");
 217                         return dh;
 218                 }
 219
 220                 ~DHParams()
 221                 {
 222                         gnutls_dh_params_deinit(dh_params);
 223                 }
 224
 225                 const gnutls_dh_params_t& get() const { return dh_params; }
 226         };
 227
 228         class X509Key
 229         {
 230                 /** Ensure that the key is deinited in case the constructor of X509Key throws
 231                  */
 232                 class RAIIKey
 233                 {
 234                  public:
 235                         gnutls_x509_privkey_t key;
 236
 237                         RAIIKey()
 238                         {
 239                                 ThrowOnError(gnutls_x509_privkey_init(&key), "gnutls_x509_privkey_init() failed");
 240                         }
 241
 242                         ~RAIIKey()
 243                         {
 244                                 gnutls_x509_privkey_deinit(key);
 245                         }
 246                 } key;
 247
 248          public:
 249                 /** Import */
 250                 X509Key(const std::string& keystr)
 251                 {
 252                         int ret = gnutls_x509_privkey_import(key.key, Datum(keystr).get(), GNUTLS_X509_FMT_PEM);
 253                         ThrowOnError(ret, "Unable to import private key");
 254                 }
 255
 256                 gnutls_x509_privkey_t& get() { return key.key; }
 257         };
 258
 259         class X509CertList
 260         {
 261                 std::vector<gnutls_x509_crt_t> certs;
 262
 263          public:
 264                 /** Import */
 265                 X509CertList(const std::string& certstr)
 266                 {
 267                         unsigned int certcount = 3;
 268                         certs.resize(certcount);
 269                         Datum datum(certstr);
 270
 271                         int ret = gnutls_x509_crt_list_import(raw(), &certcount, datum.get(), GNUTLS_X509_FMT_PEM, GNUTLS_X509_CRT_LIST_IMPORT_FAIL_IF_EXCEED);
 272                         if (ret == GNUTLS_E_SHORT_MEMORY_BUFFER)
 273                         {
 274                                 // the buffer wasn't big enough to hold all certs but gnutls changed certcount to the number of available certs,
 275                                 // try again with a bigger buffer
 276                                 certs.resize(certcount);
 277                                 ret = gnutls_x509_crt_list_import(raw(), &certcount, datum.get(), GNUTLS_X509_FMT_PEM, GNUTLS_X509_CRT_LIST_IMPORT_FAIL_IF_EXCEED);
 278                         }
 279
 280                         ThrowOnError(ret, "Unable to load certificates");
 281
 282                         // Resize the vector to the actual number of certs because we rely on its size being correct
 283                         // when deallocating the certs
 284                         certs.resize(certcount);
 285                 }
 286
 287                 ~X509CertList()
 288                 {
 289                         for (std::vector<gnutls_x509_crt_t>::iterator i = certs.begin(); i != certs.end(); ++i)
 290                                 gnutls_x509_crt_deinit(*i);
 291                 }
 292
 293                 gnutls_x509_crt_t* raw() { return &certs[0]; }
 294                 unsigned int size() const { return certs.size(); }
 295         };
 296
 297         class X509CRL : public refcountbase
 298         {
 299                 class RAIICRL
 300                 {
 301                  public:
 302                         gnutls_x509_crl_t crl;
 303
 304                         RAIICRL()
 305                         {
 306                                 ThrowOnError(gnutls_x509_crl_init(&crl), "gnutls_x509_crl_init() failed");
 307                         }
 308
 309                         ~RAIICRL()
 310                         {
 311                                 gnutls_x509_crl_deinit(crl);
 312                         }
 313                 } crl;
 314
 315          public:
 316                 /** Import */
 317                 X509CRL(const std::string& crlstr)
 318                 {
 319                         int ret = gnutls_x509_crl_import(get(), Datum(crlstr).get(), GNUTLS_X509_FMT_PEM);
 320                         ThrowOnError(ret, "Unable to load certificate revocation list");
 321                 }
 322
 323                 gnutls_x509_crl_t& get() { return crl.crl; }
 324         };
 325
 326 #ifdef GNUTLS_NEW_PRIO_API
 327         class Priority
 328         {
 329                 gnutls_priority_t priority;
 330
 331          public:
 332                 Priority(const std::string& priorities)
 333                 {
 334                         // Try to set the priorities for ciphers, kex methods etc. to the user supplied string
 335                         // If the user did not supply anything then the string is already set to "NORMAL"
 336                         const char* priocstr = priorities.c_str();
 337                         const char* prioerror;
 338
 339                         int ret = gnutls_priority_init(&priority, priocstr, &prioerror);
 340                         if (ret < 0)
 341                         {
 342                                 // gnutls did not understand the user supplied string
 343                                 throw Exception("Unable to initialize priorities to \"" + priorities + "\": " + gnutls_strerror(ret) + " Syntax error at position " + ConvToStr((unsigned int) (prioerror - priocstr)));
 344                         }
 345                 }
 346
 347                 ~Priority()
 348                 {
 349                         gnutls_priority_deinit(priority);
 350                 }
 351
 352                 void SetupSession(gnutls_session_t sess)
 353                 {
 354                         gnutls_priority_set(sess, priority);
 355                 }
 356         };
 357 #else
 358         /** Dummy class, used when gnutls_priority_set() is not available
 359          */
 360         class Priority
 361         {
 362          public:
 363                 Priority(const std::string& priorities)
 364                 {
 365                         if (priorities != "NORMAL")
 366                                 throw Exception("You've set a non-default priority string, but GnuTLS lacks support for it");
 367                 }
 368
 369                 static void SetupSession(gnutls_session_t sess)
 370                 {
 371                         // Always set the default priorities
 372                         gnutls_set_default_priority(sess);
 373                 }
 374         };
 375 #endif
 376
 377         class CertCredentials
 378         {
 379                 /** DH parameters associated with these credentials
 380                  */
 381                 std::auto_ptr<DHParams> dh;
 382
 383          protected:
 384                 gnutls_certificate_credentials_t cred;
 385
 386          public:
 387                 CertCredentials()
 388                 {
 389                         ThrowOnError(gnutls_certificate_allocate_credentials(&cred), "Cannot allocate certificate credentials");
 390                 }
 391
 392                 ~CertCredentials()
 393                 {
 394                         gnutls_certificate_free_credentials(cred);
 395                 }
 396
 397                 /** Associates these credentials with the session
 398                  */
 399                 void SetupSession(gnutls_session_t sess)
 400                 {
 401                         gnutls_credentials_set(sess, GNUTLS_CRD_CERTIFICATE, cred);
 402                 }
 403
 404                 /** Set the given DH parameters to be used with these credentials
 405                  */
 406                 void SetDH(std::auto_ptr<DHParams>& DH)
 407                 {
 408                         dh = DH;
 409                         gnutls_certificate_set_dh_params(cred, dh->get());
 410                 }
 411         };
 412
 413         class X509Credentials : public CertCredentials
 414         {
 415                 /** Private key
 416                  */
 417                 X509Key key;
 418
 419                 /** Certificate list, presented to the peer
 420                  */
 421                 X509CertList certs;
 422
 423                 /** Trusted CA, may be NULL
 424                  */
 425                 std::auto_ptr<X509CertList> trustedca;
 426
 427                 /** Certificate revocation list, may be NULL
 428                  */
 429                 std::auto_ptr<X509CRL> crl;
 430
 431                 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);
 432
 433          public:
 434                 X509Credentials(const std::string& certstr, const std::string& keystr)
 435                         : key(keystr)
 436                         , certs(certstr)
 437                 {
 438                         // Throwing is ok here, the destructor of Credentials is called in that case
 439                         int ret = gnutls_certificate_set_x509_key(cred, certs.raw(), certs.size(), key.get());
 440                         ThrowOnError(ret, "Unable to set cert/key pair");
 441
 442 #ifdef GNUTLS_NEW_CERT_CALLBACK_API
 443                         gnutls_certificate_set_retrieve_function(cred, cert_callback);
 444 #else
 445                         gnutls_certificate_client_set_retrieve_function(cred, cert_callback);
 446 #endif
 447                 }
 448
 449                 /** Sets the trusted CA and the certificate revocation list
 450                  * to use when verifying certificates
 451                  */
 452                 void SetCA(std::auto_ptr<X509CertList>& certlist, std::auto_ptr<X509CRL>& CRL)
 453                 {
 454                         // Do nothing if certlist is NULL
 455                         if (certlist.get())
 456                         {
 457                                 int ret = gnutls_certificate_set_x509_trust(cred, certlist->raw(), certlist->size());
 458                                 ThrowOnError(ret, "gnutls_certificate_set_x509_trust() failed");
 459
 460                                 if (CRL.get())
 461                                 {
 462                                         ret = gnutls_certificate_set_x509_crl(cred, &CRL->get(), 1);
 463                                         ThrowOnError(ret, "gnutls_certificate_set_x509_crl() failed");
 464                                 }
 465
 466                                 trustedca = certlist;
 467                                 crl = CRL;
 468                         }
 469                 }
 470         };
 471
 472         class DataReader
 473         {
 474                 int retval;
 475 #ifdef INSPIRCD_GNUTLS_HAS_RECV_PACKET
 476                 gnutls_packet_t packet;
 477
 478          public:
 479                 DataReader(gnutls_session_t sess)
 480                 {
 481                         // Using the packet API avoids the final copy of the data which GnuTLS does if we supply
 482                         // our own buffer. Instead, we get the buffer containing the data from GnuTLS and copy it
 483                         // to the recvq directly from there in appendto().
 484                         retval = gnutls_record_recv_packet(sess, &packet);
 485                 }
 486
 487                 void appendto(std::string& recvq)
 488                 {
 489                         // Copy data from GnuTLS buffers to recvq
 490                         gnutls_datum_t datum;
 491                         gnutls_packet_get(packet, &datum, NULL);
 492                         recvq.append(reinterpret_cast<const char*>(datum.data), datum.size);
 493
 494                         gnutls_packet_deinit(packet);
 495                 }
 496 #else
 497                 char* const buffer;
 498
 499          public:
 500                 DataReader(gnutls_session_t sess)
 501                         : buffer(ServerInstance->GetReadBuffer())
 502                 {
 503                         // Read data from GnuTLS buffers into ReadBuffer
 504                         retval = gnutls_record_recv(sess, buffer, ServerInstance->Config->NetBufferSize);
 505                 }
 506
 507                 void appendto(std::string& recvq)
 508                 {
 509                         // Copy data from ReadBuffer to recvq
 510                         recvq.append(buffer, retval);
 511                 }
 512 #endif
 513
 514                 int ret() const { return retval; }
 515         };
 516
 517         class Profile : public refcountbase
 518         {
 519                 /** Name of this profile
 520                  */
 521                 const std::string name;
 522
 523                 /** X509 certificate(s) and key
 524                  */
 525                 X509Credentials x509cred;
 526
 527                 /** The minimum length in bits for the DH prime to be accepted as a client
 528                  */
 529                 unsigned int min_dh_bits;
 530
 531                 /** Hashing algorithm to use when generating certificate fingerprints
 532                  */
 533                 Hash hash;
 534
 535                 /** Priorities for ciphers, compression methods, etc.
 536                  */
 537                 Priority priority;
 538
 539                 /** Rough max size of records to send
 540                  */
 541                 const unsigned int outrecsize;
 542
 543                 Profile(const std::string& profilename, const std::string& certstr, const std::string& keystr,
 544                                 std::auto_ptr<DHParams>& DH, unsigned int mindh, const std::string& hashstr,
 545                                 const std::string& priostr, std::auto_ptr<X509CertList>& CA, std::auto_ptr<X509CRL>& CRL,
 546                                 unsigned int recsize)
 547                         : name(profilename)
 548                         , x509cred(certstr, keystr)
 549                         , min_dh_bits(mindh)
 550                         , hash(hashstr)
 551                         , priority(priostr)
 552                         , outrecsize(recsize)
 553                 {
 554                         x509cred.SetDH(DH);
 555                         x509cred.SetCA(CA, CRL);
 556                 }
 557
 558                 static std::string ReadFile(const std::string& filename)
 559                 {
 560                         FileReader reader(filename);
 561                         std::string ret = reader.GetString();
 562                         if (ret.empty())
 563                                 throw Exception("Cannot read file " + filename);
 564                         return ret;
 565                 }
 566
 567          public:
 568                 static reference<Profile> Create(const std::string& profilename, ConfigTag* tag)
 569                 {
 570                         std::string certstr = ReadFile(tag->getString("certfile", "cert.pem"));
 571                         std::string keystr = ReadFile(tag->getString("keyfile", "key.pem"));
 572
 573                         std::auto_ptr<DHParams> dh;
 574                         int gendh = tag->getInt("gendh");
 575                         if (gendh)
 576                         {
 577                                 gendh = (gendh < 1024 ? 1024 : gendh);
 578                                 dh = DHParams::Generate(gendh);
 579                         }
 580                         else
 581                                 dh = DHParams::Import(ReadFile(tag->getString("dhfile", "dhparams.pem")));
 582
 583                         // Use default priority string if this tag does not specify one
 584                         std::string priostr = tag->getString("priority", "NORMAL");
 585                         unsigned int mindh = tag->getInt("mindhbits", 1024);
 586                         std::string hashstr = tag->getString("hash", "md5");
 587
 588                         // Load trusted CA and revocation list, if set
 589                         std::auto_ptr<X509CertList> ca;
 590                         std::auto_ptr<X509CRL> crl;
 591                         std::string filename = tag->getString("cafile");
 592                         if (!filename.empty())
 593                         {
 594                                 ca.reset(new X509CertList(ReadFile(filename)));
 595
 596                                 filename = tag->getString("crlfile");
 597                                 if (!filename.empty())
 598                                         crl.reset(new X509CRL(ReadFile(filename)));
 599                         }
 600
 601 #ifdef INSPIRCD_GNUTLS_HAS_CORK
 602                         // If cork support is available outrecsize represents the (rough) max amount of data we give GnuTLS while corked
 603                         unsigned int outrecsize = tag->getInt("outrecsize", 2048, 512);
 604 #else
 605                         unsigned int outrecsize = tag->getInt("outrecsize", 2048, 512, 16384);
 606 #endif
 607                         return new Profile(profilename, certstr, keystr, dh, mindh, hashstr, priostr, ca, crl, outrecsize);
 608                 }
 609
 610                 /** Set up the given session with the settings in this profile
 611                  */
 612                 void SetupSession(gnutls_session_t sess)
 613                 {
 614                         priority.SetupSession(sess);
 615                         x509cred.SetupSession(sess);
 616                         gnutls_dh_set_prime_bits(sess, min_dh_bits);
 617
 618                         // Request client certificate if we are a server, no-op if we're a client
 619                         gnutls_certificate_server_set_request(sess, GNUTLS_CERT_REQUEST);
 620                 }
 621
 622                 const std::string& GetName() const { return name; }
 623                 X509Credentials& GetX509Credentials() { return x509cred; }
 624                 gnutls_digest_algorithm_t GetHash() const { return hash.get(); }
 625                 unsigned int GetOutgoingRecordSize() const { return outrecsize; }
 626         };
 627 }
 628
 629 class GnuTLSIOHook : public SSLIOHook
 630 {
 631  private:
 632         gnutls_session_t sess;
 633         issl_status status;
 634         reference<GnuTLS::Profile> profile;
 635 #ifdef INSPIRCD_GNUTLS_HAS_CORK
 636         size_t gbuffersize;
 637 #endif
 638
 639         void CloseSession()
 640         {
 641                 if (this->sess)
 642                 {
 643                         gnutls_bye(this->sess, GNUTLS_SHUT_WR);
 644                         gnutls_deinit(this->sess);
 645                 }
 646                 sess = NULL;
 647                 certificate = NULL;
 648                 status = ISSL_NONE;
 649         }
 650
 651         // Returns 1 if handshake succeeded, 0 if it is still in progress, -1 if it failed
 652         int Handshake(StreamSocket* user)
 653         {
 654                 int ret = gnutls_handshake(this->sess);
 655
 656                 if (ret < 0)
 657                 {
 658                         if(ret == GNUTLS_E_AGAIN || ret == GNUTLS_E_INTERRUPTED)
 659                         {
 660                                 // Handshake needs resuming later, read() or write() would have blocked.
 661                                 this->status = ISSL_HANDSHAKING;
 662
 663                                 if (gnutls_record_get_direction(this->sess) == 0)
 664                                 {
 665                                         // gnutls_handshake() wants to read() again.
 666                                         SocketEngine::ChangeEventMask(user, FD_WANT_POLL_READ | FD_WANT_NO_WRITE);
 667                                 }
 668                                 else
 669                                 {
 670                                         // gnutls_handshake() wants to write() again.
 671                                         SocketEngine::ChangeEventMask(user, FD_WANT_NO_READ | FD_WANT_SINGLE_WRITE);
 672                                 }
 673
 674                                 return 0;
 675                         }
 676                         else
 677                         {
 678                                 user->SetError("Handshake Failed - " + std::string(gnutls_strerror(ret)));
 679                                 CloseSession();
 680                                 return -1;
 681                         }
 682                 }
 683                 else
 684                 {
 685                         // Change the seesion state
 686                         this->status = ISSL_HANDSHAKEN;
 687
 688                         VerifyCertificate();
 689
 690                         // Finish writing, if any left
 691                         SocketEngine::ChangeEventMask(user, FD_WANT_POLL_READ | FD_WANT_NO_WRITE | FD_ADD_TRIAL_WRITE);
 692
 693                         return 1;
 694                 }
 695         }
 696
 697         void VerifyCertificate()
 698         {
 699                 unsigned int certstatus;
 700                 const gnutls_datum_t* cert_list;
 701                 int ret;
 702                 unsigned int cert_list_size;
 703                 gnutls_x509_crt_t cert;
 704                 char str[512];
 705                 unsigned char digest[512];
 706                 size_t digest_size = sizeof(digest);
 707                 size_t name_size = sizeof(str);
 708                 ssl_cert* certinfo = new ssl_cert;
 709                 this->certificate = certinfo;
 710
 711                 /* This verification function uses the trusted CAs in the credentials
 712                  * structure. So you must have installed one or more CA certificates.
 713                  */
 714                 ret = gnutls_certificate_verify_peers2(this->sess, &certstatus);
 715
 716                 if (ret < 0)
 717                 {
 718                         certinfo->error = std::string(gnutls_strerror(ret));
 719                         return;
 720                 }
 721
 722                 certinfo->invalid = (certstatus & GNUTLS_CERT_INVALID);
 723                 certinfo->unknownsigner = (certstatus & GNUTLS_CERT_SIGNER_NOT_FOUND);
 724                 certinfo->revoked = (certstatus & GNUTLS_CERT_REVOKED);
 725                 certinfo->trusted = !(certstatus & GNUTLS_CERT_SIGNER_NOT_CA);
 726
 727                 /* Up to here the process is the same for X.509 certificates and
 728                  * OpenPGP keys. From now on X.509 certificates are assumed. This can
 729                  * be easily extended to work with openpgp keys as well.
 730                  */
 731                 if (gnutls_certificate_type_get(this->sess) != GNUTLS_CRT_X509)
 732                 {
 733                         certinfo->error = "No X509 keys sent";
 734                         return;
 735                 }
 736
 737                 ret = gnutls_x509_crt_init(&cert);
 738                 if (ret < 0)
 739                 {
 740                         certinfo->error = gnutls_strerror(ret);
 741                         return;
 742                 }
 743
 744                 cert_list_size = 0;
 745                 cert_list = gnutls_certificate_get_peers(this->sess, &cert_list_size);
 746                 if (cert_list == NULL)
 747                 {
 748                         certinfo->error = "No certificate was found";
 749                         goto info_done_dealloc;
 750                 }
 751
 752                 /* This is not a real world example, since we only check the first
 753                  * certificate in the given chain.
 754                  */
 755
 756                 ret = gnutls_x509_crt_import(cert, &cert_list[0], GNUTLS_X509_FMT_DER);
 757                 if (ret < 0)
 758                 {
 759                         certinfo->error = gnutls_strerror(ret);
 760                         goto info_done_dealloc;
 761                 }
 762
 763                 if (gnutls_x509_crt_get_dn(cert, str, &name_size) == 0)
 764                 {
 765                         std::string& dn = certinfo->dn;
 766                         dn = str;
 767                         // Make sure there are no chars in the string that we consider invalid
 768                         if (dn.find_first_of("\r\n") != std::string::npos)
 769                                 dn.clear();
 770                 }
 771
 772                 name_size = sizeof(str);
 773                 if (gnutls_x509_crt_get_issuer_dn(cert, str, &name_size) == 0)
 774                 {
 775                         std::string& issuer = certinfo->issuer;
 776                         issuer = str;
 777                         if (issuer.find_first_of("\r\n") != std::string::npos)
 778                                 issuer.clear();
 779                 }
 780
 781                 if ((ret = gnutls_x509_crt_get_fingerprint(cert, profile->GetHash(), digest, &digest_size)) < 0)
 782                 {
 783                         certinfo->error = gnutls_strerror(ret);
 784                 }
 785                 else
 786                 {
 787                         certinfo->fingerprint = BinToHex(digest, digest_size);
 788                 }
 789
 790                 /* Beware here we do not check for errors.
 791                  */
 792                 if ((gnutls_x509_crt_get_expiration_time(cert) < ServerInstance->Time()) || (gnutls_x509_crt_get_activation_time(cert) > ServerInstance->Time()))
 793                 {
 794                         certinfo->error = "Not activated, or expired certificate";
 795                 }
 796
 797 info_done_dealloc:
 798                 gnutls_x509_crt_deinit(cert);
 799         }
 800
 801         // Returns 1 if application I/O should proceed, 0 if it must wait for the underlying protocol to progress, -1 on fatal error
 802         int PrepareIO(StreamSocket* sock)
 803         {
 804                 if (status == ISSL_HANDSHAKEN)
 805                         return 1;
 806                 else if (status == ISSL_HANDSHAKING)
 807                 {
 808                         // The handshake isn't finished, try to finish it
 809                         return Handshake(sock);
 810                 }
 811
 812                 CloseSession();
 813                 sock->SetError("No SSL session");
 814                 return -1;
 815         }
 816
 817 #ifdef INSPIRCD_GNUTLS_HAS_CORK
 818         int FlushBuffer(StreamSocket* sock)
 819         {
 820                 // If GnuTLS has some data buffered, write it
 821                 if (gbuffersize)
 822                         return HandleWriteRet(sock, gnutls_record_uncork(this->sess, 0));
 823                 return 1;
 824         }
 825 #endif
 826
 827         int HandleWriteRet(StreamSocket* sock, int ret)
 828         {
 829                 if (ret > 0)
 830                 {
 831 #ifdef INSPIRCD_GNUTLS_HAS_CORK
 832                         gbuffersize -= ret;
 833                         if (gbuffersize)
 834                         {
 835                                 SocketEngine::ChangeEventMask(sock, FD_WANT_SINGLE_WRITE);
 836                                 return 0;
 837                         }
 838 #endif
 839                         return ret;
 840                 }
 841                 else if (ret == GNUTLS_E_AGAIN || ret == GNUTLS_E_INTERRUPTED || ret == 0)
 842                 {
 843                         SocketEngine::ChangeEventMask(sock, FD_WANT_SINGLE_WRITE);
 844                         return 0;
 845                 }
 846                 else // (ret < 0)
 847                 {
 848                         sock->SetError(gnutls_strerror(ret));
 849                         CloseSession();
 850                         return -1;
 851                 }
 852         }
 853
 854         static const char* UnknownIfNULL(const char* str)
 855         {
 856                 return str ? str : "UNKNOWN";
 857         }
 858
 859         static ssize_t gnutls_pull_wrapper(gnutls_transport_ptr_t session_wrap, void* buffer, size_t size)
 860         {
 861                 StreamSocket* sock = reinterpret_cast<StreamSocket*>(session_wrap);
 862 #ifdef _WIN32
 863                 GnuTLSIOHook* session = static_cast<GnuTLSIOHook*>(sock->GetIOHook());
 864 #endif
 865
 866                 if (sock->GetEventMask() & FD_READ_WILL_BLOCK)
 867                 {
 868 #ifdef _WIN32
 869                         gnutls_transport_set_errno(session->sess, EAGAIN);
 870 #else
 871                         errno = EAGAIN;
 872 #endif
 873                         return -1;
 874                 }
 875
 876                 int rv = SocketEngine::Recv(sock, reinterpret_cast<char *>(buffer), size, 0);
 877
 878 #ifdef _WIN32
 879                 if (rv < 0)
 880                 {
 881                         /* Windows doesn't use errno, but gnutls does, so check SocketEngine::IgnoreError()
 882                          * and then set errno appropriately.
 883                          * The gnutls library may also have a different errno variable than us, see
 884                          * gnutls_transport_set_errno(3).
 885                          */
 886                         gnutls_transport_set_errno(session->sess, SocketEngine::IgnoreError() ? EAGAIN : errno);
 887                 }
 888 #endif
 889
 890                 if (rv < (int)size)
 891                         SocketEngine::ChangeEventMask(sock, FD_READ_WILL_BLOCK);
 892                 return rv;
 893         }
 894
 895 #ifdef INSPIRCD_GNUTLS_HAS_VECTOR_PUSH
 896         static ssize_t VectorPush(gnutls_transport_ptr_t transportptr, const giovec_t* iov, int iovcnt)
 897         {
 898                 StreamSocket* sock = reinterpret_cast<StreamSocket*>(transportptr);
 899 #ifdef _WIN32
 900                 GnuTLSIOHook* session = static_cast<GnuTLSIOHook*>(sock->GetIOHook());
 901 #endif
 902
 903                 if (sock->GetEventMask() & FD_WRITE_WILL_BLOCK)
 904                 {
 905 #ifdef _WIN32
 906                         gnutls_transport_set_errno(session->sess, EAGAIN);
 907 #else
 908                         errno = EAGAIN;
 909 #endif
 910                         return -1;
 911                 }
 912
 913                 // Cast the giovec_t to iovec not to IOVector so the correct function is called on Windows
 914                 int ret = SocketEngine::WriteV(sock, reinterpret_cast<const iovec*>(iov), iovcnt);
 915 #ifdef _WIN32
 916                 // See the function above for more info about the usage of gnutls_transport_set_errno() on Windows
 917                 if (ret < 0)
 918                         gnutls_transport_set_errno(session->sess, SocketEngine::IgnoreError() ? EAGAIN : errno);
 919 #endif
 920
 921                 int size = 0;
 922                 for (int i = 0; i < iovcnt; i++)
 923                         size += iov[i].iov_len;
 924
 925                 if (ret < size)
 926                         SocketEngine::ChangeEventMask(sock, FD_WRITE_WILL_BLOCK);
 927                 return ret;
 928         }
 929
 930 #else // INSPIRCD_GNUTLS_HAS_VECTOR_PUSH
 931         static ssize_t gnutls_push_wrapper(gnutls_transport_ptr_t session_wrap, const void* buffer, size_t size)
 932         {
 933                 StreamSocket* sock = reinterpret_cast<StreamSocket*>(session_wrap);
 934 #ifdef _WIN32
 935                 GnuTLSIOHook* session = static_cast<GnuTLSIOHook*>(sock->GetIOHook());
 936 #endif
 937
 938                 if (sock->GetEventMask() & FD_WRITE_WILL_BLOCK)
 939                 {
 940 #ifdef _WIN32
 941                         gnutls_transport_set_errno(session->sess, EAGAIN);
 942 #else
 943                         errno = EAGAIN;
 944 #endif
 945                         return -1;
 946                 }
 947
 948                 int rv = SocketEngine::Send(sock, reinterpret_cast<const char *>(buffer), size, 0);
 949
 950 #ifdef _WIN32
 951                 if (rv < 0)
 952                 {
 953                         /* Windows doesn't use errno, but gnutls does, so check SocketEngine::IgnoreError()
 954                          * and then set errno appropriately.
 955                          * The gnutls library may also have a different errno variable than us, see
 956                          * gnutls_transport_set_errno(3).
 957                          */
 958                         gnutls_transport_set_errno(session->sess, SocketEngine::IgnoreError() ? EAGAIN : errno);
 959                 }
 960 #endif
 961
 962                 if (rv < (int)size)
 963                         SocketEngine::ChangeEventMask(sock, FD_WRITE_WILL_BLOCK);
 964                 return rv;
 965         }
 966 #endif // INSPIRCD_GNUTLS_HAS_VECTOR_PUSH
 967
 968  public:
 969         GnuTLSIOHook(IOHookProvider* hookprov, StreamSocket* sock, inspircd_gnutls_session_init_flags_t flags, const reference<GnuTLS::Profile>& sslprofile)
 970                 : SSLIOHook(hookprov)
 971                 , sess(NULL)
 972                 , status(ISSL_NONE)
 973                 , profile(sslprofile)
 974 #ifdef INSPIRCD_GNUTLS_HAS_CORK
 975                 , gbuffersize(0)
 976 #endif
 977         {
 978                 gnutls_init(&sess, flags);
 979                 gnutls_transport_set_ptr(sess, reinterpret_cast<gnutls_transport_ptr_t>(sock));
 980 #ifdef INSPIRCD_GNUTLS_HAS_VECTOR_PUSH
 981                 gnutls_transport_set_vec_push_function(sess, VectorPush);
 982 #else
 983                 gnutls_transport_set_push_function(sess, gnutls_push_wrapper);
 984 #endif
 985                 gnutls_transport_set_pull_function(sess, gnutls_pull_wrapper);
 986                 profile->SetupSession(sess);
 987
 988                 sock->AddIOHook(this);
 989                 Handshake(sock);
 990         }
 991
 992         void OnStreamSocketClose(StreamSocket* user) CXX11_OVERRIDE
 993         {
 994                 CloseSession();
 995         }
 996
 997         int OnStreamSocketRead(StreamSocket* user, std::string& recvq) CXX11_OVERRIDE
 998         {
 999                 // Finish handshake if needed
1000                 int prepret = PrepareIO(user);
1001                 if (prepret <= 0)
1002                         return prepret;
1003
1004                 // If we resumed the handshake then this->status will be ISSL_HANDSHAKEN.
1005                 {
1006                         GnuTLS::DataReader reader(sess);
1007                         int ret = reader.ret();
1008                         if (ret > 0)
1009                         {
1010                                 reader.appendto(recvq);
1011                                 return 1;
1012                         }
1013                         else if (ret == GNUTLS_E_AGAIN || ret == GNUTLS_E_INTERRUPTED)
1014                         {
1015                                 return 0;
1016                         }
1017                         else if (ret == 0)
1018                         {
1019                                 user->SetError("Connection closed");
1020                                 CloseSession();
1021                                 return -1;
1022                         }
1023                         else
1024                         {
1025                                 user->SetError(gnutls_strerror(ret));
1026                                 CloseSession();
1027                                 return -1;
1028                         }
1029                 }
1030         }
1031
1032         int OnStreamSocketWrite(StreamSocket* user) CXX11_OVERRIDE
1033         {
1034                 // Finish handshake if needed
1035                 int prepret = PrepareIO(user);
1036                 if (prepret <= 0)
1037                         return prepret;
1038
1039                 // Session is ready for transferring application data
1040                 StreamSocket::SendQueue& sendq = user->GetSendQ();
1041
1042 #ifdef INSPIRCD_GNUTLS_HAS_CORK
1043                 while (true)
1044                 {
1045                         // If there is something in the GnuTLS buffer try to send() it
1046                         int ret = FlushBuffer(user);
1047                         if (ret <= 0)
1048                                 return ret; // Couldn't flush entire buffer, retry later (or close on error)
1049
1050                         // GnuTLS buffer is empty, if the sendq is empty as well then break to set FD_WANT_NO_WRITE
1051                         if (sendq.empty())
1052                                 break;
1053
1054                         // GnuTLS buffer is empty but sendq is not, begin sending data from the sendq
1055                         gnutls_record_cork(this->sess);
1056                         while ((!sendq.empty()) && (gbuffersize < profile->GetOutgoingRecordSize()))
1057                         {
1058                                 const StreamSocket::SendQueue::Element& elem = sendq.front();
1059                                 gbuffersize += elem.length();
1060                                 ret = gnutls_record_send(this->sess, elem.data(), elem.length());
1061                                 if (ret < 0)
1062                                 {
1063                                         CloseSession();
1064                                         return -1;
1065                                 }
1066                                 sendq.pop_front();
1067                         }
1068                 }
1069 #else
1070                 int ret = 0;
1071
1072                 while (!sendq.empty())
1073                 {
1074                         FlattenSendQueue(sendq, profile->GetOutgoingRecordSize());
1075                         const StreamSocket::SendQueue::Element& buffer = sendq.front();
1076                         ret = HandleWriteRet(user, gnutls_record_send(this->sess, buffer.data(), buffer.length()));
1077
1078                         if (ret <= 0)
1079                                 return ret;
1080                         else if (ret < (int)buffer.length())
1081                         {
1082                                 sendq.erase_front(ret);
1083                                 SocketEngine::ChangeEventMask(user, FD_WANT_SINGLE_WRITE);
1084                                 return 0;
1085                         }
1086
1087                         // Wrote entire record, continue sending
1088                         sendq.pop_front();
1089                 }
1090 #endif
1091
1092                 SocketEngine::ChangeEventMask(user, FD_WANT_NO_WRITE);
1093                 return 1;
1094         }
1095
1096         void TellCiphersAndFingerprint(LocalUser* user)
1097         {
1098                 if (sess)
1099                 {
1100                         std::string text = "*** You are connected using SSL cipher '";
1101                         GetCiphersuite(text);
1102                         text += '\'';
1103                         if (!certificate->fingerprint.empty())
1104                                 text += " and your SSL certificate fingerprint is " + certificate->fingerprint;
1105
1106                         user->WriteNotice(text);
1107                 }
1108         }
1109
1110         void GetCiphersuite(std::string& out) const
1111         {
1112                 out.append(UnknownIfNULL(gnutls_protocol_get_name(gnutls_protocol_get_version(sess)))).push_back('-');
1113                 out.append(UnknownIfNULL(gnutls_kx_get_name(gnutls_kx_get(sess)))).push_back('-');
1114                 out.append(UnknownIfNULL(gnutls_cipher_get_name(gnutls_cipher_get(sess)))).push_back('-');
1115                 out.append(UnknownIfNULL(gnutls_mac_get_name(gnutls_mac_get(sess))));
1116         }
1117
1118         GnuTLS::Profile* GetProfile() { return profile; }
1119         bool IsHandshakeDone() const { return (status == ISSL_HANDSHAKEN); }
1120 };
1121
1122 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)
1123 {
1124 #ifndef GNUTLS_NEW_CERT_CALLBACK_API
1125         st->type = GNUTLS_CRT_X509;
1126 #else
1127         st->cert_type = GNUTLS_CRT_X509;
1128         st->key_type = GNUTLS_PRIVKEY_X509;
1129 #endif
1130         StreamSocket* sock = reinterpret_cast<StreamSocket*>(gnutls_transport_get_ptr(sess));
1131         GnuTLS::X509Credentials& cred = static_cast<GnuTLSIOHook*>(sock->GetIOHook())->GetProfile()->GetX509Credentials();
1132
1133         st->ncerts = cred.certs.size();
1134         st->cert.x509 = cred.certs.raw();
1135         st->key.x509 = cred.key.get();
1136         st->deinit_all = 0;
1137
1138         return 0;
1139 }
1140
1141 class GnuTLSIOHookProvider : public refcountbase, public IOHookProvider
1142 {
1143         reference<GnuTLS::Profile> profile;
1144
1145  public:
1146         GnuTLSIOHookProvider(Module* mod, reference<GnuTLS::Profile>& prof)
1147                 : IOHookProvider(mod, "ssl/" + prof->GetName(), IOHookProvider::IOH_SSL)
1148                 , profile(prof)
1149         {
1150                 ServerInstance->Modules->AddService(*this);
1151         }
1152
1153         ~GnuTLSIOHookProvider()
1154         {
1155                 ServerInstance->Modules->DelService(*this);
1156         }
1157
1158         void OnAccept(StreamSocket* sock, irc::sockets::sockaddrs* client, irc::sockets::sockaddrs* server) CXX11_OVERRIDE
1159         {
1160                 new GnuTLSIOHook(this, sock, GNUTLS_SERVER, profile);
1161         }
1162
1163         void OnConnect(StreamSocket* sock) CXX11_OVERRIDE
1164         {
1165                 new GnuTLSIOHook(this, sock, GNUTLS_CLIENT, profile);
1166         }
1167 };
1168
1169 class ModuleSSLGnuTLS : public Module
1170 {
1171         typedef std::vector<reference<GnuTLSIOHookProvider> > ProfileList;
1172
1173         // First member of the class, gets constructed first and destructed last
1174         GnuTLS::Init libinit;
1175         RandGen randhandler;
1176         ProfileList profiles;
1177
1178         void ReadProfiles()
1179         {
1180                 // First, store all profiles in a new, temporary container. If no problems occur, swap the two
1181                 // containers; this way if something goes wrong we can go back and continue using the current profiles,
1182                 // avoiding unpleasant situations where no new SSL connections are possible.
1183                 ProfileList newprofiles;
1184
1185                 ConfigTagList tags = ServerInstance->Config->ConfTags("sslprofile");
1186                 if (tags.first == tags.second)
1187                 {
1188                         // No <sslprofile> tags found, create a profile named "gnutls" from settings in the <gnutls> block
1189                         const std::string defname = "gnutls";
1190                         ConfigTag* tag = ServerInstance->Config->ConfValue(defname);
1191                         ServerInstance->Logs->Log(MODNAME, LOG_DEFAULT, "No <sslprofile> tags found; using settings from the <gnutls> tag");
1192
1193                         try
1194                         {
1195                                 reference<GnuTLS::Profile> profile(GnuTLS::Profile::Create(defname, tag));
1196                                 newprofiles.push_back(new GnuTLSIOHookProvider(this, profile));
1197                         }
1198                         catch (CoreException& ex)
1199                         {
1200                                 throw ModuleException("Error while initializing the default SSL profile - " + ex.GetReason());
1201                         }
1202                 }
1203
1204                 for (ConfigIter i = tags.first; i != tags.second; ++i)
1205                 {
1206                         ConfigTag* tag = i->second;
1207                         if (tag->getString("provider") != "gnutls")
1208                                 continue;
1209
1210                         std::string name = tag->getString("name");
1211                         if (name.empty())
1212                         {
1213                                 ServerInstance->Logs->Log(MODNAME, LOG_DEFAULT, "Ignoring <sslprofile> tag without name at " + tag->getTagLocation());
1214                                 continue;
1215                         }
1216
1217                         reference<GnuTLS::Profile> profile;
1218                         try
1219                         {
1220                                 profile = GnuTLS::Profile::Create(name, tag);
1221                         }
1222                         catch (CoreException& ex)
1223                         {
1224                                 throw ModuleException("Error while initializing SSL profile \"" + name + "\" at " + tag->getTagLocation() + " - " + ex.GetReason());
1225                         }
1226
1227                         newprofiles.push_back(new GnuTLSIOHookProvider(this, profile));
1228                 }
1229
1230                 // New profiles are ok, begin using them
1231                 // Old profiles are deleted when their refcount drops to zero
1232                 profiles.swap(newprofiles);
1233         }
1234
1235  public:
1236         ModuleSSLGnuTLS()
1237         {
1238 #ifndef GNUTLS_HAS_RND
1239                 gcry_control (GCRYCTL_INITIALIZATION_FINISHED, 0);
1240 #endif
1241         }
1242
1243         void init() CXX11_OVERRIDE
1244         {
1245                 ServerInstance->Logs->Log(MODNAME, LOG_DEFAULT, "GnuTLS lib version %s module was compiled for " GNUTLS_VERSION, gnutls_check_version(NULL));
1246                 ReadProfiles();
1247                 ServerInstance->GenRandom = &randhandler;
1248         }
1249
1250         void OnModuleRehash(User* user, const std::string &param) CXX11_OVERRIDE
1251         {
1252                 if(param != "ssl")
1253                         return;
1254
1255                 try
1256                 {
1257                         ReadProfiles();
1258                 }
1259                 catch (ModuleException& ex)
1260                 {
1261                         ServerInstance->Logs->Log(MODNAME, LOG_DEFAULT, ex.GetReason() + " Not applying settings.");
1262                 }
1263         }
1264
1265         ~ModuleSSLGnuTLS()
1266         {
1267                 ServerInstance->GenRandom = &ServerInstance->HandleGenRandom;
1268         }
1269
1270         void OnCleanup(int target_type, void* item) CXX11_OVERRIDE
1271         {
1272                 if(target_type == TYPE_USER)
1273                 {
1274                         LocalUser* user = IS_LOCAL(static_cast<User*>(item));
1275
1276                         if (user && user->eh.GetIOHook() && user->eh.GetIOHook()->prov->creator == this)
1277                         {
1278                                 // User is using SSL, they're a local user, and they're using one of *our* SSL ports.
1279                                 // Potentially there could be multiple SSL modules loaded at once on different ports.
1280                                 ServerInstance->Users->QuitUser(user, "SSL module unloading");
1281                         }
1282                 }
1283         }
1284
1285         Version GetVersion() CXX11_OVERRIDE
1286         {
1287                 return Version("Provides SSL support for clients", VF_VENDOR);
1288         }
1289
1290         void OnUserConnect(LocalUser* user) CXX11_OVERRIDE
1291         {
1292                 IOHook* hook = user->eh.GetIOHook();
1293                 if (hook && hook->prov->creator == this)
1294                         static_cast<GnuTLSIOHook*>(hook)->TellCiphersAndFingerprint(user);
1295         }
1296
1297         ModResult OnCheckReady(LocalUser* user) CXX11_OVERRIDE
1298         {
1299                 if ((user->eh.GetIOHook()) && (user->eh.GetIOHook()->prov->creator == this))
1300                 {
1301                         GnuTLSIOHook* iohook = static_cast<GnuTLSIOHook*>(user->eh.GetIOHook());
1302                         if (!iohook->IsHandshakeDone())
1303                                 return MOD_RES_DENY;
1304                 }
1305
1306                 return MOD_RES_PASSTHRU;
1307         }
1308 };
1309
1310 MODULE_INIT(ModuleSSLGnuTLS)