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