Add counter system for umodes to get rid of some O(n)

[user/henk/code/inspircd.git] / include / dns.h

diff --git a/include/dns.h b/include/dns.h
index 6c690dc8756550a4c958d26239ed6f4133793e9d..038323a4362b6b3e557cb3068614f16058915e71 100644 (file)
--- a/include/dns.h
+++ b/include/dns.h
@@ -2,14 +2,11 @@
  *       | Inspire Internet Relay Chat Daemon |
  *       +------------------------------------+
  *
- *  InspIRCd is copyright (C) 2002-2006 ChatSpike-Dev.
- *                     E-mail:
- *              <brain@chatspike.net>
- *              <Craig@chatspike.net>
+ *  InspIRCd: (C) 2002-2007 InspIRCd Development Team
+ * See: http://www.inspircd.org/wiki/index.php/Credits
  *
- * Written by Craig Edwards, Craig McLure, and others.
  * This program is free but copyrighted software; see
- *          the file COPYING for details.
+ *            the file COPYING for details.
  *
  * ---------------------------------------------------
  */
@@ -48,6 +45,7 @@ using irc::sockets::insp_sockaddr;
 using irc::sockets::insp_inaddr;
 
 class InspIRCd;
+class Module;
 
 /**
  * Result status, used internally
@@ -68,7 +66,9 @@ enum ResolverError
        RESOLVER_NSDOWN         =       1,
        RESOLVER_NXDOMAIN       =       2,
        RESOLVER_NOTREADY       =       3,
-       RESOLVER_BADIP          =       4
+       RESOLVER_BADIP          =       4,
+       RESOLVER_TIMEOUT        =       5,
+       RESLOVER_FORCEUNLOAD    =       6
 };
 
 /**
@@ -86,21 +86,12 @@ class DNSHeader;
  */
 class ResourceRecord;
 
-/**
- * A set of requests keyed by request id
- */
-typedef std::map<int,DNSRequest*> requestlist;
-
-/**
- * An iterator into a set of requests
- */
-typedef requestlist::iterator requestlist_iter;
-
 /**
  * Query and resource record types
  */
 enum QueryType
 {
+       DNS_QUERY_NONE  = 0,      /* Uninitialized Query */
        DNS_QUERY_A     = 1,      /* 'A' record: an ipv4 address */
        DNS_QUERY_CNAME = 5,      /* 'CNAME' record: An alias */
        DNS_QUERY_PTR   = 12,     /* 'PTR' record: a hostname */
@@ -145,6 +136,10 @@ class Resolver : public Extensible
         * Pointer to creator
         */
        InspIRCd* ServerInstance;
+       /**
+        * Pointer to creator module (if any, or NULL)
+        */
+       Module* Creator;
        /**
         * The input data, either a host or an IP address
         */
@@ -192,8 +187,14 @@ class Resolver : public Extensible
         * event a lookup could not be allocated, or a similar hard error occurs such as
         * the network being down. This will also be thrown if an invalid IP address is
         * passed when resolving a 'PTR' record.
+        *
+        * NOTE: If you are instantiating your DNS lookup from a module, you should set the
+        * value of creator to point at your Module class. This way if your module is unloaded
+        * whilst lookups are in progress, they can be safely removed and your module will not
+        * crash the server.
         */
-       Resolver(InspIRCd* Instance, const std::string &source, QueryType qt);
+       Resolver(InspIRCd* Instance, const std::string &source, QueryType qt, Module* creator = NULL);
+
        /**
         * The default destructor does nothing.
         */
@@ -219,6 +220,11 @@ class Resolver : public Extensible
         * this method will return -1.
         */
        int GetId();
+
+       /**
+        * Returns the creator module, or NULL
+        */
+       Module* GetCreator();
 };
 
 /** DNS is a singleton class used by the core to dispatch dns
@@ -238,11 +244,6 @@ class DNS : public EventHandler
         */
        static const int MAX_REQUEST_ID = 0xFFFF;
 
-       /**
-        * Requests that are currently 'in flight'
-        */
-       requestlist requests;
-
        /**
         * Server address being used currently
         */
@@ -253,11 +254,6 @@ class DNS : public EventHandler
         */
        int currid;
 
-       /**
-        * Currently active Resolver classes
-        */
-       Resolver* Classes[MAX_REQUEST_ID];
-
        /**
         * We have to turn off a few checks on received packets
         * when people are using 4in6 (e.g. ::ffff:xxxx). This is
@@ -272,6 +268,14 @@ class DNS : public EventHandler
        int MakePayload(const char* name, const QueryType rr, const unsigned short rr_class, unsigned char* payload);
 
  public:
+       /**
+        * Currently active Resolver classes
+        */
+       Resolver* Classes[MAX_REQUEST_ID];
+       /**
+        * Requests that are currently 'in flight'
+        */
+       DNSRequest* requests[MAX_REQUEST_ID];
        /**
         * The port number DNS requests are made on,
         * and replies have as a source-port number.
@@ -330,7 +334,7 @@ class DNS : public EventHandler
         * Handle a SocketEngine read event
         * Inherited from EventHandler
         */
-       void HandleEvent(EventType et);
+       void HandleEvent(EventType et, int errornum = 0);
 
        /**
         * Add a Resolver* to the list of active classes
@@ -364,6 +368,13 @@ class DNS : public EventHandler
         * Turn an in6_addr into a .ip6.arpa domain
         */
        static void MakeIP6Int(char* query, const in6_addr *ip);
+
+       /**
+        * Clean out all dns resolvers owned by a particular
+        * module, to make unloading a module safe if there
+        * are dns requests currently in progress.
+        */
+       void CleanResolvers(Module* module);
 };
 
 #endif