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