Fix test client error cheecking on result types

[user/henk/code/inspircd.git] / src / dnsqueue.cpp

diff --git a/src/dnsqueue.cpp b/src/dnsqueue.cpp
index 975884796f7dd24f9963c58dfb98cc77b9257077..fb0e0bd75999900c96170afed78f019b5253c63f 100644 (file)
--- a/src/dnsqueue.cpp
+++ b/src/dnsqueue.cpp
@@ -38,16 +38,19 @@ extern InspIRCd* ServerInstance;
 class Lookup;
 
 Lookup* dnslist[MAX_DESCRIPTORS];
+Lookup* user_fd_to_dns[MAX_DESCRIPTORS];
+extern userrec* fd_ref_table[MAX_DESCRIPTORS];
 
 //enum LookupState { reverse, forward };
 
 class Lookup {
 private:
-       DNS resolver1;
-       DNS resolver2;
        char u[NICKMAX];
        std::string hostname;
 public:
+       DNS resolver1;
+       DNS resolver2;
+
        Lookup()
        {
                *u = 0;
@@ -71,7 +74,7 @@ public:
                if (usr)
                {
                        resolver1.SetNS(std::string(Config->DNSServer));
-                       if (!resolver1.ReverseLookup(std::string(usr->host)))
+                       if (!resolver1.ReverseLookup(std::string(usr->host), true))
                        {
                                return false;
                        }
@@ -81,6 +84,7 @@ public:
                        if (resolver1.GetFD() != -1)
                        {
                                dnslist[resolver1.GetFD()] = this;
+                               user_fd_to_dns[usr->fd] = this;
                                return true;
                        }
                }
@@ -109,17 +113,18 @@ public:
                                                }
                                                if ((hostname != "") && (usr->registered != 7))
                                                {
-                                                       if ((std::string((char*)inet_ntoa(usr->ip4)) == ip) && (hostname.length() < 65))
+                                                       if ((std::string(inet_ntoa(usr->ip4)) == ip) && (hostname.length() < 65))
                                                        {
-                                                               strlcpy(usr->host,hostname.c_str(),64);
-                                                               strlcpy(usr->dhost,hostname.c_str(),64);
-                                                               /*address_cache::iterator address = addrcache.find(usr->ip4);
-                                                               if (address == addrcache.end())
+                                                               if ((hostname.find_last_of(".in-addr.arpa") == hostname.length() - 1) && (hostname.find_last_of(".in-addr.arpa") != std::string::npos))
                                                                {
-                                                                       log(DEBUG,"Caching hostname %s -> %s",(char*)inet_ntoa(usr->ip4),hostname.c_str());
-                                                                       addrcache[usr->ip4] = new std::string(hostname);
-                                                               }*/
-                                                               WriteServ(usr->fd,"NOTICE Auth :*** Found your hostname");
+                                                                       WriteServ(usr->fd,"NOTICE Auth :*** Your ISP are muppets -- reverse resolution resolves back to same reverse .arpa domain (!)");
+                                                               }
+                                                               else
+                                                               {
+                                                                       strlcpy(usr->host,hostname.c_str(),64);
+                                                                       strlcpy(usr->dhost,hostname.c_str(),64);
+                                                                       WriteServ(usr->fd,"NOTICE Auth :*** Found your hostname");
+                                                               }
                                                        }
                                                        usr->dns_done = true;
                                                        return true;
@@ -148,7 +153,10 @@ public:
                                if ((usr) && (usr->dns_done))
                                {
                                        if (resolver1.GetFD() != -1)
+                                       {
                                                dnslist[resolver1.GetFD()] = NULL;
+                                               user_fd_to_dns[usr->fd] = NULL;
+                                       }
                                        return true;
                                }
                                if (resolver1.GetFD() != -1)
@@ -157,18 +165,23 @@ public:
                                        hostname = resolver1.GetResult();
                                        if (usr)
                                        {
+                                               user_fd_to_dns[usr->fd] = NULL;
                                                if ((usr->registered > 3) || (hostname == ""))
                                                {
                                                        WriteServ(usr->fd,"NOTICE Auth :*** Could not resolve your hostname -- Using your IP address instead");
                                                        usr->dns_done = true;
                                                        return true;
                                                }
-                                       }
-                                       if (hostname != "")
-                                       {
-                                               resolver2.ForwardLookup(hostname);
-                                               if (resolver2.GetFD() != -1)
-                                                       dnslist[resolver2.GetFD()] = this;
+                                               if (hostname != "")
+                                               {
+                                                       resolver2.ForwardLookup(hostname, true);
+                                                       if (resolver2.GetFD() != -1)
+                                                       {
+                                                               dnslist[resolver2.GetFD()] = this;
+                                                               if (usr)
+                                                                       user_fd_to_dns[usr->fd] = this;
+                                                       }
+                                               }
                                        }
                                }
                        }
@@ -225,6 +238,43 @@ bool lookup_dns(const std::string &nick)
        return false;
 }
 
+void ZapThisDns(int fd)
+{
+#ifdef THREADED_DNS
+/*     if (fd_ref_table[fd])
+       {
+                       if (fd_ref_table[fd]->registered >= 3)
+               {
+                       log(DEBUG,"Joining thread for user %d",fd);
+                       if (pthread_join(fd_ref_table[fd]->dnsthread, NULL))
+                       {
+                               log(DEBUG,"Can't pthread_join(): %s", strerror(errno));
+                       }
+               }
+       }*/
+#else
+       if ((fd < 0) || (fd > MAX_DESCRIPTORS))
+               return;
+
+       Lookup *x = user_fd_to_dns[fd];
+
+       if (x)
+       {
+               if (x->resolver1.GetFD() > 0)
+               {
+                       log(DEBUG,"Whacked resolver1");
+                       dns_close(x->resolver1.GetFD());
+               }
+
+               if (x->resolver2.GetFD() > 0)
+               {
+                       log(DEBUG,"Whacked resolver2");
+                       dns_close(x->resolver2.GetFD());
+               }
+       }
+#endif
+}
+
 void dns_poll(int fdcheck)
 {
        /* Check the given file descriptor is in valid range */