]> git.netwichtig.de Git - user/henk/code/inspircd.git/blobdiff - src/modules/m_ident.cpp
Fix memory leak and invalid vtable location on unload of m_sslinfo
[user/henk/code/inspircd.git] / src / modules / m_ident.cpp
index 04ad7ba4aec9398f36efcc89f278f4d92c6b8b46..690d7b15c58c88389441beef94a0bc94c514dbed 100644 (file)
@@ -79,61 +79,39 @@ class IdentRequestSocket : public EventHandler
        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(InspIRCd *Server, User* u) : user(u), ServerInstance(Server), 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");
 
                done = false;
 
-               /* We allocate two of these because sizeof(sockaddr_in6) > sizeof(sockaddr_in) */
-               irc::sockets::sockaddrs s;
-               irc::sockets::sockaddrs addr;
+               irc::sockets::sockaddrs bindaddr;
+               irc::sockets::sockaddrs connaddr;
+
+               memcpy(&bindaddr, &user->server_sa, sizeof(bindaddr));
+               memcpy(&connaddr, &user->client_sa, sizeof(connaddr));
 
-#ifdef IPV6
-               /* Horrid icky nasty ugly berkely socket crap. */
-               if (v6)
+               if (connaddr.sa.sa_family == AF_INET6)
                {
-                       if (inet_pton(AF_INET6, user->GetIPString(), &addr.in6.sin6_addr) > 0)
-                       {
-                               addr.in6.sin6_family = AF_INET6;
-                               addr.in6.sin6_port = htons(113);
-                               size = sizeof(sockaddr_in6);
-                               inet_pton(AF_INET6, bindip.c_str(), &s.in6.sin6_addr);
-                               s.in6.sin6_family = AF_INET6;
-                               s.in6.sin6_port = 0;
-                       }
+                       bindaddr.in6.sin6_port = 0;
+                       connaddr.in6.sin6_port = htons(113);
                }
                else
-#endif
                {
-                       if (inet_aton(user->GetIPString(), &addr.in4.sin_addr) > 0)
-                       {
-                               addr.in4.sin_family = AF_INET;
-                               addr.in4.sin_port = htons(113);
-                               size = sizeof(sockaddr_in);
-                               inet_aton(bindip.c_str(), &s.in4.sin_addr);
-                               s.in4.sin_family = AF_INET;
-                               s.in4.sin_port = 0;
-                       }
+                       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.sa, size) < 0)
+               if (ServerInstance->SE->Bind(GetFd(), &bindaddr.sa, size) < 0)
                {
                        this->Close();
                        throw ModuleException("failed to bind()");
@@ -142,7 +120,7 @@ class IdentRequestSocket : public EventHandler
                ServerInstance->SE->NonBlocking(GetFd());
 
                /* Attempt connection (nonblocking) */
-               if (ServerInstance->SE->Connect(this, &addr.sa, size) == -1 && errno != EINPROGRESS)
+               if (ServerInstance->SE->Connect(this, &connaddr.sa, size) == -1 && errno != EINPROGRESS)
                {
                        this->Close();
                        throw ModuleException("connect() failed");
@@ -165,30 +143,16 @@ class IdentRequestSocket : public EventHandler
        {
                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). */
-               irc::sockets::sockaddrs laddr, raddr;
-
-               socklen_t laddrsz = sizeof(laddr);
-               socklen_t raddrsz = sizeof(raddr);
-
-               if ((getsockname(user->GetFd(), &laddr.sa, &laddrsz) != 0) || (getpeername(user->GetFd(), &raddr.sa, &raddrsz) != 0))
-               {
-                       done = true;
-                       return;
-               }
-
                char req[32];
 
                /* Build request in the form 'localport,remoteport\r\n' */
                int req_size;
-#ifdef IPV6
-               if (raddr.sa.sa_family == AF_INET6)
-                       req_size = snprintf(req, sizeof(req), "%d,%d\r\n", ntohs(raddr.in6.sin6_port), ntohs(laddr.in6.sin6_port));
+               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
-#endif
-                       req_size = snprintf(req, sizeof(req), "%d,%d\r\n", ntohs(raddr.in4.sin_port), ntohs(laddr.in4.sin_port));
+                       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!
@@ -323,7 +287,7 @@ class ModuleIdent : public Module
        ModuleIdent(InspIRCd *Me) : Module(Me)
        {
                Conf = new ConfigReader(ServerInstance);
-               OnRehash(NULL, "");
+               OnRehash(NULL);
                Implementation eventlist[] = { I_OnRehash, I_OnUserRegister, I_OnCheckReady, I_OnCleanup, I_OnUserDisconnect };
                ServerInstance->Modules->Attach(eventlist, this, 5);
        }
@@ -338,7 +302,7 @@ class ModuleIdent : public Module
                return Version("$Id$", VF_VENDOR, API_VERSION);
        }
 
-       virtual void OnRehash(User *user, const std::string &param)
+       virtual void OnRehash(User *user)
        {
                delete Conf;
                Conf = new ConfigReader(ServerInstance);
@@ -348,7 +312,7 @@ class ModuleIdent : public Module
                        RequestTimeout = 5;
        }
 
-       virtual int OnUserRegister(User *user)
+       virtual ModResult OnUserRegister(User *user)
        {
                for (int j = 0; j < Conf->Enumerate("connect"); j++)
                {
@@ -359,7 +323,7 @@ class ModuleIdent : public Module
                                bool useident = Conf->ReadFlag("connect", "useident", "yes", j);
 
                                if (!useident)
-                                       return 0;
+                                       return MOD_RES_PASSTHRU;
                        }
                }
 
@@ -371,54 +335,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(ServerInstance, user);
+                       user->Extend("ident_socket", 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))
                {
                        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 +377,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 +396,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)