]> git.netwichtig.de Git - user/henk/code/inspircd.git/blobdiff - src/modules/m_ident.cpp
Replace std::deque with std::vector in spanningtree and related modules
[user/henk/code/inspircd.git] / src / modules / m_ident.cpp
index b715f437cff6fd47d1cfd74f71ff4850a036c53f..80f2c0b996ab847ae31578578b1da65984b08d45 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.
@@ -80,88 +80,50 @@ class IdentRequestSocket : public EventHandler
        std::string result;             /* Holds the ident string if done */
  public:
 
-       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)
        {
                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) */
-               sockaddr* s = new sockaddr[2];
-               sockaddr* addr = new sockaddr[2];
+               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)
                {
-                       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))
                {
@@ -179,32 +141,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). */
-               #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;
-               }
-
                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!
@@ -339,7 +285,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);
        }
@@ -354,7 +300,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);
@@ -370,9 +316,9 @@ class ModuleIdent : public Module
                {
                        std::string hostn = Conf->ReadValue("connect","allow",j);
                        /* XXX: Fixme: does not respect port, limit, etc */
-                       if ((InspIRCd::MatchCIDR(user->GetIPString(),hostn)) || (InspIRCd::Match(user->host,hostn)))
+                       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", j);
+                               bool useident = Conf->ReadFlag("connect", "useident", "yes", j);
 
                                if (!useident)
                                        return 0;
@@ -387,39 +333,16 @@ 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;
        }
 
@@ -471,7 +394,7 @@ 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);