]> git.netwichtig.de Git - user/henk/code/inspircd.git/blobdiff - src/modules/m_sslinfo.cpp
Fix ParamModeBase::OnUnset() not being virtual.
[user/henk/code/inspircd.git] / src / modules / m_sslinfo.cpp
index 85e7b77dd3f560ddbefdd02d3f232cf0e6a47731..54aeb9755a7c193ffdfa06f13a18244f1d69bc2d 100644 (file)
@@ -19,6 +19,7 @@
 
 #include "inspircd.h"
 #include "modules/ssl.h"
+#include "modules/webirc.h"
 #include "modules/whois.h"
 
 enum
@@ -30,7 +31,8 @@ enum
        RPL_WHOISSECURE = 671
 };
 
-class SSLCertExt : public ExtensionItem {
+class SSLCertExt : public ExtensionItem
+{
  public:
        SSLCertExt(Module* parent)
                : ExtensionItem("ssl_cert", ExtensionItem::EXT_USER, parent)
@@ -41,6 +43,7 @@ class SSLCertExt : public ExtensionItem {
        {
                return static_cast<ssl_cert*>(get_raw(item));
        }
+
        void set(Extensible* item, ssl_cert* value)
        {
                value->refcount_inc();
@@ -49,6 +52,12 @@ class SSLCertExt : public ExtensionItem {
                        delete old;
        }
 
+       void unset(Extensible* container)
+       {
+               void* old = unset_raw(container);
+               delete static_cast<std::string*>(old);
+       }
+
        std::string serialize(SerializeFormat format, const Extensible* container, void* item) const CXX11_OVERRIDE
        {
                return static_cast<ssl_cert*>(item)->GetMetaLine();
@@ -79,7 +88,7 @@ class SSLCertExt : public ExtensionItem {
                }
        }
 
-       void free(void* item) CXX11_OVERRIDE
+       void free(Extensible* container, void* item) CXX11_OVERRIDE
        {
                ssl_cert* old = static_cast<ssl_cert*>(item);
                if (old && old->refcount_dec())
@@ -87,19 +96,58 @@ class SSLCertExt : public ExtensionItem {
        }
 };
 
-/** Handle /SSLINFO
- */
+class UserCertificateAPIImpl : public UserCertificateAPIBase
+{
+ public:
+       LocalIntExt nosslext;
+       SSLCertExt sslext;
+
+       UserCertificateAPIImpl(Module* mod)
+               : UserCertificateAPIBase(mod)
+               , nosslext("no_ssl_cert", ExtensionItem::EXT_USER, mod)
+               , sslext(mod)
+       {
+       }
+
+       ssl_cert* GetCertificate(User* user) CXX11_OVERRIDE
+       {
+               ssl_cert* cert = sslext.get(user);
+               if (cert)
+                       return cert;
+
+               LocalUser* luser = IS_LOCAL(user);
+               if (!luser || nosslext.get(luser))
+                       return NULL;
+
+               cert = SSLClientCert::GetCertificate(&luser->eh);
+               if (!cert)
+                       return NULL;
+
+               SetCertificate(user, cert);
+               return cert;
+       }
+
+       void SetCertificate(User* user, ssl_cert* cert) CXX11_OVERRIDE
+       {
+               ServerInstance->Logs->Log(MODNAME, LOG_DEBUG, "Setting SSL certificate for %s: %s",
+                       user->GetFullHost().c_str(), cert->GetMetaLine().c_str());
+               sslext.set(user, cert);
+       }
+};
+
 class CommandSSLInfo : public Command
 {
  public:
-       SSLCertExt CertExt;
+       UserCertificateAPIImpl sslapi;
 
-       CommandSSLInfo(Module* Creator) : Command(Creator, "SSLINFO", 1), CertExt(Creator)
+       CommandSSLInfo(Module* Creator)
+               : Command(Creator, "SSLINFO", 1)
+               , sslapi(Creator)
        {
                this->syntax = "<nick>";
        }
 
-       CmdResult Handle(const std::vector<std::string>& parameters, User* user) CXX11_OVERRIDE
+       CmdResult Handle(User* user, const Params& parameters) CXX11_OVERRIDE
        {
                User* target = ServerInstance->FindNickOnly(parameters[0]);
 
@@ -114,7 +162,7 @@ class CommandSSLInfo : public Command
                        user->WriteNotice("*** You cannot view SSL certificate information for other users");
                        return CMD_FAILURE;
                }
-               ssl_cert* cert = CertExt.get(target);
+               ssl_cert* cert = sslapi.GetCertificate(target);
                if (!cert)
                {
                        user->WriteNotice("*** No SSL certificate for this user");
@@ -133,32 +181,19 @@ class CommandSSLInfo : public Command
        }
 };
 
-class UserCertificateAPIImpl : public UserCertificateAPIBase
-{
-       SSLCertExt& ext;
-
- public:
-       UserCertificateAPIImpl(Module* mod, SSLCertExt& certext)
-               : UserCertificateAPIBase(mod), ext(certext)
-       {
-       }
-
-       ssl_cert* GetCertificate(User* user) CXX11_OVERRIDE
-       {
-               return ext.get(user);
-       }
-};
-
-class ModuleSSLInfo : public Module, public Whois::EventListener
+class ModuleSSLInfo
+       : public Module
+       , public WebIRC::EventListener
+       , public Whois::EventListener
 {
+ private:
        CommandSSLInfo cmd;
-       UserCertificateAPIImpl APIImpl;
 
  public:
        ModuleSSLInfo()
-               : Whois::EventListener(this)
+               : WebIRC::EventListener(this)
+               , Whois::EventListener(this)
                , cmd(this)
-               , APIImpl(this, cmd.CertExt)
        {
        }
 
@@ -169,7 +204,7 @@ class ModuleSSLInfo : public Module, public Whois::EventListener
 
        void OnWhois(Whois::Context& whois) CXX11_OVERRIDE
        {
-               ssl_cert* cert = cmd.CertExt.get(whois.GetTarget());
+               ssl_cert* cert = cmd.sslapi.GetCertificate(whois.GetTarget());
                if (cert)
                {
                        whois.SendLine(RPL_WHOISSECURE, "is using a secure connection");
@@ -179,7 +214,7 @@ class ModuleSSLInfo : public Module, public Whois::EventListener
                }
        }
 
-       ModResult OnPreCommand(std::string &command, std::vector<std::string> &parameters, LocalUser *user, bool validated, const std::string &original_line) CXX11_OVERRIDE
+       ModResult OnPreCommand(std::string& command, CommandBase::Params& parameters, LocalUser* user, bool validated) CXX11_OVERRIDE
        {
                if ((command == "OPER") && (validated))
                {
@@ -187,7 +222,7 @@ class ModuleSSLInfo : public Module, public Whois::EventListener
                        if (i != ServerInstance->Config->oper_blocks.end())
                        {
                                OperInfo* ifo = i->second;
-                               ssl_cert* cert = cmd.CertExt.get(user);
+                               ssl_cert* cert = cmd.sslapi.GetCertificate(user);
 
                                if (ifo->oper_block->getBool("sslonly") && !cert)
                                {
@@ -210,13 +245,6 @@ class ModuleSSLInfo : public Module, public Whois::EventListener
                return MOD_RES_PASSTHRU;
        }
 
-       void OnUserConnect(LocalUser* user) CXX11_OVERRIDE
-       {
-               ssl_cert* cert = SSLClientCert::GetCertificate(&user->eh);
-               if (cert)
-                       cmd.CertExt.set(user, cert);
-       }
-
        void OnPostConnect(User* user) CXX11_OVERRIDE
        {
                LocalUser* const localuser = IS_LOCAL(user);
@@ -272,6 +300,38 @@ class ModuleSSLInfo : public Module, public Whois::EventListener
                        return MOD_RES_DENY;
                return MOD_RES_PASSTHRU;
        }
+
+       void OnWebIRCAuth(LocalUser* user, const WebIRC::FlagMap* flags) CXX11_OVERRIDE
+       {
+               // We are only interested in connection flags. If none have been
+               // given then we have nothing to do.
+               if (!flags)
+                       return;
+
+               // We only care about the tls connection flag if the connection
+               // between the gateway and the server is secure.
+               if (!cmd.sslapi.GetCertificate(user))
+                       return;
+
+               WebIRC::FlagMap::const_iterator iter = flags->find("secure");
+               if (iter == flags->end())
+               {
+                       // If this is not set then the connection between the client and
+                       // the gateway is not secure.
+                       cmd.sslapi.nosslext.set(user, 1);
+                       cmd.sslapi.sslext.unset(user);
+                       return;
+               }
+
+               // Create a fake ssl_cert for the user.
+               ssl_cert* cert = new ssl_cert;
+               cert->error = "WebIRC users can not specify valid certs yet";
+               cert->invalid = true;
+               cert->revoked = true;
+               cert->trusted = false;
+               cert->unknownsigner = true;
+               cmd.sslapi.SetCertificate(user, cert);
+       }
 };
 
 MODULE_INIT(ModuleSSLInfo)