]> git.netwichtig.de Git - user/henk/code/inspircd.git/blobdiff - src/modules/m_ident.cpp
Remove InspIRCd* parameters and fields
[user/henk/code/inspircd.git] / src / modules / m_ident.cpp
index 3eded4daf59e2eb7a21eb59076d8c6647a05bfb3..d74873e00770d9aaeb3b8c93552f5a7babff61ff 100644 (file)
@@ -2,8 +2,8 @@
  *       | Inspire Internet Relay Chat Daemon |
  *       +------------------------------------+
  *
- *  InspIRCd: (C) 2002-2008 InspIRCd Development Team
- * See: http://www.inspircd.org/wiki/index.php/Credits
+ *  InspIRCd: (C) 2002-2009 InspIRCd Development Team
+ * See: http://wiki.inspircd.org/Credits
  *
  * This program is free but copyrighted software; see
  *         the file COPYING for details.
@@ -12,9 +12,6 @@
  */
 
 #include "inspircd.h"
-#include "users.h"
-#include "channels.h"
-#include "modules.h"
 
 /* $ModDesc: Provides support for RFC1413 ident lookups */
 
 class IdentRequestSocket : public EventHandler
 {
  private:
-
-        User *user;                    /* User we are attached to */
-        InspIRCd* ServerInstance;      /* Server instance */
-        bool done;                     /* True if lookup is finished */
-        std::string result;            /* Holds the ident string if done */
-
+       User *user;                     /* User we are attached to */
+       bool done;                      /* True if lookup is finished */
+       std::string result;             /* Holds the ident string if done */
  public:
+       time_t age;
 
-       IdentRequestSocket(InspIRCd *Server, User* u, const std::string &bindip) : user(u), ServerInstance(Server), result(u->ident)
+       IdentRequestSocket(User* u) : user(u), result(u->ident)
        {
+               age = ServerInstance->Time();
                socklen_t size = 0;
-#ifdef IPV6
-               /* Does this look like a v6 ip address? */
-               bool v6 = false;
-               if ((bindip.empty()) || bindip.find(':') != std::string::npos)
-               v6 = true;
-
-               if (v6)
-                       SetFd(socket(AF_INET6, SOCK_STREAM, 0));
-               else
-#endif
-                       SetFd(socket(AF_INET, SOCK_STREAM, 0));
+
+               SetFd(socket(user->server_sa.sa.sa_family, SOCK_STREAM, 0));
 
                if (GetFd() == -1)
                        throw ModuleException("Could not create socket");
 
-               /* We allocate two of these because sizeof(sockaddr_in6) > sizeof(sockaddr_in) */
-               sockaddr* s = new sockaddr[2];
-               sockaddr* addr = new sockaddr[2];
+               done = false;
+
+               irc::sockets::sockaddrs bindaddr;
+               irc::sockets::sockaddrs connaddr;
 
-#ifdef IPV6
-               /* Horrid icky nasty ugly berkely socket crap. */
-               if (v6)
+               memcpy(&bindaddr, &user->server_sa, sizeof(bindaddr));
+               memcpy(&connaddr, &user->client_sa, sizeof(connaddr));
+
+               if (connaddr.sa.sa_family == AF_INET6)
                {
-                       in6_addr addy;
-                       in6_addr n;
-                       if (inet_pton(AF_INET6, user->GetIPString(), &addy) > 0)
-                       {
-                               ((sockaddr_in6*)addr)->sin6_family = AF_INET6;
-                               memcpy(&((sockaddr_in6*)addr)->sin6_addr, &addy, sizeof(addy));
-                               ((sockaddr_in6*)addr)->sin6_port = htons(113);
-                               size = sizeof(sockaddr_in6);
-                               inet_pton(AF_INET6, bindip.c_str(), &n);
-                               memcpy(&((sockaddr_in6*)s)->sin6_addr, &n, sizeof(sockaddr_in6));
-                               ((sockaddr_in6*)s)->sin6_port = 0;
-                               ((sockaddr_in6*)s)->sin6_family = AF_INET6;
-                       }
+                       bindaddr.in6.sin6_port = 0;
+                       connaddr.in6.sin6_port = htons(113);
                }
                else
-#endif
                {
-                       in_addr addy;
-                       in_addr n;
-                       if (inet_aton(user->GetIPString(), &addy) > 0)
-                       {
-                               ((sockaddr_in*)addr)->sin_family = AF_INET;
-                               ((sockaddr_in*)addr)->sin_addr = addy;
-                               ((sockaddr_in*)addr)->sin_port = htons(113);
-                               size = sizeof(sockaddr_in);
-                               inet_aton(bindip.c_str(), &n);
-                               ((sockaddr_in*)s)->sin_addr = n;
-                               ((sockaddr_in*)s)->sin_port = 0;
-                               ((sockaddr_in*)s)->sin_family = AF_INET;
-                       }
+                       bindaddr.in4.sin_port = 0;
+                       connaddr.in4.sin_port = htons(113);
                }
 
                /* Attempt to bind (ident requests must come from the ip the query is referring to */
-               if (ServerInstance->SE->Bind(GetFd(), s, size) < 0)
+               if (ServerInstance->SE->Bind(GetFd(), &bindaddr.sa, size) < 0)
                {
                        this->Close();
-                       delete[] s;
-                       delete[] addr;
                        throw ModuleException("failed to bind()");
                }
 
-               delete[] s;
                ServerInstance->SE->NonBlocking(GetFd());
 
                /* Attempt connection (nonblocking) */
-               if (ServerInstance->SE->Connect(this, (sockaddr*)addr, size) == -1 && errno != EINPROGRESS)
+               if (ServerInstance->SE->Connect(this, &connaddr.sa, size) == -1 && errno != EINPROGRESS)
                {
                        this->Close();
-                       delete[] addr;
                        throw ModuleException("connect() failed");
                }
 
-               delete[] addr;
-
                /* Add fd to socket engine */
-               if (!ServerInstance->SE->AddFd(this))
+               if (!ServerInstance->SE->AddFd(this, FD_WANT_NO_READ | FD_WANT_POLL_WRITE))
                {
                        this->Close();
                        throw ModuleException("out of fds");
                }
-
-               /* Important: We set WantWrite immediately after connect()
-                * because a successful connection will trigger a writability event
-                */
-               ServerInstance->SE->WantWrite(this);
        }
 
        virtual void OnConnected()
        {
                ServerInstance->Logs->Log("m_ident",DEBUG,"OnConnected()");
-
-               /* Both sockaddr_in and sockaddr_in6 can be safely casted to sockaddr, especially since the
-                * only members we use are in a part of the struct that should always be identical (at the
-                * byte level). */
-               #ifndef IPV6
-               sockaddr_in laddr, raddr;
-               #else
-               sockaddr_in6 laddr, raddr;
-               #endif
-
-               socklen_t laddrsz = sizeof(laddr);
-               socklen_t raddrsz = sizeof(raddr);
-
-               if ((getsockname(user->GetFd(), (sockaddr*) &laddr, &laddrsz) != 0) || (getpeername(user->GetFd(), (sockaddr*) &raddr, &raddrsz) != 0))
-               {
-                       done = true;
-                       return;
-               }
+               ServerInstance->SE->ChangeEventMask(this, FD_WANT_POLL_READ | FD_WANT_NO_WRITE);
 
                char req[32];
 
                /* Build request in the form 'localport,remoteport\r\n' */
-               #ifndef IPV6
-               int req_size = snprintf(req, sizeof(req), "%d,%d\r\n", ntohs(raddr.sin_port), ntohs(laddr.sin_port));
-               #else
-               int req_size = snprintf(req, sizeof(req), "%d,%d\r\n", ntohs(raddr.sin6_port), ntohs(laddr.sin6_port));
-               #endif
+               int req_size;
+               if (user->client_sa.sa.sa_family == AF_INET6)
+                       req_size = snprintf(req, sizeof(req), "%d,%d\r\n",
+                               ntohs(user->client_sa.in6.sin6_port), ntohs(user->server_sa.in6.sin6_port));
+               else
+                       req_size = snprintf(req, sizeof(req), "%d,%d\r\n",
+                               ntohs(user->client_sa.in4.sin_port), ntohs(user->server_sa.in4.sin_port));
 
                /* Send failed if we didnt write the whole ident request --
                 * might as well give up if this happens!
@@ -335,34 +275,53 @@ class IdentRequestSocket : public EventHandler
 
 class ModuleIdent : public Module
 {
- private:
        int RequestTimeout;
+       ConfigReader *Conf;
+       SimpleExtItem<IdentRequestSocket> ext;
  public:
-       ModuleIdent(InspIRCd *Me)
-               : Module(Me)
+       ModuleIdent() : ext("ident_socket", this)
        {
-               OnRehash(NULL, "");
+               Conf = new ConfigReader;
+               OnRehash(NULL);
                Implementation eventlist[] = { I_OnRehash, I_OnUserRegister, I_OnCheckReady, I_OnCleanup, I_OnUserDisconnect };
                ServerInstance->Modules->Attach(eventlist, this, 5);
        }
 
-       virtual Version GetVersion()
+       ~ModuleIdent()
        {
-               return Version(1, 2, 1, 0, VF_VENDOR, API_VERSION);
+               delete Conf;
        }
 
+       virtual Version GetVersion()
+       {
+               return Version("Provides support for RFC1413 ident lookups", VF_VENDOR, API_VERSION);
+       }
 
-       virtual void OnRehash(User *user, const std::string &param)
+       virtual void OnRehash(User *user)
        {
-               ConfigReader MyConf(ServerInstance);
+               delete Conf;
+               Conf = new ConfigReader;
 
-               RequestTimeout = MyConf.ReadInteger("ident", "timeout", 0, true);
+               RequestTimeout = Conf->ReadInteger("ident", "timeout", 0, true);
                if (!RequestTimeout)
                        RequestTimeout = 5;
        }
 
-       virtual int OnUserRegister(User *user)
+       virtual ModResult OnUserRegister(User *user)
        {
+               for (int j = 0; j < Conf->Enumerate("connect"); j++)
+               {
+                       std::string hostn = Conf->ReadValue("connect","allow",j);
+                       /* XXX: Fixme: does not respect port, limit, etc */
+                       if ((InspIRCd::MatchCIDR(user->GetIPString(),hostn, ascii_case_insensitive_map)) || (InspIRCd::Match(user->host,hostn, ascii_case_insensitive_map)))
+                       {
+                               bool useident = Conf->ReadFlag("connect", "useident", "yes", j);
+
+                               if (!useident)
+                                       return MOD_RES_PASSTHRU;
+                       }
+               }
+
                /* User::ident is currently the username field from USER; with m_ident loaded, that
                 * should be preceded by a ~. The field is actually IdentMax+2 characters wide. */
                if (user->ident.length() > ServerInstance->Config->Limits.IdentMax + 1)
@@ -371,54 +330,31 @@ class ModuleIdent : public Module
 
                user->WriteServ("NOTICE Auth :*** Looking up your ident...");
 
-               // Get the IP that the user is connected to, and bind to that for the outgoing connection
-               #ifndef IPV6
-               sockaddr_in laddr;
-               #else
-               sockaddr_in6 laddr;
-               #endif
-               socklen_t laddrsz = sizeof(laddr);
-
-               if (getsockname(user->GetFd(), (sockaddr*) &laddr, &laddrsz) != 0)
-               {
-                       user->WriteServ("NOTICE Auth :*** Could not find your ident, using %s instead.", user->ident.c_str());
-                       return 0;
-               }
-
-               #ifndef IPV6
-               const char *ip = inet_ntoa(laddr.sin_addr);
-               #else
-               char ip[INET6_ADDRSTRLEN + 1];
-               inet_ntop(laddr.sin6_family, &laddr.sin6_addr, ip, INET6_ADDRSTRLEN);
-               #endif
-
-               IdentRequestSocket *isock = NULL;
                try
                {
-                       isock = new IdentRequestSocket(ServerInstance, user, ip);
+                       IdentRequestSocket *isock = new IdentRequestSocket(user);
+                       ext.set(user, isock);
                }
                catch (ModuleException &e)
                {
                        ServerInstance->Logs->Log("m_ident",DEBUG,"Ident exception: %s", e.GetReason());
-                       return 0;
                }
 
-               user->Extend("ident_socket", isock);
-               return 0;
+               return MOD_RES_PASSTHRU;
        }
 
        /* This triggers pretty regularly, we can use it in preference to
         * creating a Timer object and especially better than creating a
         * Timer per ident lookup!
         */
-       virtual bool OnCheckReady(User *user)
+       virtual ModResult OnCheckReady(User *user)
        {
                /* Does user have an ident socket attached at all? */
-               IdentRequestSocket *isock = NULL;
-               if (!user->GetExt("ident_socket", isock))
+               IdentRequestSocket *isock = ext.get(user);
+               if (!isock)
                {
                        ServerInstance->Logs->Log("m_ident",DEBUG, "No ident socket :(");
-                       return true;
+                       return MOD_RES_PASSTHRU;
                }
 
                ServerInstance->Logs->Log("m_ident",DEBUG, "Has ident_socket");
@@ -436,14 +372,14 @@ class ModuleIdent : public Module
                         * we call this to clean up the user
                         */
                        OnUserDisconnect(user);
-                       return true;
+                       return MOD_RES_PASSTHRU;
                }
 
                /* Got a result yet? */
                if (!isock->HasResult())
                {
                        ServerInstance->Logs->Log("m_ident",DEBUG, "No result yet");
-                       return false;
+                       return MOD_RES_DENY;
                }
 
                ServerInstance->Logs->Log("m_ident",DEBUG, "Yay, result!");
@@ -455,11 +391,11 @@ class ModuleIdent : public Module
                        user->WriteServ("NOTICE Auth :*** Could not find your ident, using %s instead.", isock->GetResult());
 
                /* Copy the ident string to the user */
-               user->ident.assign(isock->GetResult(), 0, ServerInstance->Config->Limits.IdentMax + 1);
+               user->ChangeIdent(isock->GetResult());
 
                /* The user isnt actually disconnecting, we call this to clean up the user */
                OnUserDisconnect(user);
-               return true;
+               return MOD_RES_PASSTHRU;
        }
 
        virtual void OnCleanup(int target_type, void *item)
@@ -472,12 +408,11 @@ class ModuleIdent : public Module
        virtual void OnUserDisconnect(User *user)
        {
                /* User disconnect (generic socket detatch event) */
-               IdentRequestSocket *isock = NULL;
-               if (user->GetExt("ident_socket", isock))
+               IdentRequestSocket *isock = ext.get(user);
+               if (isock)
                {
                        isock->Close();
-                       delete isock;
-                       user->Shrink("ident_socket");
+                       ext.unset(user);
                }
        }
 };