* | Inspire Internet Relay Chat Daemon |
* +------------------------------------+
*
- * Inspire is copyright (C) 2002-2004 ChatSpike-Dev.
+ * InspIRCd is copyright (C) 2002-2006 ChatSpike-Dev.
* E-mail:
* <brain@chatspike.net>
* <Craig@chatspike.net>
* ---------------------------------------------------
*/
-/* Now with added unF! ;) */
-
-using namespace std;
-
-#include "inspircd.h"
-#include "inspircd_io.h"
-#include "inspircd_util.h"
#include "inspircd_config.h"
+#include "inspircd.h"
+#include "configreader.h"
#include <unistd.h>
-#include <fcntl.h>
#include <sys/errno.h>
#include <sys/ioctl.h>
#include <sys/utsname.h>
-#include <cstdio>
-#include <time.h>
#include <string>
-#ifdef GCC3
-#include <ext/hash_map>
-#else
-#include <hash_map>
-#endif
-#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 "dynamic.h"
-#include "wildcard.h"
-#include "message.h"
-#include "mode.h"
-#include "commands.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/poll.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"
+#include "socket.h"
-#ifdef GCC3
-#define nspace __gnu_cxx
-#else
-#define nspace std
-#endif
-
-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);
- }
- };
-}
-
+extern ServerConfig* Config;
+extern InspIRCd* ServerInstance;
-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);
- }
+class Lookup;
-};
+Lookup* dnslist[MAX_DESCRIPTORS];
+Lookup* user_fd_to_dns[MAX_DESCRIPTORS];
+extern userrec* fd_ref_table[MAX_DESCRIPTORS];
-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 std::deque<command_t> command_table;
-
-extern user_hash clientlist;
-extern chan_hash chanlist;
-extern user_hash whowas;
-extern command_table cmdlist;
-extern address_cache IP;
-
-extern ClassVector Classes;
-
-extern char DNSServer[MAXBUF];
+//enum LookupState { reverse, forward };
class Lookup {
private:
- DNS* resolver;
- userrec* u;
+ char u[NICKMAX];
+ std::string hostname;
public:
+ DNS resolver1;
+ DNS resolver2;
+
Lookup()
{
- u = NULL;
- resolver = NULL;
+ *u = 0;
+ hostname = "";
}
- ~Lookup()
+ void Reset()
{
- if (resolver)
- delete resolver;
+ *u = 0;
+ hostname = "";
}
- Lookup(userrec* user)
+ ~Lookup()
{
- u = user;
- log(DEBUG,"New Lookup class with DNSServer set to '%s'",DNSServer);
- resolver = new DNS(std::string(DNSServer));
- resolver->ReverseLookup(std::string(user->host));
}
- bool Done()
+ bool DoLookup(std::string nick)
{
- if (resolver->HasResult())
+ hostname = "";
+ userrec* usr = Find(nick);
+ if (usr)
{
- log(DEBUG,"DNS Result available!");
- std::string hostname = resolver->GetResult();
- if (u)
+ resolver1.SetNS(std::string(Config->DNSServer));
+ if (!resolver1.ReverseLookup(std::string(usr->host), true))
{
- log(DEBUG,"Applying hostname lookup to %s: %s",u->nick,hostname.c_str());
- if (hostname != "")
- strlcpy(u->host,hostname.c_str(),MAXBUF);
- u->dns_done = true;
+ return false;
+ }
+ strlcpy(u,nick.c_str(),NICKMAX-1);
+
+ /* ASSOCIATE WITH DNS LOOKUP LIST */
+ if (resolver1.GetFD() != -1)
+ {
+ dnslist[resolver1.GetFD()] = this;
+ user_fd_to_dns[usr->fd] = this;
return true;
}
}
return false;
}
- int GetFD()
+ bool Done(int fdcheck)
{
- if (u)
+ if (hostname != "")
{
- return u->fd;
+ // doing forward lookup
+ userrec* usr = NULL;
+ if (resolver2.HasResult(fdcheck))
+ {
+ if (resolver2.GetFD() != -1)
+ {
+ dnslist[resolver2.GetFD()] = NULL;
+ std::string ip = resolver2.GetResultIP();
+ usr = Find(u);
+ if (usr)
+ {
+ if (usr->registered > 3)
+ {
+ usr->dns_done = true;
+ return true;
+ }
+ if ((hostname != "") && (usr->registered != 7))
+ {
+ if ((std::string(inet_ntoa(usr->ip4)) == ip) && (hostname.length() < 65))
+ {
+ if ((hostname.find_last_of(".in-addr.arpa") == hostname.length() - 1) && (hostname.find_last_of(".in-addr.arpa") != std::string::npos))
+ {
+ 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;
+ }
+ }
+ }
+ else
+ {
+ usr = Find(u);
+ if (usr)
+ {
+ usr->dns_done = true;
+ }
+ return true;
+ }
+ }
+ return false;
}
- else return 0;
+ else
+ {
+ // doing reverse lookup
+ userrec* usr = NULL;
+ if (resolver1.HasResult(fdcheck))
+ {
+ usr = Find(u);
+ 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)
+ {
+ dnslist[resolver1.GetFD()] = NULL;
+ 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, true);
+ if (resolver2.GetFD() != -1)
+ {
+ dnslist[resolver2.GetFD()] = this;
+ if (usr)
+ user_fd_to_dns[usr->fd] = this;
+ }
+ }
+ }
+ }
+ }
+ }
+ return false;
}
-};
-
-typedef std::vector<Lookup> dns_queue;
-dns_queue dnsq;
+ int GetFD()
+ {
+ userrec* usr = Find(u);
+ if (!usr)
+ return 0;
+ if (usr->dns_done)
+ return 0;
+ return usr->fd;
+ }
+};
-bool lookup_dns(userrec* u)
+bool lookup_dns(const std::string &nick)
{
- // place a new user into the queue...
- log(DEBUG,"Queueing DNS lookup for %s",u->nick);
- Lookup L(u);
- dnsq.push_back(L);
- return true;
+ /* First attempt to find the nickname */
+ userrec* u = Find(nick);
+ if (u)
+ {
+ /* Check the cache */
+ /*address_cache::iterator address = addrcache.find(u->ip4);
+ if (address != addrcache.end())
+ {
+ WriteServ(u->fd,"NOTICE Auth :*** Found your hostname (cached)");
+ log(DEBUG,"Found cached host");
+ strlcpy(u->host,address->second->c_str(),MAXBUF);
+ strlcpy(u->dhost,address->second->c_str(),MAXBUF);
+ u->dns_done = true;
+ return true;
+ }*/
+ /* If the user exists, create a new
+ * lookup object, and associate it
+ * with the user. The lookup object
+ * will maintain the reference table
+ * which we use for quickly finding
+ * dns results. Please note that we
+ * do not associate a lookup with a
+ * userrec* pointer and we use the
+ * nickname instead because, by the
+ * time the DNS lookup has completed,
+ * the nickname could have quit and
+ * if we then try and access the
+ * pointer we get a nice segfault.
+ */
+ Lookup* L = new Lookup();
+ L->DoLookup(nick);
+ return true;
+ }
+ return false;
}
-void dns_poll()
+void ZapThisDns(int fd)
{
- // do we have items in the queue?
- if (dnsq.size())
+#ifdef THREADED_DNS
+/* if (fd_ref_table[fd])
{
- log(DEBUG,"DNS items pending...");
- // are any ready, or stale?
- if (dnsq[0].Done() || (!dnsq[0].GetFD()))
+ if (fd_ref_table[fd]->registered >= 3)
{
- // if they are take them out...
- log(DEBUG,"****** DNS lookup for fd %d is complete. ******",dnsq[0].GetFD());
- dnsq.erase(dnsq.begin());
+ 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 */
+ if ((fdcheck < 0) || (fdcheck > MAX_DESCRIPTORS))
+ return;
+
+ /* Try and find the file descriptor in our list of
+ * active DNS lookups
+ */
+ Lookup *x = dnslist[fdcheck];
+ if (x)
+ {
+ /* If it exists check if its a valid fd still */
+ if (x->GetFD() != -1)
+ {
+ /* Check if its done, if it is delete it */
+ if (x->Done(fdcheck))
+ {
+ /* We don't need to delete the file descriptor
+ * from the socket engine, as dns.cpp tracks it
+ * for us if we are in single-threaded country.
+ */
+ DELETE(x);
+ }
+ }
+ else
+ {
+ /* its fd is dodgy, the dns code probably
+ * bashed it due to error. Free the class.
+ */
+ DELETE(x);
+ }
+ /* If we got down here, the dns lookup was valid, BUT,
+ * its still in progress. Be patient, and wait for
+ * more socketengine events to complete the lookups.
+ */
+ return;
+ }
+ /* This FD doesnt belong here, lets be rid of it,
+ * just to be safe so we dont get any more events
+ * about it.
+ */
+ if (ServerInstance && ServerInstance->SE)
+ ServerInstance->SE->DelFd(fdcheck);
+}