]> git.netwichtig.de Git - user/henk/code/inspircd.git/blob - src/modules/extra/m_ssl_openssl.cpp
fc7212999640d80e7b569bca1045df037ce60fa1
[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 (SSL_set_fd(sess, sock->GetFd()) == 0)
548                         throw ModuleException("Can't set fd with SSL_set_fd: " + ConvToStr(sock->GetFd()));
549
550                 SSL_set_ex_data(sess, exdataindex, this);
551                 sock->AddIOHook(this);
552                 Handshake(sock);
553         }
554
555         void OnStreamSocketClose(StreamSocket* user) CXX11_OVERRIDE
556         {
557                 CloseSession();
558         }
559
560         int OnStreamSocketRead(StreamSocket* user, std::string& recvq) CXX11_OVERRIDE
561         {
562                 // Finish handshake if needed
563                 int prepret = PrepareIO(user);
564                 if (prepret <= 0)
565                         return prepret;
566
567                 // If we resumed the handshake then this->status will be ISSL_OPEN
568                 {
569                         ERR_clear_error();
570                         char* buffer = ServerInstance->GetReadBuffer();
571                         size_t bufsiz = ServerInstance->Config->NetBufferSize;
572                         int ret = SSL_read(sess, buffer, bufsiz);
573
574                         if (!CheckRenego(user))
575                                 return -1;
576
577                         if (ret > 0)
578                         {
579                                 recvq.append(buffer, ret);
580                                 if (data_to_write)
581                                         SocketEngine::ChangeEventMask(user, FD_WANT_POLL_READ | FD_WANT_SINGLE_WRITE);
582                                 return 1;
583                         }
584                         else if (ret == 0)
585                         {
586                                 // Client closed connection.
587                                 CloseSession();
588                                 user->SetError("Connection closed");
589                                 return -1;
590                         }
591                         else // if (ret < 0)
592                         {
593                                 int err = SSL_get_error(sess, ret);
594
595                                 if (err == SSL_ERROR_WANT_READ)
596                                 {
597                                         SocketEngine::ChangeEventMask(user, FD_WANT_POLL_READ);
598                                         return 0;
599                                 }
600                                 else if (err == SSL_ERROR_WANT_WRITE)
601                                 {
602                                         SocketEngine::ChangeEventMask(user, FD_WANT_NO_READ | FD_WANT_SINGLE_WRITE);
603                                         return 0;
604                                 }
605                                 else
606                                 {
607                                         CloseSession();
608                                         return -1;
609                                 }
610                         }
611                 }
612         }
613
614         int OnStreamSocketWrite(StreamSocket* user) CXX11_OVERRIDE
615         {
616                 // Finish handshake if needed
617                 int prepret = PrepareIO(user);
618                 if (prepret <= 0)
619                         return prepret;
620
621                 data_to_write = true;
622
623                 // Session is ready for transferring application data
624                 StreamSocket::SendQueue& sendq = user->GetSendQ();
625                 while (!sendq.empty())
626                 {
627                         ERR_clear_error();
628                         FlattenSendQueue(sendq, profile->GetOutgoingRecordSize());
629                         const StreamSocket::SendQueue::Element& buffer = sendq.front();
630                         int ret = SSL_write(sess, buffer.data(), buffer.size());
631
632                         if (!CheckRenego(user))
633                                 return -1;
634
635                         if (ret == (int)buffer.length())
636                         {
637                                 // Wrote entire record, continue sending
638                                 sendq.pop_front();
639                         }
640                         else if (ret > 0)
641                         {
642                                 sendq.erase_front(ret);
643                                 SocketEngine::ChangeEventMask(user, FD_WANT_SINGLE_WRITE);
644                                 return 0;
645                         }
646                         else if (ret == 0)
647                         {
648                                 CloseSession();
649                                 return -1;
650                         }
651                         else // if (ret < 0)
652                         {
653                                 int err = SSL_get_error(sess, ret);
654
655                                 if (err == SSL_ERROR_WANT_WRITE)
656                                 {
657                                         SocketEngine::ChangeEventMask(user, FD_WANT_SINGLE_WRITE);
658                                         return 0;
659                                 }
660                                 else if (err == SSL_ERROR_WANT_READ)
661                                 {
662                                         SocketEngine::ChangeEventMask(user, FD_WANT_POLL_READ);
663                                         return 0;
664                                 }
665                                 else
666                                 {
667                                         CloseSession();
668                                         return -1;
669                                 }
670                         }
671                 }
672
673                 data_to_write = false;
674                 SocketEngine::ChangeEventMask(user, FD_WANT_POLL_READ | FD_WANT_NO_WRITE);
675                 return 1;
676         }
677
678         void TellCiphersAndFingerprint(LocalUser* user)
679         {
680                 if (sess)
681                 {
682                         std::string text = "*** You are connected using SSL cipher '";
683                         GetCiphersuite(text);
684                         text += '\'';
685                         const std::string& fingerprint = certificate->fingerprint;
686                         if (!fingerprint.empty())
687                                 text += " and your SSL certificate fingerprint is " + fingerprint;
688
689                         user->WriteNotice(text);
690                 }
691         }
692
693         void GetCiphersuite(std::string& out) const
694         {
695                 out.append(SSL_get_version(sess)).push_back('-');
696                 out.append(SSL_get_cipher(sess));
697         }
698
699         bool IsHandshakeDone() const { return (status == ISSL_OPEN); }
700 };
701
702 static void StaticSSLInfoCallback(const SSL* ssl, int where, int rc)
703 {
704         OpenSSLIOHook* hook = static_cast<OpenSSLIOHook*>(SSL_get_ex_data(ssl, exdataindex));
705         hook->SSLInfoCallback(where, rc);
706 }
707
708 class OpenSSLIOHookProvider : public refcountbase, public IOHookProvider
709 {
710         reference<OpenSSL::Profile> profile;
711
712  public:
713         OpenSSLIOHookProvider(Module* mod, reference<OpenSSL::Profile>& prof)
714                 : IOHookProvider(mod, "ssl/" + prof->GetName(), IOHookProvider::IOH_SSL)
715                 , profile(prof)
716         {
717                 ServerInstance->Modules->AddService(*this);
718         }
719
720         ~OpenSSLIOHookProvider()
721         {
722                 ServerInstance->Modules->DelService(*this);
723         }
724
725         void OnAccept(StreamSocket* sock, irc::sockets::sockaddrs* client, irc::sockets::sockaddrs* server) CXX11_OVERRIDE
726         {
727                 new OpenSSLIOHook(this, sock, profile->CreateServerSession(), profile);
728         }
729
730         void OnConnect(StreamSocket* sock) CXX11_OVERRIDE
731         {
732                 new OpenSSLIOHook(this, sock, profile->CreateClientSession(), profile);
733         }
734 };
735
736 class ModuleSSLOpenSSL : public Module
737 {
738         typedef std::vector<reference<OpenSSLIOHookProvider> > ProfileList;
739
740         ProfileList profiles;
741
742         void ReadProfiles()
743         {
744                 ProfileList newprofiles;
745                 ConfigTagList tags = ServerInstance->Config->ConfTags("sslprofile");
746                 if (tags.first == tags.second)
747                 {
748                         // Create a default profile named "openssl"
749                         const std::string defname = "openssl";
750                         ConfigTag* tag = ServerInstance->Config->ConfValue(defname);
751                         ServerInstance->Logs->Log(MODNAME, LOG_DEFAULT, "No <sslprofile> tags found, using settings from the <openssl> tag");
752
753                         try
754                         {
755                                 reference<OpenSSL::Profile> profile(new OpenSSL::Profile(defname, tag));
756                                 newprofiles.push_back(new OpenSSLIOHookProvider(this, profile));
757                         }
758                         catch (OpenSSL::Exception& ex)
759                         {
760                                 throw ModuleException("Error while initializing the default SSL profile - " + ex.GetReason());
761                         }
762                 }
763
764                 for (ConfigIter i = tags.first; i != tags.second; ++i)
765                 {
766                         ConfigTag* tag = i->second;
767                         if (tag->getString("provider") != "openssl")
768                                 continue;
769
770                         std::string name = tag->getString("name");
771                         if (name.empty())
772                         {
773                                 ServerInstance->Logs->Log(MODNAME, LOG_DEFAULT, "Ignoring <sslprofile> tag without name at " + tag->getTagLocation());
774                                 continue;
775                         }
776
777                         reference<OpenSSL::Profile> profile;
778                         try
779                         {
780                                 profile = new OpenSSL::Profile(name, tag);
781                         }
782                         catch (CoreException& ex)
783                         {
784                                 throw ModuleException("Error while initializing SSL profile \"" + name + "\" at " + tag->getTagLocation() + " - " + ex.GetReason());
785                         }
786
787                         newprofiles.push_back(new OpenSSLIOHookProvider(this, profile));
788                 }
789
790                 profiles.swap(newprofiles);
791         }
792
793  public:
794         ModuleSSLOpenSSL()
795         {
796                 // Initialize OpenSSL
797                 SSL_library_init();
798                 SSL_load_error_strings();
799         }
800
801         void init() CXX11_OVERRIDE
802         {
803                 ServerInstance->Logs->Log(MODNAME, LOG_DEFAULT, "OpenSSL lib version \"%s\" module was compiled for \"" OPENSSL_VERSION_TEXT "\"", SSLeay_version(SSLEAY_VERSION));
804
805                 // Register application specific data
806                 char exdatastr[] = "inspircd";
807                 exdataindex = SSL_get_ex_new_index(0, exdatastr, NULL, NULL, NULL);
808                 if (exdataindex < 0)
809                         throw ModuleException("Failed to register application specific data");
810
811                 ReadProfiles();
812         }
813
814         void OnModuleRehash(User* user, const std::string &param) CXX11_OVERRIDE
815         {
816                 if (param != "ssl")
817                         return;
818
819                 try
820                 {
821                         ReadProfiles();
822                 }
823                 catch (ModuleException& ex)
824                 {
825                         ServerInstance->Logs->Log(MODNAME, LOG_DEFAULT, ex.GetReason() + " Not applying settings.");
826                 }
827         }
828
829         void OnUserConnect(LocalUser* user) CXX11_OVERRIDE
830         {
831                 IOHook* hook = user->eh.GetIOHook();
832                 if (hook && hook->prov->creator == this)
833                         static_cast<OpenSSLIOHook*>(hook)->TellCiphersAndFingerprint(user);
834         }
835
836         void OnCleanup(int target_type, void* item) CXX11_OVERRIDE
837         {
838                 if (target_type == TYPE_USER)
839                 {
840                         LocalUser* user = IS_LOCAL((User*)item);
841
842                         if (user && user->eh.GetIOHook() && user->eh.GetIOHook()->prov->creator == this)
843                         {
844                                 // User is using SSL, they're a local user, and they're using one of *our* SSL ports.
845                                 // Potentially there could be multiple SSL modules loaded at once on different ports.
846                                 ServerInstance->Users->QuitUser(user, "SSL module unloading");
847                         }
848                 }
849         }
850
851         ModResult OnCheckReady(LocalUser* user) CXX11_OVERRIDE
852         {
853                 if ((user->eh.GetIOHook()) && (user->eh.GetIOHook()->prov->creator == this))
854                 {
855                         OpenSSLIOHook* iohook = static_cast<OpenSSLIOHook*>(user->eh.GetIOHook());
856                         if (!iohook->IsHandshakeDone())
857                                 return MOD_RES_DENY;
858                 }
859
860                 return MOD_RES_PASSTHRU;
861         }
862
863         Version GetVersion() CXX11_OVERRIDE
864         {
865                 return Version("Provides SSL support for clients", VF_VENDOR);
866         }
867 };
868
869 MODULE_INIT(ModuleSSLOpenSSL)