]> git.netwichtig.de Git - user/henk/code/inspircd.git/blob - src/modules/extra/m_ssl_openssl.cpp
m_ssl_openssl Verify DH params being non-NULL before setting it on the context
[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  /* HACK: This prevents OpenSSL on OS X 10.7 and later from spewing deprecation
25   * warnings for every single function call. As far as I (SaberUK) know, Apple
26   * have no plans to remove OpenSSL so this warning just causes needless spam.
27   */
28 #ifdef __APPLE__
29 # define __AVAILABILITYMACROS__
30 # define DEPRECATED_IN_MAC_OS_X_VERSION_10_7_AND_LATER
31 #endif
32  
33 #include "inspircd.h"
34 #include <openssl/ssl.h>
35 #include <openssl/err.h>
36 #include "ssl.h"
37
38 #ifdef _WIN32
39 # pragma comment(lib, "ssleay32.lib")
40 # pragma comment(lib, "libeay32.lib")
41 # undef MAX_DESCRIPTORS
42 # define MAX_DESCRIPTORS 10000
43 #endif
44
45 /* $ModDesc: Provides SSL support for clients */
46
47 /* $LinkerFlags: if("USE_FREEBSD_BASE_SSL") -lssl -lcrypto */
48 /* $CompileFlags: if(!"USE_FREEBSD_BASE_SSL") pkgconfversion("openssl","0.9.7") pkgconfincludes("openssl","/openssl/ssl.h","") */
49 /* $LinkerFlags: if(!"USE_FREEBSD_BASE_SSL") rpath("pkg-config --libs openssl") pkgconflibs("openssl","/libssl.so","-lssl -lcrypto -ldl") */
50
51 /* $NoPedantic */
52
53
54 class ModuleSSLOpenSSL;
55
56 enum issl_status { ISSL_NONE, ISSL_HANDSHAKING, ISSL_OPEN };
57
58 static bool SelfSigned = false;
59
60 #ifdef INSPIRCD_OPENSSL_ENABLE_RENEGO_DETECTION
61 static ModuleSSLOpenSSL* opensslmod = NULL;
62 #endif
63
64 char* get_error()
65 {
66         return ERR_error_string(ERR_get_error(), NULL);
67 }
68
69 static int error_callback(const char *str, size_t len, void *u);
70
71 /** Represents an SSL user's extra data
72  */
73 class issl_session
74 {
75 public:
76         SSL* sess;
77         issl_status status;
78         reference<ssl_cert> cert;
79
80         bool outbound;
81         bool data_to_write;
82
83         issl_session()
84                 : sess(NULL)
85                 , status(ISSL_NONE)
86         {
87                 outbound = false;
88                 data_to_write = false;
89         }
90 };
91
92 static int OnVerify(int preverify_ok, X509_STORE_CTX *ctx)
93 {
94         /* XXX: This will allow self signed certificates.
95          * In the future if we want an option to not allow this,
96          * we can just return preverify_ok here, and openssl
97          * will boot off self-signed and invalid peer certs.
98          */
99         int ve = X509_STORE_CTX_get_error(ctx);
100
101         SelfSigned = (ve == X509_V_ERR_DEPTH_ZERO_SELF_SIGNED_CERT);
102
103         return 1;
104 }
105
106 class ModuleSSLOpenSSL : public Module
107 {
108         issl_session* sessions;
109
110         SSL_CTX* ctx;
111         SSL_CTX* clictx;
112
113         long ctx_options;
114         long clictx_options;
115
116         std::string sslports;
117         bool use_sha;
118
119         ServiceProvider iohook;
120
121         static void SetContextOptions(SSL_CTX* ctx, long defoptions, const std::string& ctxname, ConfigTag* tag)
122         {
123                 long setoptions = tag->getInt(ctxname + "setoptions");
124                 // User-friendly config options for setting context options
125 #ifdef SSL_OP_CIPHER_SERVER_PREFERENCE
126                 if (tag->getBool("cipherserverpref"))
127                         setoptions |= SSL_OP_CIPHER_SERVER_PREFERENCE;
128 #endif
129 #ifdef SSL_OP_NO_COMPRESSION
130                 if (!tag->getBool("compression", true))
131                         setoptions |= SSL_OP_NO_COMPRESSION;
132 #endif
133                 if (!tag->getBool("sslv3", true))
134                         setoptions |= SSL_OP_NO_SSLv3;
135                 if (!tag->getBool("tlsv1", true))
136                         setoptions |= SSL_OP_NO_TLSv1;
137
138                 long clearoptions = tag->getInt(ctxname + "clearoptions");
139                 ServerInstance->Logs->Log("m_ssl_openssl", DEBUG, "Setting OpenSSL %s context options, default: %ld set: %ld clear: %ld", ctxname.c_str(), defoptions, setoptions, clearoptions);
140
141                 // Clear everything
142                 SSL_CTX_clear_options(ctx, SSL_CTX_get_options(ctx));
143
144                 // Set the default options and what is in the conf
145                 SSL_CTX_set_options(ctx, defoptions | setoptions);
146                 long final = SSL_CTX_clear_options(ctx, clearoptions);
147                 ServerInstance->Logs->Log("m_ssl_openssl", DEFAULT, "OpenSSL %s context options: %ld", ctxname.c_str(), final);
148         }
149
150 #ifdef INSPIRCD_OPENSSL_ENABLE_ECDH
151         void SetupECDH(ConfigTag* tag)
152         {
153                 std::string curvename = tag->getString("ecdhcurve", "prime256v1");
154                 if (curvename.empty())
155                         return;
156
157                 int nid = OBJ_sn2nid(curvename.c_str());
158                 if (nid == 0)
159                 {
160                         ServerInstance->Logs->Log("m_ssl_openssl", DEFAULT, "m_ssl_openssl.so: Unknown curve: \"%s\"", curvename.c_str());
161                         return;
162                 }
163
164                 EC_KEY* eckey = EC_KEY_new_by_curve_name(nid);
165                 if (!eckey)
166                 {
167                         ServerInstance->Logs->Log("m_ssl_openssl", DEFAULT, "m_ssl_openssl.so: Unable to create EC key object");
168                         return;
169                 }
170
171                 ERR_clear_error();
172                 if (SSL_CTX_set_tmp_ecdh(ctx, eckey) < 0)
173                 {
174                         ServerInstance->Logs->Log("m_ssl_openssl", DEFAULT, "m_ssl_openssl.so: Couldn't set ECDH parameters");
175                         ERR_print_errors_cb(error_callback, this);
176                 }
177
178                 EC_KEY_free(eckey);
179         }
180 #endif
181
182 #ifdef INSPIRCD_OPENSSL_ENABLE_RENEGO_DETECTION
183         static void SSLInfoCallback(const SSL* ssl, int where, int rc)
184         {
185                 int fd = SSL_get_fd(const_cast<SSL*>(ssl));
186                 issl_session& session = opensslmod->sessions[fd];
187
188                 if ((where & SSL_CB_HANDSHAKE_START) && (session.status == ISSL_OPEN))
189                 {
190                         // The other side is trying to renegotiate, kill the connection and change status
191                         // to ISSL_NONE so CheckRenego() closes the session
192                         session.status = ISSL_NONE;
193                         ServerInstance->SE->Shutdown(fd, 2);
194                 }
195         }
196
197         bool CheckRenego(StreamSocket* sock, issl_session* session)
198         {
199                 if (session->status != ISSL_NONE)
200                         return true;
201
202                 ServerInstance->Logs->Log("m_ssl_openssl", DEBUG, "Session %p killed, attempted to renegotiate", (void*)session->sess);
203                 CloseSession(session);
204                 sock->SetError("Renegotiation is not allowed");
205                 return false;
206         }
207 #endif
208
209  public:
210
211         ModuleSSLOpenSSL() : iohook(this, "ssl/openssl", SERVICE_IOHOOK)
212         {
213 #ifdef INSPIRCD_OPENSSL_ENABLE_RENEGO_DETECTION
214                 opensslmod = this;
215 #endif
216                 sessions = new issl_session[ServerInstance->SE->GetMaxFds()];
217
218                 /* Global SSL library initialization*/
219                 SSL_library_init();
220                 SSL_load_error_strings();
221
222                 /* Build our SSL contexts:
223                  * NOTE: OpenSSL makes us have two contexts, one for servers and one for clients. ICK.
224                  */
225                 ctx = SSL_CTX_new( SSLv23_server_method() );
226                 clictx = SSL_CTX_new( SSLv23_client_method() );
227
228                 SSL_CTX_set_mode(ctx, SSL_MODE_ENABLE_PARTIAL_WRITE | SSL_MODE_ACCEPT_MOVING_WRITE_BUFFER);
229                 SSL_CTX_set_mode(clictx, SSL_MODE_ENABLE_PARTIAL_WRITE | SSL_MODE_ACCEPT_MOVING_WRITE_BUFFER);
230
231                 SSL_CTX_set_verify(ctx, SSL_VERIFY_PEER | SSL_VERIFY_CLIENT_ONCE, OnVerify);
232                 SSL_CTX_set_verify(clictx, SSL_VERIFY_PEER | SSL_VERIFY_CLIENT_ONCE, OnVerify);
233
234                 SSL_CTX_set_session_cache_mode(ctx, SSL_SESS_CACHE_OFF);
235                 SSL_CTX_set_session_cache_mode(clictx, SSL_SESS_CACHE_OFF);
236
237                 long opts = SSL_OP_NO_SSLv2 | SSL_OP_SINGLE_DH_USE;
238                 // Only turn options on if they exist
239 #ifdef SSL_OP_SINGLE_ECDH_USE
240                 opts |= SSL_OP_SINGLE_ECDH_USE;
241 #endif
242 #ifdef SSL_OP_NO_TICKET
243                 opts |= SSL_OP_NO_TICKET;
244 #endif
245
246                 ctx_options = SSL_CTX_set_options(ctx, opts);
247                 clictx_options = SSL_CTX_set_options(clictx, opts);
248         }
249
250         void init()
251         {
252                 // Needs the flag as it ignores a plain /rehash
253                 OnModuleRehash(NULL,"ssl");
254                 Implementation eventlist[] = { I_On005Numeric, I_OnRehash, I_OnModuleRehash, I_OnHookIO, I_OnUserConnect };
255                 ServerInstance->Modules->Attach(eventlist, this, sizeof(eventlist)/sizeof(Implementation));
256                 ServerInstance->Modules->AddService(iohook);
257         }
258
259         void OnHookIO(StreamSocket* user, ListenSocket* lsb)
260         {
261                 if (!user->GetIOHook() && lsb->bind_tag->getString("ssl") == "openssl")
262                 {
263                         /* Hook the user with our module */
264                         user->AddIOHook(this);
265                 }
266         }
267
268         void OnRehash(User* user)
269         {
270                 sslports.clear();
271
272                 ConfigTag* Conf = ServerInstance->Config->ConfValue("openssl");
273
274 #ifdef INSPIRCD_OPENSSL_ENABLE_RENEGO_DETECTION
275                 // Set the callback if we are not allowing renegotiations, unset it if we do
276                 if (Conf->getBool("renegotiation", true))
277                 {
278                         SSL_CTX_set_info_callback(ctx, NULL);
279                         SSL_CTX_set_info_callback(clictx, NULL);
280                 }
281                 else
282                 {
283                         SSL_CTX_set_info_callback(ctx, SSLInfoCallback);
284                         SSL_CTX_set_info_callback(clictx, SSLInfoCallback);
285                 }
286 #endif
287
288                 if (Conf->getBool("showports", true))
289                 {
290                         sslports = Conf->getString("advertisedports");
291                         if (!sslports.empty())
292                                 return;
293
294                         for (size_t i = 0; i < ServerInstance->ports.size(); i++)
295                         {
296                                 ListenSocket* port = ServerInstance->ports[i];
297                                 if (port->bind_tag->getString("ssl") != "openssl")
298                                         continue;
299
300                                 const std::string& portid = port->bind_desc;
301                                 ServerInstance->Logs->Log("m_ssl_openssl", DEFAULT, "m_ssl_openssl.so: Enabling SSL for port %s", portid.c_str());
302
303                                 if (port->bind_tag->getString("type", "clients") == "clients" && port->bind_addr != "127.0.0.1")
304                                 {
305                                         /*
306                                          * Found an SSL port for clients that is not bound to 127.0.0.1 and handled by us, display
307                                          * the IP:port in ISUPPORT.
308                                          *
309                                          * We used to advertise all ports seperated by a ';' char that matched the above criteria,
310                                          * but this resulted in too long ISUPPORT lines if there were lots of ports to be displayed.
311                                          * To solve this by default we now only display the first IP:port found and let the user
312                                          * configure the exact value for the 005 token, if necessary.
313                                          */
314                                         sslports = portid;
315                                         break;
316                                 }
317                         }
318                 }
319         }
320
321         void OnModuleRehash(User* user, const std::string &param)
322         {
323                 if (param != "ssl")
324                         return;
325
326                 std::string keyfile;
327                 std::string certfile;
328                 std::string cafile;
329                 std::string dhfile;
330                 OnRehash(user);
331
332                 ConfigTag* conf = ServerInstance->Config->ConfValue("openssl");
333
334                 cafile   = conf->getString("cafile", CONFIG_PATH "/ca.pem");
335                 certfile = conf->getString("certfile", CONFIG_PATH "/cert.pem");
336                 keyfile  = conf->getString("keyfile", CONFIG_PATH "/key.pem");
337                 dhfile   = conf->getString("dhfile", CONFIG_PATH "/dhparams.pem");
338                 std::string hash = conf->getString("hash", "md5");
339                 if (hash != "sha1" && hash != "md5")
340                         throw ModuleException("Unknown hash type " + hash);
341                 use_sha = (hash == "sha1");
342
343                 if (conf->getBool("customcontextoptions"))
344                 {
345                         SetContextOptions(ctx, ctx_options, "server", conf);
346                         SetContextOptions(clictx, clictx_options, "client", conf);
347                 }
348
349                 std::string ciphers = conf->getString("ciphers", "");
350
351                 if (!ciphers.empty())
352                 {
353                         ERR_clear_error();
354                         if ((!SSL_CTX_set_cipher_list(ctx, ciphers.c_str())) || (!SSL_CTX_set_cipher_list(clictx, ciphers.c_str())))
355                         {
356                                 ServerInstance->Logs->Log("m_ssl_openssl",DEFAULT, "m_ssl_openssl.so: Can't set cipher list to %s.", ciphers.c_str());
357                                 ERR_print_errors_cb(error_callback, this);
358                         }
359                 }
360
361                 /* Load our keys and certificates
362                  * NOTE: OpenSSL's error logging API sucks, don't blame us for this clusterfuck.
363                  */
364                 ERR_clear_error();
365                 if ((!SSL_CTX_use_certificate_chain_file(ctx, certfile.c_str())) || (!SSL_CTX_use_certificate_chain_file(clictx, certfile.c_str())))
366                 {
367                         ServerInstance->Logs->Log("m_ssl_openssl",DEFAULT, "m_ssl_openssl.so: Can't read certificate file %s. %s", certfile.c_str(), strerror(errno));
368                         ERR_print_errors_cb(error_callback, this);
369                 }
370
371                 ERR_clear_error();
372                 if (((!SSL_CTX_use_PrivateKey_file(ctx, keyfile.c_str(), SSL_FILETYPE_PEM))) || (!SSL_CTX_use_PrivateKey_file(clictx, keyfile.c_str(), SSL_FILETYPE_PEM)))
373                 {
374                         ServerInstance->Logs->Log("m_ssl_openssl",DEFAULT, "m_ssl_openssl.so: Can't read key file %s. %s", keyfile.c_str(), strerror(errno));
375                         ERR_print_errors_cb(error_callback, this);
376                 }
377
378                 /* Load the CAs we trust*/
379                 ERR_clear_error();
380                 if (((!SSL_CTX_load_verify_locations(ctx, cafile.c_str(), 0))) || (!SSL_CTX_load_verify_locations(clictx, cafile.c_str(), 0)))
381                 {
382                         ServerInstance->Logs->Log("m_ssl_openssl",DEFAULT, "m_ssl_openssl.so: 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", cafile.c_str(), strerror(errno));
383                         ERR_print_errors_cb(error_callback, this);
384                 }
385
386 #ifdef _WIN32
387                 BIO* dhpfile = BIO_new_file(dhfile.c_str(), "r");
388 #else
389                 FILE* dhpfile = fopen(dhfile.c_str(), "r");
390 #endif
391                 DH* ret;
392
393                 if (dhpfile == NULL)
394                 {
395                         ServerInstance->Logs->Log("m_ssl_openssl",DEFAULT, "m_ssl_openssl.so Couldn't open DH file %s: %s", dhfile.c_str(), strerror(errno));
396                         throw ModuleException("Couldn't open DH file " + dhfile + ": " + strerror(errno));
397                 }
398                 else
399                 {
400 #ifdef _WIN32
401                         ret = PEM_read_bio_DHparams(dhpfile, NULL, NULL, NULL);
402                         BIO_free(dhpfile);
403 #else
404                         ret = PEM_read_DHparams(dhpfile, NULL, NULL, NULL);
405 #endif
406
407                         ERR_clear_error();
408                         if (ret)
409                         {
410                                 if ((SSL_CTX_set_tmp_dh(ctx, ret) < 0) || (SSL_CTX_set_tmp_dh(clictx, ret) < 0))
411                                 {
412                                         ServerInstance->Logs->Log("m_ssl_openssl", DEFAULT, "m_ssl_openssl.so: Couldn't set DH parameters %s. SSL errors follow:", dhfile.c_str());
413                                         ERR_print_errors_cb(error_callback, this);
414                                 }
415                                 DH_free(ret);
416                         }
417                         else
418                         {
419                                 ServerInstance->Logs->Log("m_ssl_openssl", DEFAULT, "m_ssl_openssl.so: Couldn't set DH parameters %s.", dhfile.c_str());
420                         }
421                 }
422
423 #ifndef _WIN32
424                 fclose(dhpfile);
425 #endif
426
427 #ifdef INSPIRCD_OPENSSL_ENABLE_ECDH
428                 SetupECDH(conf);
429 #endif
430         }
431
432         void On005Numeric(std::string &output)
433         {
434                 if (!sslports.empty())
435                         output.append(" SSL=" + sslports);
436         }
437
438         ~ModuleSSLOpenSSL()
439         {
440                 SSL_CTX_free(ctx);
441                 SSL_CTX_free(clictx);
442                 delete[] sessions;
443         }
444
445         void OnUserConnect(LocalUser* user)
446         {
447                 if (user->eh.GetIOHook() == this)
448                 {
449                         if (sessions[user->eh.GetFd()].sess)
450                         {
451                                 if (!sessions[user->eh.GetFd()].cert->fingerprint.empty())
452                                         user->WriteServ("NOTICE %s :*** You are connected using SSL cipher \"%s\""
453                                                 " and your SSL fingerprint is %s", user->nick.c_str(), SSL_get_cipher(sessions[user->eh.GetFd()].sess), sessions[user->eh.GetFd()].cert->fingerprint.c_str());
454                                 else
455                                         user->WriteServ("NOTICE %s :*** You are connected using SSL cipher \"%s\"", user->nick.c_str(), SSL_get_cipher(sessions[user->eh.GetFd()].sess));
456                         }
457                 }
458         }
459
460         void OnCleanup(int target_type, void* item)
461         {
462                 if (target_type == TYPE_USER)
463                 {
464                         LocalUser* user = IS_LOCAL((User*)item);
465
466                         if (user && user->eh.GetIOHook() == this)
467                         {
468                                 // User is using SSL, they're a local user, and they're using one of *our* SSL ports.
469                                 // Potentially there could be multiple SSL modules loaded at once on different ports.
470                                 ServerInstance->Users->QuitUser(user, "SSL module unloading");
471                         }
472                 }
473         }
474
475         Version GetVersion()
476         {
477                 return Version("Provides SSL support for clients", VF_VENDOR);
478         }
479
480         void OnRequest(Request& request)
481         {
482                 if (strcmp("GET_SSL_CERT", request.id) == 0)
483                 {
484                         SocketCertificateRequest& req = static_cast<SocketCertificateRequest&>(request);
485                         int fd = req.sock->GetFd();
486                         issl_session* session = &sessions[fd];
487
488                         req.cert = session->cert;
489                 }
490                 else if (!strcmp("GET_RAW_SSL_SESSION", request.id))
491                 {
492                         SSLRawSessionRequest& req = static_cast<SSLRawSessionRequest&>(request);
493                         if ((req.fd >= 0) && (req.fd < ServerInstance->SE->GetMaxFds()))
494                                 req.data = reinterpret_cast<void*>(sessions[req.fd].sess);
495                 }
496         }
497
498         void OnStreamSocketAccept(StreamSocket* user, irc::sockets::sockaddrs* client, irc::sockets::sockaddrs* server)
499         {
500                 int fd = user->GetFd();
501
502                 issl_session* session = &sessions[fd];
503
504                 session->sess = SSL_new(ctx);
505                 session->status = ISSL_NONE;
506                 session->outbound = false;
507                 session->data_to_write = false;
508
509                 if (session->sess == NULL)
510                         return;
511
512                 if (SSL_set_fd(session->sess, fd) == 0)
513                 {
514                         ServerInstance->Logs->Log("m_ssl_openssl",DEBUG,"BUG: Can't set fd with SSL_set_fd: %d", fd);
515                         return;
516                 }
517
518                 Handshake(user, session);
519         }
520
521         void OnStreamSocketConnect(StreamSocket* user)
522         {
523                 int fd = user->GetFd();
524                 /* Are there any possibilities of an out of range fd? Hope not, but lets be paranoid */
525                 if ((fd < 0) || (fd > ServerInstance->SE->GetMaxFds() -1))
526                         return;
527
528                 issl_session* session = &sessions[fd];
529
530                 session->sess = SSL_new(clictx);
531                 session->status = ISSL_NONE;
532                 session->outbound = true;
533                 session->data_to_write = false;
534
535                 if (session->sess == NULL)
536                         return;
537
538                 if (SSL_set_fd(session->sess, fd) == 0)
539                 {
540                         ServerInstance->Logs->Log("m_ssl_openssl",DEBUG,"BUG: Can't set fd with SSL_set_fd: %d", fd);
541                         return;
542                 }
543
544                 Handshake(user, session);
545         }
546
547         void OnStreamSocketClose(StreamSocket* user)
548         {
549                 int fd = user->GetFd();
550                 /* Are there any possibilities of an out of range fd? Hope not, but lets be paranoid */
551                 if ((fd < 0) || (fd > ServerInstance->SE->GetMaxFds() - 1))
552                         return;
553
554                 CloseSession(&sessions[fd]);
555         }
556
557         int OnStreamSocketRead(StreamSocket* user, std::string& recvq)
558         {
559                 int fd = user->GetFd();
560                 /* Are there any possibilities of an out of range fd? Hope not, but lets be paranoid */
561                 if ((fd < 0) || (fd > ServerInstance->SE->GetMaxFds() - 1))
562                         return -1;
563
564                 issl_session* session = &sessions[fd];
565
566                 if (!session->sess)
567                 {
568                         CloseSession(session);
569                         return -1;
570                 }
571
572                 if (session->status == ISSL_HANDSHAKING)
573                 {
574                         // The handshake isn't finished and it wants to read, try to finish it.
575                         if (!Handshake(user, session))
576                         {
577                                 // Couldn't resume handshake.
578                                 if (session->status == ISSL_NONE)
579                                         return -1;
580                                 return 0;
581                         }
582                 }
583
584                 // If we resumed the handshake then session->status will be ISSL_OPEN
585
586                 if (session->status == ISSL_OPEN)
587                 {
588                         ERR_clear_error();
589                         char* buffer = ServerInstance->GetReadBuffer();
590                         size_t bufsiz = ServerInstance->Config->NetBufferSize;
591                         int ret = SSL_read(session->sess, buffer, bufsiz);
592
593 #ifdef INSPIRCD_OPENSSL_ENABLE_RENEGO_DETECTION
594                         if (!CheckRenego(user, session))
595                                 return -1;
596 #endif
597
598                         if (ret > 0)
599                         {
600                                 recvq.append(buffer, ret);
601
602                                 int mask = 0;
603                                 // Schedule a read if there is still data in the OpenSSL buffer
604                                 if (SSL_pending(session->sess) > 0)
605                                         mask |= FD_ADD_TRIAL_READ;
606                                 if (session->data_to_write)
607                                         mask |= FD_WANT_POLL_READ | FD_WANT_SINGLE_WRITE;
608                                 if (mask != 0)
609                                         ServerInstance->SE->ChangeEventMask(user, mask);
610                                 return 1;
611                         }
612                         else if (ret == 0)
613                         {
614                                 // Client closed connection.
615                                 CloseSession(session);
616                                 user->SetError("Connection closed");
617                                 return -1;
618                         }
619                         else if (ret < 0)
620                         {
621                                 int err = SSL_get_error(session->sess, ret);
622
623                                 if (err == SSL_ERROR_WANT_READ)
624                                 {
625                                         ServerInstance->SE->ChangeEventMask(user, FD_WANT_POLL_READ);
626                                         return 0;
627                                 }
628                                 else if (err == SSL_ERROR_WANT_WRITE)
629                                 {
630                                         ServerInstance->SE->ChangeEventMask(user, FD_WANT_NO_READ | FD_WANT_SINGLE_WRITE);
631                                         return 0;
632                                 }
633                                 else
634                                 {
635                                         CloseSession(session);
636                                         return -1;
637                                 }
638                         }
639                 }
640
641                 return 0;
642         }
643
644         int OnStreamSocketWrite(StreamSocket* user, std::string& buffer)
645         {
646                 int fd = user->GetFd();
647
648                 issl_session* session = &sessions[fd];
649
650                 if (!session->sess)
651                 {
652                         CloseSession(session);
653                         return -1;
654                 }
655
656                 session->data_to_write = true;
657
658                 if (session->status == ISSL_HANDSHAKING)
659                 {
660                         if (!Handshake(user, session))
661                         {
662                                 // Couldn't resume handshake.
663                                 if (session->status == ISSL_NONE)
664                                         return -1;
665                                 return 0;
666                         }
667                 }
668
669                 if (session->status == ISSL_OPEN)
670                 {
671                         ERR_clear_error();
672                         int ret = SSL_write(session->sess, buffer.data(), buffer.size());
673
674 #ifdef INSPIRCD_OPENSSL_ENABLE_RENEGO_DETECTION
675                         if (!CheckRenego(user, session))
676                                 return -1;
677 #endif
678
679                         if (ret == (int)buffer.length())
680                         {
681                                 session->data_to_write = false;
682                                 ServerInstance->SE->ChangeEventMask(user, FD_WANT_POLL_READ | FD_WANT_NO_WRITE);
683                                 return 1;
684                         }
685                         else if (ret > 0)
686                         {
687                                 buffer = buffer.substr(ret);
688                                 ServerInstance->SE->ChangeEventMask(user, FD_WANT_SINGLE_WRITE);
689                                 return 0;
690                         }
691                         else if (ret == 0)
692                         {
693                                 CloseSession(session);
694                                 return -1;
695                         }
696                         else if (ret < 0)
697                         {
698                                 int err = SSL_get_error(session->sess, ret);
699
700                                 if (err == SSL_ERROR_WANT_WRITE)
701                                 {
702                                         ServerInstance->SE->ChangeEventMask(user, FD_WANT_SINGLE_WRITE);
703                                         return 0;
704                                 }
705                                 else if (err == SSL_ERROR_WANT_READ)
706                                 {
707                                         ServerInstance->SE->ChangeEventMask(user, FD_WANT_POLL_READ);
708                                         return 0;
709                                 }
710                                 else
711                                 {
712                                         CloseSession(session);
713                                         return -1;
714                                 }
715                         }
716                 }
717                 return 0;
718         }
719
720         bool Handshake(StreamSocket* user, issl_session* session)
721         {
722                 int ret;
723
724                 ERR_clear_error();
725                 if (session->outbound)
726                         ret = SSL_connect(session->sess);
727                 else
728                         ret = SSL_accept(session->sess);
729
730                 if (ret < 0)
731                 {
732                         int err = SSL_get_error(session->sess, ret);
733
734                         if (err == SSL_ERROR_WANT_READ)
735                         {
736                                 ServerInstance->SE->ChangeEventMask(user, FD_WANT_POLL_READ | FD_WANT_NO_WRITE);
737                                 session->status = ISSL_HANDSHAKING;
738                                 return true;
739                         }
740                         else if (err == SSL_ERROR_WANT_WRITE)
741                         {
742                                 ServerInstance->SE->ChangeEventMask(user, FD_WANT_NO_READ | FD_WANT_SINGLE_WRITE);
743                                 session->status = ISSL_HANDSHAKING;
744                                 return true;
745                         }
746                         else
747                         {
748                                 CloseSession(session);
749                         }
750
751                         return false;
752                 }
753                 else if (ret > 0)
754                 {
755                         // Handshake complete.
756                         VerifyCertificate(session, user);
757
758                         session->status = ISSL_OPEN;
759
760                         ServerInstance->SE->ChangeEventMask(user, FD_WANT_POLL_READ | FD_WANT_NO_WRITE | FD_ADD_TRIAL_WRITE);
761
762                         return true;
763                 }
764                 else if (ret == 0)
765                 {
766                         CloseSession(session);
767                 }
768                 return false;
769         }
770
771         void CloseSession(issl_session* session)
772         {
773                 if (session->sess)
774                 {
775                         SSL_shutdown(session->sess);
776                         SSL_free(session->sess);
777                 }
778
779                 session->sess = NULL;
780                 session->status = ISSL_NONE;
781                 session->cert = NULL;
782         }
783
784         void VerifyCertificate(issl_session* session, StreamSocket* user)
785         {
786                 if (!session->sess || !user)
787                         return;
788
789                 X509* cert;
790                 ssl_cert* certinfo = new ssl_cert;
791                 session->cert = certinfo;
792                 unsigned int n;
793                 unsigned char md[EVP_MAX_MD_SIZE];
794                 const EVP_MD *digest = use_sha ? EVP_sha1() : EVP_md5();
795
796                 cert = SSL_get_peer_certificate((SSL*)session->sess);
797
798                 if (!cert)
799                 {
800                         certinfo->error = "Could not get peer certificate: "+std::string(get_error());
801                         return;
802                 }
803
804                 certinfo->invalid = (SSL_get_verify_result(session->sess) != X509_V_OK);
805
806                 if (!SelfSigned)
807                 {
808                         certinfo->unknownsigner = false;
809                         certinfo->trusted = true;
810                 }
811                 else
812                 {
813                         certinfo->unknownsigner = true;
814                         certinfo->trusted = false;
815                 }
816
817                 char buf[512];
818                 X509_NAME_oneline(X509_get_subject_name(cert), buf, sizeof(buf));
819                 certinfo->dn = buf;
820                 // Make sure there are no chars in the string that we consider invalid
821                 if (certinfo->dn.find_first_of("\r\n") != std::string::npos)
822                         certinfo->dn.clear();
823
824                 X509_NAME_oneline(X509_get_issuer_name(cert), buf, sizeof(buf));
825                 certinfo->issuer = buf;
826                 if (certinfo->issuer.find_first_of("\r\n") != std::string::npos)
827                         certinfo->issuer.clear();
828
829                 if (!X509_digest(cert, digest, md, &n))
830                 {
831                         certinfo->error = "Out of memory generating fingerprint";
832                 }
833                 else
834                 {
835                         certinfo->fingerprint = irc::hex(md, n);
836                 }
837
838                 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))
839                 {
840                         certinfo->error = "Not activated, or expired certificate";
841                 }
842
843                 X509_free(cert);
844         }
845 };
846
847 static int error_callback(const char *str, size_t len, void *u)
848 {
849         ServerInstance->Logs->Log("m_ssl_openssl",DEFAULT, "SSL error: " + std::string(str, len - 1));
850
851         //
852         // XXX: Remove this line, it causes valgrind warnings...
853         //
854         // MD_update(&m, buf, j);
855         //
856         //
857         // ... ONLY JOKING! :-)
858         //
859
860         return 0;
861 }
862
863 MODULE_INIT(ModuleSSLOpenSSL)