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>
#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"
#include "xline.h"
#include "inspstring.h"
#include "dnsqueue.h"
-#include <stdlib.h>
#include <time.h>
#include <sys/types.h>
#include <sys/socket.h>
#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"
+#include "helperfuncs.h"
+#include "hashcomp.h"
+#include "socketengine.h"
-#ifdef GCC3
-#define nspace __gnu_cxx
-#else
-#define nspace std
-#endif
+extern SocketEngine* SE;
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);
- 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;
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()
{
strcpy(u,"");
- resolver = NULL;
}
void Reset()
{
strcpy(u,"");
- if (resolver)
- delete resolver;
- resolver = NULL;
+ log(DEBUG,"Reset class Lookup");
}
~Lookup()
{
- if (resolver)
- delete resolver;
- resolver = NULL;
}
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 = new DNS(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()
+ bool Done(int fdcheck)
{
- 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(fdcheck))
{
- 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(fdcheck))
{
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;
}
}
};
-Lookup dnsq[MAXBUF];
+Lookup dnsq[255];
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())
{
}
}
// 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;
}
return false;
}
-void dns_poll()
+void dns_poll(int fdcheck)
{
// do we have items in the queue?
for (int j = 0; j <= max_fd_alloc; j++)
// are any ready, or stale?
if (dnsq[j].GetFD())
{
- if (dnsq[j].Done())
+ if (dnsq[j].Done(fdcheck))
{
dnsq[j].Reset();
}
}
// 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;
}
-