Add timeouts to the http module. Two seperate timeouts, 60 seconds to receive headers...

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

diff --git a/include/dns.h b/include/dns.h
index f66d2d3c406410338d7e1b8c5b88bd827fa7acab..6c690dc8756550a4c958d26239ed6f4133793e9d 100644 (file)
--- a/include/dns.h
+++ b/include/dns.h
@@ -37,8 +37,17 @@ Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 
 #include <string>
 #include "inspircd_config.h"
-#include "socket.h"
 #include "base.h"
+#include "socketengine.h"
+#include "socket.h"
+
+using namespace std;
+using irc::sockets::insp_aton;
+using irc::sockets::insp_ntoa;
+using irc::sockets::insp_sockaddr;
+using irc::sockets::insp_inaddr;
+
+class InspIRCd;
 
 /**
  * Result status, used internally
@@ -132,6 +141,10 @@ enum ForceProtocol
 class Resolver : public Extensible
 {
  protected:
+       /**
+        * Pointer to creator
+        */
+       InspIRCd* ServerInstance;
        /**
         * The input data, either a host or an IP address
         */
@@ -180,7 +193,7 @@ class Resolver : public Extensible
         * the network being down. This will also be thrown if an invalid IP address is
         * passed when resolving a 'PTR' record.
         */
-       Resolver(const std::string &source, QueryType qt);
+       Resolver(InspIRCd* Instance, const std::string &source, QueryType qt);
        /**
         * The default destructor does nothing.
         */
@@ -213,10 +226,12 @@ class Resolver : public Extensible
  * back to Resolver objects, based upon the request ID. You
  * should never use this class yourself.
  */
-class DNS : public Extensible
+class DNS : public EventHandler
 {
  private:
 
+       InspIRCd* ServerInstance;
+
        /**
         * The maximum value of a dns request id,
         * 16 bits wide, 0xFFFF.
@@ -233,11 +248,6 @@ class DNS : public Extensible
         */
        insp_inaddr myserver;
 
-       /**
-        * File descriptor being used to perform queries
-        */
-       static int MasterSocket;
-
        /**
         * A counter used to form part of the pseudo-random id
         */
@@ -280,11 +290,6 @@ class DNS : public Extensible
         * Empty out a header into a data stream ready for transmission "on the wire"
         */
        static void EmptyHeader(unsigned char *output, const DNSHeader *header, const int length);
-       /**
-        * Get the master socket fd, used internally
-        */
-       static int GetMasterSocket();
-
        /**
         * Start the lookup of an ipv4 from a hostname
         */
@@ -323,8 +328,9 @@ class DNS : public Extensible
 
        /**
         * Handle a SocketEngine read event
+        * Inherited from EventHandler
         */
-       void MarshallReads(int fd);
+       void HandleEvent(EventType et);
 
        /**
         * Add a Resolver* to the list of active classes
@@ -340,7 +346,7 @@ class DNS : public Extensible
         * The constructor initialises the dns socket,
         * and clears the request lists.
         */
-       DNS();
+       DNS(InspIRCd* Instance);
 
        /**
         * Destructor