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