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