]> git.netwichtig.de Git - user/henk/code/inspircd.git/blob - src/modules/extra/m_ssl_openssl.cpp
6d3eef3933af3009d51876296e55072a333e1a44
[user/henk/code/inspircd.git] / src / modules / extra / m_ssl_openssl.cpp
1 /*
2  * InspIRCd -- Internet Relay Chat Daemon
3  *
4  *   Copyright (C) 2009-2010 Daniel De Graaf <danieldg@inspircd.org>
5  *   Copyright (C) 2008 Pippijn van Steenhoven <pip88nl@gmail.com>
6  *   Copyright (C) 2006-2008 Craig Edwards <craigedwards@brainbox.cc>
7  *   Copyright (C) 2008 Thomas Stagner <aquanight@inspircd.org>
8  *   Copyright (C) 2007 Dennis Friis <peavey@inspircd.org>
9  *   Copyright (C) 2006 Oliver Lupton <oliverlupton@gmail.com>
10  *
11  * This file is part of InspIRCd.  InspIRCd is free software: you can
12  * redistribute it and/or modify it under the terms of the GNU General Public
13  * License as published by the Free Software Foundation, version 2.
14  *
15  * This program is distributed in the hope that it will be useful, but WITHOUT
16  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
17  * FOR A PARTICULAR PURPOSE.  See the GNU General Public License for more
18  * details.
19  *
20  * You should have received a copy of the GNU General Public License
21  * along with this program.  If not, see <http://www.gnu.org/licenses/>.
22  */
23
24
25 #include "inspircd.h"
26 #include "iohook.h"
27 #include "modules/ssl.h"
28
29 // Ignore OpenSSL deprecation warnings on OS X Lion and newer.
30 #if defined __APPLE__
31 # pragma GCC diagnostic ignored "-Wdeprecated-declarations"
32 #endif
33
34 // Fix warnings about the use of `long long` on C++03.
35 #if defined __clang__
36 # pragma clang diagnostic ignored "-Wc++11-long-long"
37 #elif defined __GNUC__
38 # pragma GCC diagnostic ignored "-Wlong-long"
39 #endif
40
41 #include <openssl/ssl.h>
42 #include <openssl/err.h>
43
44 #ifdef _WIN32
45 # pragma comment(lib, "ssleay32.lib")
46 # pragma comment(lib, "libeay32.lib")
47 #endif
48
49 /* $CompileFlags: pkgconfversion("openssl","0.9.7") pkgconfincludes("openssl","/openssl/ssl.h","") */
50 /* $LinkerFlags: rpath("pkg-config --libs openssl") pkgconflibs("openssl","/libssl.so","-lssl -lcrypto") */
51
52 #if ((OPENSSL_VERSION_NUMBER >= 0x10000000L) && (!(defined(OPENSSL_NO_ECDH))))
53 // OpenSSL 0.9.8 includes some ECC support, but it's unfinished. Enable only for 1.0.0 and later.
54 #define INSPIRCD_OPENSSL_ENABLE_ECDH
55 #endif
56
57 enum issl_status { ISSL_NONE, ISSL_HANDSHAKING, ISSL_OPEN };
58
59 static bool SelfSigned = false;
60 static int exdataindex;
61
62 char* get_error()
63 {
64         return ERR_error_string(ERR_get_error(), NULL);
65 }
66
67 static int OnVerify(int preverify_ok, X509_STORE_CTX* ctx);
68 static void StaticSSLInfoCallback(const SSL* ssl, int where, int rc);
69
70 namespace OpenSSL
71 {
72         class Exception : public ModuleException
73         {
74          public:
75                 Exception(const std::string& reason)
76                         : ModuleException(reason) { }
77         };
78
79         class DHParams
80         {
81                 DH* dh;
82
83          public:
84                 DHParams(const std::string& filename)
85                 {
86                         BIO* dhpfile = BIO_new_file(filename.c_str(), "r");
87                         if (dhpfile == NULL)
88                                 throw Exception("Couldn't open DH file " + filename);
89
90                         dh = PEM_read_bio_DHparams(dhpfile, NULL, NULL, NULL);
91                         BIO_free(dhpfile);
92
93                         if (!dh)
94                                 throw Exception("Couldn't read DH params from file " + filename);
95                 }
96
97                 ~DHParams()
98                 {
99                         DH_free(dh);
100                 }
101
102                 DH* get()
103                 {
104                         return dh;
105                 }
106         };
107
108         class Context
109         {
110                 SSL_CTX* const ctx;
111                 long ctx_options;
112
113          public:
114                 Context(SSL_CTX* context)
115                         : ctx(context)
116                 {
117                         // Sane default options for OpenSSL see https://www.openssl.org/docs/ssl/SSL_CTX_set_options.html
118                         // and when choosing a cipher, use the server's preferences instead of the client preferences.
119                         long opts = SSL_OP_NO_SSLv2 | SSL_OP_NO_SESSION_RESUMPTION_ON_RENEGOTIATION | SSL_OP_CIPHER_SERVER_PREFERENCE | SSL_OP_SINGLE_DH_USE;
120                         // Only turn options on if they exist
121 #ifdef SSL_OP_SINGLE_ECDH_USE
122                         opts |= SSL_OP_SINGLE_ECDH_USE;
123 #endif
124 #ifdef SSL_OP_NO_TICKET
125                         opts |= SSL_OP_NO_TICKET;
126 #endif
127
128                         ctx_options = SSL_CTX_set_options(ctx, opts);
129
130                         long mode = SSL_MODE_ENABLE_PARTIAL_WRITE | SSL_MODE_ACCEPT_MOVING_WRITE_BUFFER;
131 #ifdef SSL_MODE_RELEASE_BUFFERS
132                         mode |= SSL_MODE_RELEASE_BUFFERS;
133 #endif
134                         SSL_CTX_set_mode(ctx, mode);
135                         SSL_CTX_set_verify(ctx, SSL_VERIFY_PEER | SSL_VERIFY_CLIENT_ONCE, OnVerify);
136                         SSL_CTX_set_session_cache_mode(ctx, SSL_SESS_CACHE_OFF);
137                         SSL_CTX_set_info_callback(ctx, StaticSSLInfoCallback);
138                 }
139
140                 ~Context()
141                 {
142                         SSL_CTX_free(ctx);
143                 }
144
145                 bool SetDH(DHParams& dh)
146                 {
147                         ERR_clear_error();
148                         return (SSL_CTX_set_tmp_dh(ctx, dh.get()) >= 0);
149                 }
150
151 #ifdef INSPIRCD_OPENSSL_ENABLE_ECDH
152                 void SetECDH(const std::string& curvename)
153                 {
154                         int nid = OBJ_sn2nid(curvename.c_str());
155                         if (nid == 0)
156                                 throw Exception("Unknown curve: " + curvename);
157
158                         EC_KEY* eckey = EC_KEY_new_by_curve_name(nid);
159                         if (!eckey)
160                                 throw Exception("Unable to create EC key object");
161
162                         ERR_clear_error();
163                         bool ret = (SSL_CTX_set_tmp_ecdh(ctx, eckey) >= 0);
164                         EC_KEY_free(eckey);
165                         if (!ret)
166                                 throw Exception("Couldn't set ECDH parameters");
167                 }
168 #endif
169
170                 bool SetCiphers(const std::string& ciphers)
171                 {
172                         ERR_clear_error();
173                         return SSL_CTX_set_cipher_list(ctx, ciphers.c_str());
174                 }
175
176                 bool SetCerts(const std::string& filename)
177                 {
178                         ERR_clear_error();
179                         return SSL_CTX_use_certificate_chain_file(ctx, filename.c_str());
180                 }
181
182                 bool SetPrivateKey(const std::string& filename)
183                 {
184                         ERR_clear_error();
185                         return SSL_CTX_use_PrivateKey_file(ctx, filename.c_str(), SSL_FILETYPE_PEM);
186                 }
187
188                 bool SetCA(const std::string& filename)
189                 {
190                         ERR_clear_error();
191                         return SSL_CTX_load_verify_locations(ctx, filename.c_str(), 0);
192                 }
193
194                 long GetDefaultContextOptions() const
195                 {
196                         return ctx_options;
197                 }
198
199                 long SetRawContextOptions(long setoptions, long clearoptions)
200                 {
201                         // Clear everything
202                         SSL_CTX_clear_options(ctx, SSL_CTX_get_options(ctx));
203
204                         // Set the default options and what is in the conf
205                         SSL_CTX_set_options(ctx, ctx_options | setoptions);
206                         return SSL_CTX_clear_options(ctx, clearoptions);
207                 }
208
209                 SSL* CreateServerSession()
210                 {
211                         SSL* sess = SSL_new(ctx);
212                         SSL_set_accept_state(sess); // Act as server
213                         return sess;
214                 }
215
216                 SSL* CreateClientSession()
217                 {
218                         SSL* sess = SSL_new(ctx);
219                         SSL_set_connect_state(sess); // Act as client
220                         return sess;
221                 }
222         };
223
224         class Profile : public refcountbase
225         {
226                 /** Name of this profile
227                  */
228                 const std::string name;
229
230                 /** DH parameters in use
231                  */
232                 DHParams dh;
233
234                 /** OpenSSL makes us have two contexts, one for servers and one for clients
235                  */
236                 Context ctx;
237                 Context clictx;
238
239                 /** Digest to use when generating fingerprints
240                  */
241                 const EVP_MD* digest;
242
243                 /** Last error, set by error_callback()
244                  */
245                 std::string lasterr;
246
247                 /** True if renegotiations are allowed, false if not
248                  */
249                 const bool allowrenego;
250
251                 /** Rough max size of records to send
252                  */
253                 const unsigned int outrecsize;
254
255                 static int error_callback(const char* str, size_t len, void* u)
256                 {
257                         Profile* profile = reinterpret_cast<Profile*>(u);
258                         profile->lasterr = std::string(str, len - 1);
259                         return 0;
260                 }
261
262                 /** Set raw OpenSSL context (SSL_CTX) options from a config tag
263                  * @param ctxname Name of the context, client or server
264                  * @param tag Config tag defining this profile
265                  * @param context Context object to manipulate
266                  */
267                 void SetContextOptions(const std::string& ctxname, ConfigTag* tag, Context& context)
268                 {
269                         long setoptions = tag->getInt(ctxname + "setoptions");
270                         long clearoptions = tag->getInt(ctxname + "clearoptions");
271 #ifdef SSL_OP_NO_COMPRESSION
272                         if (!tag->getBool("compression", true))
273                                 setoptions |= SSL_OP_NO_COMPRESSION;
274 #endif
275                         if (!tag->getBool("sslv3", true))
276                                 setoptions |= SSL_OP_NO_SSLv3;
277                         if (!tag->getBool("tlsv1", true))
278                                 setoptions |= SSL_OP_NO_TLSv1;
279
280                         if (!setoptions && !clearoptions)
281                                 return; // Nothing to do
282
283                         ServerInstance->Logs->Log(MODNAME, LOG_DEBUG, "Setting %s %s context options, default: %ld set: %ld clear: %ld", name.c_str(), ctxname.c_str(), ctx.GetDefaultContextOptions(), setoptions, clearoptions);
284                         long final = context.SetRawContextOptions(setoptions, clearoptions);
285                         ServerInstance->Logs->Log(MODNAME, LOG_DEFAULT, "%s %s context options: %ld", name.c_str(), ctxname.c_str(), final);
286                 }
287
288          public:
289                 Profile(const std::string& profilename, ConfigTag* tag)
290                         : name(profilename)
291                         , dh(ServerInstance->Config->Paths.PrependConfig(tag->getString("dhfile", "dh.pem")))
292                         , ctx(SSL_CTX_new(SSLv23_server_method()))
293                         , clictx(SSL_CTX_new(SSLv23_client_method()))
294                         , allowrenego(tag->getBool("renegotiation")) // Disallow by default
295                         , outrecsize(tag->getInt("outrecsize", 2048, 512, 16384))
296                 {
297                         if ((!ctx.SetDH(dh)) || (!clictx.SetDH(dh)))
298                                 throw Exception("Couldn't set DH parameters");
299
300                         std::string hash = tag->getString("hash", "md5");
301                         digest = EVP_get_digestbyname(hash.c_str());
302                         if (digest == NULL)
303                                 throw Exception("Unknown hash type " + hash);
304
305                         std::string ciphers = tag->getString("ciphers");
306                         if (!ciphers.empty())
307                         {
308                                 if ((!ctx.SetCiphers(ciphers)) || (!clictx.SetCiphers(ciphers)))
309                                 {
310                                         ERR_print_errors_cb(error_callback, this);
311                                         throw Exception("Can't set cipher list to \"" + ciphers + "\" " + lasterr);
312                                 }
313                         }
314
315 #ifdef INSPIRCD_OPENSSL_ENABLE_ECDH
316                         std::string curvename = tag->getString("ecdhcurve", "prime256v1");
317                         if (!curvename.empty())
318                                 ctx.SetECDH(curvename);
319 #endif
320
321                         SetContextOptions("server", tag, ctx);
322                         SetContextOptions("client", tag, clictx);
323
324                         /* Load our keys and certificates
325                          * NOTE: OpenSSL's error logging API sucks, don't blame us for this clusterfuck.
326                          */
327                         std::string filename = ServerInstance->Config->Paths.PrependConfig(tag->getString("certfile", "cert.pem"));
328                         if ((!ctx.SetCerts(filename)) || (!clictx.SetCerts(filename)))
329                         {
330                                 ERR_print_errors_cb(error_callback, this);
331                                 throw Exception("Can't read certificate file: " + lasterr);
332                         }
333
334                         filename = ServerInstance->Config->Paths.PrependConfig(tag->getString("keyfile", "key.pem"));
335                         if ((!ctx.SetPrivateKey(filename)) || (!clictx.SetPrivateKey(filename)))
336                         {
337                                 ERR_print_errors_cb(error_callback, this);
338                                 throw Exception("Can't read key file: " + lasterr);
339                         }
340
341                         // Load the CAs we trust
342                         filename = ServerInstance->Config->Paths.PrependConfig(tag->getString("cafile", "ca.pem"));
343                         if ((!ctx.SetCA(filename)) || (!clictx.SetCA(filename)))
344                         {
345                                 ERR_print_errors_cb(error_callback, this);
346                                 ServerInstance->Logs->Log(MODNAME, LOG_DEFAULT, "Can't read CA list from %s. This is only a problem if you want to verify client certificates, otherwise it's safe to ignore this message. Error: %s", filename.c_str(), lasterr.c_str());
347                         }
348                 }
349
350                 const std::string& GetName() const { return name; }
351                 SSL* CreateServerSession() { return ctx.CreateServerSession(); }
352                 SSL* CreateClientSession() { return clictx.CreateClientSession(); }
353                 const EVP_MD* GetDigest() { return digest; }
354                 bool AllowRenegotiation() const { return allowrenego; }
355                 unsigned int GetOutgoingRecordSize() const { return outrecsize; }
356         };
357 }
358
359 static int OnVerify(int preverify_ok, X509_STORE_CTX *ctx)
360 {
361         /* XXX: This will allow self signed certificates.
362          * In the future if we want an option to not allow this,
363          * we can just return preverify_ok here, and openssl
364          * will boot off self-signed and invalid peer certs.
365          */
366         int ve = X509_STORE_CTX_get_error(ctx);
367
368         SelfSigned = (ve == X509_V_ERR_DEPTH_ZERO_SELF_SIGNED_CERT);
369
370         return 1;
371 }
372
373 class OpenSSLIOHook : public SSLIOHook
374 {
375  private:
376         SSL* sess;
377         issl_status status;
378         bool data_to_write;
379         reference<OpenSSL::Profile> profile;
380
381         // Returns 1 if handshake succeeded, 0 if it is still in progress, -1 if it failed
382         int Handshake(StreamSocket* user)
383         {
384                 ERR_clear_error();
385                 int ret = SSL_do_handshake(sess);
386                 if (ret < 0)
387                 {
388                         int err = SSL_get_error(sess, ret);
389
390                         if (err == SSL_ERROR_WANT_READ)
391                         {
392                                 SocketEngine::ChangeEventMask(user, FD_WANT_POLL_READ | FD_WANT_NO_WRITE);
393                                 this->status = ISSL_HANDSHAKING;
394                                 return 0;
395                         }
396                         else if (err == SSL_ERROR_WANT_WRITE)
397                         {
398                                 SocketEngine::ChangeEventMask(user, FD_WANT_NO_READ | FD_WANT_SINGLE_WRITE);
399                                 this->status = ISSL_HANDSHAKING;
400                                 return 0;
401                         }
402                         else
403                         {
404                                 CloseSession();
405                                 return -1;
406                         }
407                 }
408                 else if (ret > 0)
409                 {
410                         // Handshake complete.
411                         VerifyCertificate();
412
413                         status = ISSL_OPEN;
414
415                         SocketEngine::ChangeEventMask(user, FD_WANT_POLL_READ | FD_WANT_NO_WRITE | FD_ADD_TRIAL_WRITE);
416
417                         return 1;
418                 }
419                 else if (ret == 0)
420                 {
421                         CloseSession();
422                 }
423                 return -1;
424         }
425
426         void CloseSession()
427         {
428                 if (sess)
429                 {
430                         SSL_shutdown(sess);
431                         SSL_free(sess);
432                 }
433                 sess = NULL;
434                 certificate = NULL;
435                 status = ISSL_NONE;
436         }
437
438         void VerifyCertificate()
439         {
440                 X509* cert;
441                 ssl_cert* certinfo = new ssl_cert;
442                 this->certificate = certinfo;
443                 unsigned int n;
444                 unsigned char md[EVP_MAX_MD_SIZE];
445
446                 cert = SSL_get_peer_certificate(sess);
447
448                 if (!cert)
449                 {
450                         certinfo->error = "Could not get peer certificate: "+std::string(get_error());
451                         return;
452                 }
453
454                 certinfo->invalid = (SSL_get_verify_result(sess) != X509_V_OK);
455
456                 if (!SelfSigned)
457                 {
458                         certinfo->unknownsigner = false;
459                         certinfo->trusted = true;
460                 }
461                 else
462                 {
463                         certinfo->unknownsigner = true;
464                         certinfo->trusted = false;
465                 }
466
467                 char buf[512];
468                 X509_NAME_oneline(X509_get_subject_name(cert), buf, sizeof(buf));
469                 certinfo->dn = buf;
470                 // Make sure there are no chars in the string that we consider invalid
471                 if (certinfo->dn.find_first_of("\r\n") != std::string::npos)
472                         certinfo->dn.clear();
473
474                 X509_NAME_oneline(X509_get_issuer_name(cert), buf, sizeof(buf));
475                 certinfo->issuer = buf;
476                 if (certinfo->issuer.find_first_of("\r\n") != std::string::npos)
477                         certinfo->issuer.clear();
478
479                 if (!X509_digest(cert, profile->GetDigest(), md, &n))
480                 {
481                         certinfo->error = "Out of memory generating fingerprint";
482                 }
483                 else
484                 {
485                         certinfo->fingerprint = BinToHex(md, n);
486                 }
487
488                 if ((ASN1_UTCTIME_cmp_time_t(X509_get_notAfter(cert), ServerInstance->Time()) == -1) || (ASN1_UTCTIME_cmp_time_t(X509_get_notBefore(cert), ServerInstance->Time()) == 0))
489                 {
490                         certinfo->error = "Not activated, or expired certificate";
491                 }
492
493                 X509_free(cert);
494         }
495
496         void SSLInfoCallback(int where, int rc)
497         {
498                 if ((where & SSL_CB_HANDSHAKE_START) && (status == ISSL_OPEN))
499                 {
500                         if (profile->AllowRenegotiation())
501                                 return;
502
503                         // The other side is trying to renegotiate, kill the connection and change status
504                         // to ISSL_NONE so CheckRenego() closes the session
505                         status = ISSL_NONE;
506                         SocketEngine::Shutdown(SSL_get_fd(sess), 2);
507                 }
508         }
509
510         bool CheckRenego(StreamSocket* sock)
511         {
512                 if (status != ISSL_NONE)
513                         return true;
514
515                 ServerInstance->Logs->Log(MODNAME, LOG_DEBUG, "Session %p killed, attempted to renegotiate", (void*)sess);
516                 CloseSession();
517                 sock->SetError("Renegotiation is not allowed");
518                 return false;
519         }
520
521         // Returns 1 if application I/O should proceed, 0 if it must wait for the underlying protocol to progress, -1 on fatal error
522         int PrepareIO(StreamSocket* sock)
523         {
524                 if (status == ISSL_OPEN)
525                         return 1;
526                 else if (status == ISSL_HANDSHAKING)
527                 {
528                         // The handshake isn't finished, try to finish it
529                         return Handshake(sock);
530                 }
531
532                 CloseSession();
533                 return -1;
534         }
535
536         // Calls our private SSLInfoCallback()
537         friend void StaticSSLInfoCallback(const SSL* ssl, int where, int rc);
538
539  public:
540         OpenSSLIOHook(IOHookProvider* hookprov, StreamSocket* sock, SSL* session, const reference<OpenSSL::Profile>& sslprofile)
541                 : SSLIOHook(hookprov)
542                 , sess(session)
543                 , status(ISSL_NONE)
544                 , data_to_write(false)
545                 , profile(sslprofile)
546         {
547                 if (sess == NULL)
548                         return;
549                 if (SSL_set_fd(sess, sock->GetFd()) == 0)
550                         throw ModuleException("Can't set fd with SSL_set_fd: " + ConvToStr(sock->GetFd()));
551
552                 SSL_set_ex_data(sess, exdataindex, this);
553                 sock->AddIOHook(this);
554                 Handshake(sock);
555         }
556
557         void OnStreamSocketClose(StreamSocket* user) CXX11_OVERRIDE
558         {
559                 CloseSession();
560         }
561
562         int OnStreamSocketRead(StreamSocket* user, std::string& recvq) CXX11_OVERRIDE
563         {
564                 // Finish handshake if needed
565                 int prepret = PrepareIO(user);
566                 if (prepret <= 0)
567                         return prepret;
568
569                 // If we resumed the handshake then this->status will be ISSL_OPEN
570                 {
571                         ERR_clear_error();
572                         char* buffer = ServerInstance->GetReadBuffer();
573                         size_t bufsiz = ServerInstance->Config->NetBufferSize;
574                         int ret = SSL_read(sess, buffer, bufsiz);
575
576                         if (!CheckRenego(user))
577                                 return -1;
578
579                         if (ret > 0)
580                         {
581                                 recvq.append(buffer, ret);
582                                 if (data_to_write)
583                                         SocketEngine::ChangeEventMask(user, FD_WANT_POLL_READ | FD_WANT_SINGLE_WRITE);
584                                 return 1;
585                         }
586                         else if (ret == 0)
587                         {
588                                 // Client closed connection.
589                                 CloseSession();
590                                 user->SetError("Connection closed");
591                                 return -1;
592                         }
593                         else // if (ret < 0)
594                         {
595                                 int err = SSL_get_error(sess, ret);
596
597                                 if (err == SSL_ERROR_WANT_READ)
598                                 {
599                                         SocketEngine::ChangeEventMask(user, FD_WANT_POLL_READ);
600                                         return 0;
601                                 }
602                                 else if (err == SSL_ERROR_WANT_WRITE)
603                                 {
604                                         SocketEngine::ChangeEventMask(user, FD_WANT_NO_READ | FD_WANT_SINGLE_WRITE);
605                                         return 0;
606                                 }
607                                 else
608                                 {
609                                         CloseSession();
610                                         return -1;
611                                 }
612                         }
613                 }
614         }
615
616         int OnStreamSocketWrite(StreamSocket* user) CXX11_OVERRIDE
617         {
618                 // Finish handshake if needed
619                 int prepret = PrepareIO(user);
620                 if (prepret <= 0)
621                         return prepret;
622
623                 data_to_write = true;
624
625                 // Session is ready for transferring application data
626                 StreamSocket::SendQueue& sendq = user->GetSendQ();
627                 while (!sendq.empty())
628                 {
629                         ERR_clear_error();
630                         FlattenSendQueue(sendq, profile->GetOutgoingRecordSize());
631                         const StreamSocket::SendQueue::Element& buffer = sendq.front();
632                         int ret = SSL_write(sess, buffer.data(), buffer.size());
633
634                         if (!CheckRenego(user))
635                                 return -1;
636
637                         if (ret == (int)buffer.length())
638                         {
639                                 // Wrote entire record, continue sending
640                                 sendq.pop_front();
641                         }
642                         else if (ret > 0)
643                         {
644                                 sendq.erase_front(ret);
645                                 SocketEngine::ChangeEventMask(user, FD_WANT_SINGLE_WRITE);
646                                 return 0;
647                         }
648                         else if (ret == 0)
649                         {
650                                 CloseSession();
651                                 return -1;
652                         }
653                         else // if (ret < 0)
654                         {
655                                 int err = SSL_get_error(sess, ret);
656
657                                 if (err == SSL_ERROR_WANT_WRITE)
658                                 {
659                                         SocketEngine::ChangeEventMask(user, FD_WANT_SINGLE_WRITE);
660                                         return 0;
661                                 }
662                                 else if (err == SSL_ERROR_WANT_READ)
663                                 {
664                                         SocketEngine::ChangeEventMask(user, FD_WANT_POLL_READ);
665                                         return 0;
666                                 }
667                                 else
668                                 {
669                                         CloseSession();
670                                         return -1;
671                                 }
672                         }
673                 }
674
675                 data_to_write = false;
676                 SocketEngine::ChangeEventMask(user, FD_WANT_POLL_READ | FD_WANT_NO_WRITE);
677                 return 1;
678         }
679
680         void TellCiphersAndFingerprint(LocalUser* user)
681         {
682                 if (sess)
683                 {
684                         std::string text = "*** You are connected using SSL cipher '";
685                         GetCiphersuite(text);
686                         text += '\'';
687                         const std::string& fingerprint = certificate->fingerprint;
688                         if (!fingerprint.empty())
689                                 text += " and your SSL certificate fingerprint is " + fingerprint;
690
691                         user->WriteNotice(text);
692                 }
693         }
694
695         void GetCiphersuite(std::string& out) const
696         {
697                 out.append(SSL_get_version(sess)).push_back('-');
698                 out.append(SSL_get_cipher(sess));
699         }
700
701         bool IsHandshakeDone() const { return (status == ISSL_OPEN); }
702 };
703
704 static void StaticSSLInfoCallback(const SSL* ssl, int where, int rc)
705 {
706         OpenSSLIOHook* hook = static_cast<OpenSSLIOHook*>(SSL_get_ex_data(ssl, exdataindex));
707         hook->SSLInfoCallback(where, rc);
708 }
709
710 class OpenSSLIOHookProvider : public refcountbase, public IOHookProvider
711 {
712         reference<OpenSSL::Profile> profile;
713
714  public:
715         OpenSSLIOHookProvider(Module* mod, reference<OpenSSL::Profile>& prof)
716                 : IOHookProvider(mod, "ssl/" + prof->GetName(), IOHookProvider::IOH_SSL)
717                 , profile(prof)
718         {
719                 ServerInstance->Modules->AddService(*this);
720         }
721
722         ~OpenSSLIOHookProvider()
723         {
724                 ServerInstance->Modules->DelService(*this);
725         }
726
727         void OnAccept(StreamSocket* sock, irc::sockets::sockaddrs* client, irc::sockets::sockaddrs* server) CXX11_OVERRIDE
728         {
729                 new OpenSSLIOHook(this, sock, profile->CreateServerSession(), profile);
730         }
731
732         void OnConnect(StreamSocket* sock) CXX11_OVERRIDE
733         {
734                 new OpenSSLIOHook(this, sock, profile->CreateClientSession(), profile);
735         }
736 };
737
738 class ModuleSSLOpenSSL : public Module
739 {
740         typedef std::vector<reference<OpenSSLIOHookProvider> > ProfileList;
741
742         ProfileList profiles;
743
744         void ReadProfiles()
745         {
746                 ProfileList newprofiles;
747                 ConfigTagList tags = ServerInstance->Config->ConfTags("sslprofile");
748                 if (tags.first == tags.second)
749                 {
750                         // Create a default profile named "openssl"
751                         const std::string defname = "openssl";
752                         ConfigTag* tag = ServerInstance->Config->ConfValue(defname);
753                         ServerInstance->Logs->Log(MODNAME, LOG_DEFAULT, "No <sslprofile> tags found, using settings from the <openssl> tag");
754
755                         try
756                         {
757                                 reference<OpenSSL::Profile> profile(new OpenSSL::Profile(defname, tag));
758                                 newprofiles.push_back(new OpenSSLIOHookProvider(this, profile));
759                         }
760                         catch (OpenSSL::Exception& ex)
761                         {
762                                 throw ModuleException("Error while initializing the default SSL profile - " + ex.GetReason());
763                         }
764                 }
765
766                 for (ConfigIter i = tags.first; i != tags.second; ++i)
767                 {
768                         ConfigTag* tag = i->second;
769                         if (tag->getString("provider") != "openssl")
770                                 continue;
771
772                         std::string name = tag->getString("name");
773                         if (name.empty())
774                         {
775                                 ServerInstance->Logs->Log(MODNAME, LOG_DEFAULT, "Ignoring <sslprofile> tag without name at " + tag->getTagLocation());
776                                 continue;
777                         }
778
779                         reference<OpenSSL::Profile> profile;
780                         try
781                         {
782                                 profile = new OpenSSL::Profile(name, tag);
783                         }
784                         catch (CoreException& ex)
785                         {
786                                 throw ModuleException("Error while initializing SSL profile \"" + name + "\" at " + tag->getTagLocation() + " - " + ex.GetReason());
787                         }
788
789                         newprofiles.push_back(new OpenSSLIOHookProvider(this, profile));
790                 }
791
792                 profiles.swap(newprofiles);
793         }
794
795  public:
796         ModuleSSLOpenSSL()
797         {
798                 // Initialize OpenSSL
799                 SSL_library_init();
800                 SSL_load_error_strings();
801         }
802
803         void init() CXX11_OVERRIDE
804         {
805                 ServerInstance->Logs->Log(MODNAME, LOG_DEFAULT, "OpenSSL lib version \"%s\" module was compiled for \"" OPENSSL_VERSION_TEXT "\"", SSLeay_version(SSLEAY_VERSION));
806
807                 // Register application specific data
808                 char exdatastr[] = "inspircd";
809                 exdataindex = SSL_get_ex_new_index(0, exdatastr, NULL, NULL, NULL);
810                 if (exdataindex < 0)
811                         throw ModuleException("Failed to register application specific data");
812
813                 ReadProfiles();
814         }
815
816         void OnModuleRehash(User* user, const std::string &param) CXX11_OVERRIDE
817         {
818                 if (param != "ssl")
819                         return;
820
821                 try
822                 {
823                         ReadProfiles();
824                 }
825                 catch (ModuleException& ex)
826                 {
827                         ServerInstance->Logs->Log(MODNAME, LOG_DEFAULT, ex.GetReason() + " Not applying settings.");
828                 }
829         }
830
831         void OnUserConnect(LocalUser* user) CXX11_OVERRIDE
832         {
833                 IOHook* hook = user->eh.GetIOHook();
834                 if (hook && hook->prov->creator == this)
835                         static_cast<OpenSSLIOHook*>(hook)->TellCiphersAndFingerprint(user);
836         }
837
838         void OnCleanup(int target_type, void* item) CXX11_OVERRIDE
839         {
840                 if (target_type == TYPE_USER)
841                 {
842                         LocalUser* user = IS_LOCAL((User*)item);
843
844                         if (user && user->eh.GetIOHook() && user->eh.GetIOHook()->prov->creator == this)
845                         {
846                                 // User is using SSL, they're a local user, and they're using one of *our* SSL ports.
847                                 // Potentially there could be multiple SSL modules loaded at once on different ports.
848                                 ServerInstance->Users->QuitUser(user, "SSL module unloading");
849                         }
850                 }
851         }
852
853         ModResult OnCheckReady(LocalUser* user) CXX11_OVERRIDE
854         {
855                 if ((user->eh.GetIOHook()) && (user->eh.GetIOHook()->prov->creator == this))
856                 {
857                         OpenSSLIOHook* iohook = static_cast<OpenSSLIOHook*>(user->eh.GetIOHook());
858                         if (!iohook->IsHandshakeDone())
859                                 return MOD_RES_DENY;
860                 }
861
862                 return MOD_RES_PASSTHRU;
863         }
864
865         Version GetVersion() CXX11_OVERRIDE
866         {
867                 return Version("Provides SSL support for clients", VF_VENDOR);
868         }
869 };
870
871 MODULE_INIT(ModuleSSLOpenSSL)