]> git.netwichtig.de Git - user/henk/code/inspircd.git/blobdiff - src/dnsqueue.cpp
EXPERIMENTAL new socket engine code
[user/henk/code/inspircd.git] / src / dnsqueue.cpp
index 80189a95e17f4815883158950e8dad92873df0f9..99f32988e55f382d60f1a2aaee2fbcc37ed9c843 100644 (file)
 
 using namespace std;
 
+#include "inspircd_config.h"
 #include "inspircd.h"
 #include "inspircd_io.h"
 #include "inspircd_util.h"
-#include "inspircd_config.h"
 #include <unistd.h>
-#include <fcntl.h>
 #include <sys/errno.h>
 #include <sys/ioctl.h>
 #include <sys/utsname.h>
@@ -38,14 +37,8 @@ using namespace std;
 #include <map>
 #include <sstream>
 #include <vector>
-#include <errno.h>
 #include <deque>
-#include <errno.h>
-#include <unistd.h>
-#include <sched.h>
-#include "connection.h"
 #include "users.h"
-#include "servers.h"
 #include "ctables.h"
 #include "globals.h"
 #include "modules.h"
@@ -57,7 +50,6 @@ using namespace std;
 #include "xline.h"
 #include "inspstring.h"
 #include "dnsqueue.h"
-#include <stdlib.h>
 #include <time.h>
 #include <sys/types.h>
 #include <sys/socket.h>
@@ -65,100 +57,27 @@ using namespace std;
 #include <sys/time.h>
 #include <netinet/in.h>
 #include <string.h>
-#include <unistd.h>
-#include <stdio.h>
-#include <errno.h>
-#include <fcntl.h>
 #include "dns.h"
-
-#ifdef GCC3
-#define nspace __gnu_cxx
-#else
-#define nspace std
-#endif
+#include "helperfuncs.h"
+#include "hashcomp.h"
 
 extern int MaxWhoResults;
 
 extern std::vector<Module*> modules;
 extern std::vector<std::string> module_names;
 extern std::vector<ircd_module*> factory;
-extern std::vector<int> fd_reap;
 
 extern int MODCOUNT;
 
-namespace nspace
-{
-#ifdef GCC34
-        template<> struct hash<in_addr>
-#else
-        template<> struct nspace::hash<in_addr>
-#endif
-        {
-                size_t operator()(const struct in_addr &a) const
-                {
-                        size_t q;
-                        memcpy(&q,&a,sizeof(size_t));
-                        return q;
-                }
-        };
-#ifdef GCC34
-        template<> struct hash<string>
-#else
-        template<> struct nspace::hash<string>
-#endif
-        {
-                size_t operator()(const string &s) const
-                {
-                        char a[MAXBUF];
-                        static struct hash<const char *> strhash;
-                        strlcpy(a,s.c_str(),MAXBUF);
-                        strlower(a);
-                        return strhash(a);
-                }
-        };
-}
-
-
-struct StrHashComp
-{
-
-       bool operator()(const string& s1, const string& s2) const
-       {
-               char a[MAXBUF],b[MAXBUF];
-               strlcpy(a,s1.c_str(),MAXBUF);
-               strlcpy(b,s2.c_str(),MAXBUF);
-                strlower(a);
-                strlower(b);
-               return (strcasecmp(a,b) == 0);
-       }
-
-};
-
-struct InAddr_HashComp
-{
-
-       bool operator()(const in_addr &s1, const in_addr &s2) const
-       {
-               size_t q;
-               size_t p;
-               
-               memcpy(&q,&s1,sizeof(size_t));
-               memcpy(&p,&s2,sizeof(size_t));
-               
-               return (q == p);
-       }
-
-};
-
-
-typedef nspace::hash_map<std::string, userrec*, nspace::hash<string>, StrHashComp> user_hash;
-typedef nspace::hash_map<std::string, chanrec*, nspace::hash<string>, StrHashComp> chan_hash;
-typedef nspace::hash_map<in_addr,string*, nspace::hash<in_addr>, InAddr_HashComp> address_cache;
+typedef nspace::hash_map<std::string, userrec*, nspace::hash<string>, irc::StrHashComp> user_hash;
+typedef nspace::hash_map<std::string, chanrec*, nspace::hash<string>, irc::StrHashComp> chan_hash;
+typedef nspace::hash_map<in_addr,string*, nspace::hash<in_addr>, irc::InAddr_HashComp> address_cache;
+typedef nspace::hash_map<std::string, WhoWasUser*, nspace::hash<string>, irc::StrHashComp> whowas_hash;
 typedef std::deque<command_t> command_table;
 
 extern user_hash clientlist;
 extern chan_hash chanlist;
-extern user_hash whowas;
+extern whowas_hash whowas;
 extern command_table cmdlist;
 
 extern ClassVector Classes;
@@ -168,10 +87,14 @@ long max_fd_alloc = 0;
 
 extern time_t TIME;
 
+//enum LookupState { reverse, forward };
+
 class Lookup {
 private:
-       DNS resolver;
+       DNS resolver1;
+       DNS resolver2;
        char u[NICKMAX];
+       std::string hostname;
 public:
        Lookup()
        {
@@ -181,6 +104,7 @@ public:
        void Reset()
        {
                strcpy(u,"");
+               log(DEBUG,"Reset class Lookup");
        }
 
        ~Lookup()
@@ -189,55 +113,109 @@ public:
 
        bool DoLookup(std::string nick)
        {
+               hostname = "";
                userrec* usr = Find(nick);
                if (usr)
                {
                        log(DEBUG,"New Lookup class for %s with DNSServer set to '%s'",nick.c_str(),DNSServer);
-                       resolver.SetNS(std::string(DNSServer));
-                       if (!resolver.ReverseLookup(std::string(usr->host)))
+                       resolver1.SetNS(std::string(DNSServer));
+                       if (!resolver1.ReverseLookup(std::string(usr->host)))
+                       {
+                               log(DEBUG,"ReverseLookup didnt return true, we're outta here");
                                return false;
+                       }
                        strlcpy(u,nick.c_str(),NICKMAX);
                        return true;
                }
+               log(DEBUG,"We couldnt find that user");
                return false;
        }
 
        bool Done()
        {
-               userrec* usr = NULL;
-               if (resolver.HasResult())
+               if (hostname != "")
                {
-                       if (resolver.GetFD() != 0)
+                       log(DEBUG,"Doing forward lookup here with host %s",hostname.c_str());
+                       // doing forward lookup
+                       userrec* usr = NULL;
+                       if (resolver2.HasResult())
                        {
-                               std::string hostname = resolver.GetResult();
-                               usr = Find(u);
-                               if (usr)
+                               log(DEBUG,"resolver2 has result");
+                               if (resolver2.GetFD() != 0)
                                {
-                                       if (usr->registered > 3)
+                                       std::string ip = resolver2.GetResultIP();
+                                       log(DEBUG,"FORWARD RESULT! %s",ip.c_str());
+
+                                       usr = Find(u);
+                                       if (usr)
                                        {
-                                               usr->dns_done = true;
-                                               return true;
+                                               if (usr->registered > 3)
+                                               {
+                                                       log(DEBUG,"Point 1: Returning true as usr->dns_done is true");
+                                                       usr->dns_done = true;
+                                                       return true;
+                                               }
+                                               if ((hostname != "") && (usr->registered != 7))
+                                               {
+                                                       if (std::string(usr->ip) == ip)
+                                                       {
+                                                               strlcpy(usr->host,hostname.c_str(),MAXBUF);
+                                                               strlcpy(usr->dhost,hostname.c_str(),MAXBUF);
+                                                               log(DEBUG,"Forward and reverse match, assigning hostname");
+                                                       }
+                                                       else
+                                                       {
+                                                               log(DEBUG,"AWOOGA! Forward lookup doesn't match reverse: R='%s',F='%s',IP='%s'",hostname.c_str(),ip.c_str(),usr->ip);
+                                                       }
+                                                       usr->dns_done = true;
+                                                       return true;
+                                               }
                                        }
-                                       if ((hostname != "") && (usr->registered != 7))
+                               }
+                               else
+                               {
+                                       usr = Find(u);
+                                       if (usr)
                                        {
-                                               strlcpy(usr->host,hostname.c_str(),MAXBUF);
-                                               strlcpy(usr->dhost,hostname.c_str(),MAXBUF);
-                                               WriteServ(usr->fd,"NOTICE Auth :Resolved your hostname: %s",hostname.c_str());
+                                               log(DEBUG,"Point 2: Returning true");
                                                usr->dns_done = true;
-                                               return true;
                                        }
-                                       usr->dns_done = true;
                                        return true;
                                }
                        }
-                       else
+                       log(DEBUG,"Returning false in forward");
+                       return false;
+               }
+               else
+               {
+                       // doing reverse lookup
+                       userrec* usr = NULL;
+                       if (resolver1.HasResult())
                        {
                                usr = Find(u);
-                               if (usr)
-                                       usr->dns_done = true;
-                               return true;
+                               if ((usr) && (usr->dns_done))
+                                       return true;
+                               if (resolver1.GetFD() != 0)
+                               {
+                                       hostname = resolver1.GetResult();
+                                       if (usr)
+                                       {
+                                               if ((usr->registered > 3) || (hostname == ""))
+                                               {
+                                                       log(DEBUG,"Hostname is blank and user->registered > 3, returning true and setting done");
+                                                       usr->dns_done = true;
+                                                       return true;
+                                               }
+                                       }
+                                       if (hostname != "")
+                                       {
+                                               log(DEBUG,"Starting forwardlookup now for host '%s'...",hostname.c_str());
+                                               resolver2.ForwardLookup(hostname);
+                                       }
+                               }
                        }
                }
+               log(DEBUG,"Returning false");
                return false;
        }
 
@@ -252,7 +230,7 @@ public:
        }
 };
 
-Lookup dnsq[MAXBUF];
+Lookup dnsq[255];
 
 bool lookup_dns(std::string nick)
 {
@@ -265,7 +243,7 @@ bool lookup_dns(std::string nick)
                Lookup L;
                if (L.DoLookup(nick))
                {
-                       for (int j = 0; j < MAXBUF; j++)
+                       for (int j = 0; j < 255; j++)
                        {
                                if (!dnsq[j].GetFD())
                                {
@@ -274,7 +252,7 @@ bool lookup_dns(std::string nick)
                                }
                        }
                        // calculate the maximum value, this saves cpu time later
-                       for (int p = 0; p < MAXBUF; p++)
+                       for (int p = 0; p < 255; p++)
                                if (dnsq[p].GetFD())
                                        max_fd_alloc = p;
                }
@@ -302,9 +280,8 @@ void dns_poll()
        }
        // looks like someones freed an item, recalculate end of list.
        if ((!dnsq[max_fd_alloc].GetFD()) && (max_fd_alloc != 0))
-               for (int p = 0; p < MAXBUF; p++)
+               for (int p = 0; p < 255; p++)
                        if (dnsq[p].GetFD())
                                max_fd_alloc = p;
 
 }
-