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